Tests/UnitTests/GTXBaseTestCase.m

//
// Copyright 2018 Google Inc.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//      http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
//

#import <UIKit/UIKit.h>
#import <XCTest/XCTest.h>

#import "GTXBaseTestCase.h"

GTXCheckHandlerBlock noOpCheckBlock = ^BOOL(id _Nonnull element, GTXErrorRefType errorOrNil) {
  return YES;
};

@implementation GTXBaseTestCase

- (NSObject *)newAccessibleElement {
  NSObject *element = [[NSObject alloc] init];
  element.isAccessibilityElement = YES;
  return element;
}

- (NSObject *)newInAccessibleElement {
  NSObject *element = [[NSObject alloc] init];
  element.isAccessibilityElement = NO;
  return element;
}

- (void)createTreeFromPreOrderTraversal:(NSArray *)preOrderTraversal {
  //   A
  //  / \
  // B   C
  //    / \
  //   D   E
  // Example Traversal: A, B, C, NSNull, NSNull, D, E, NSNull, (D's children).
  NSMutableArray *queue = [[NSMutableArray alloc] init];
  NSMutableArray *children = [[NSMutableArray alloc] init];
  for (id element in preOrderTraversal) {
    if ([element isKindOfClass:[NSNull class]]) {
      // End of list has been reached.
      [[queue firstObject] setAccessibilityElements:children];
      [queue removeObjectAtIndex:0];
      children = [[NSMutableArray alloc] init];
    } else {
      if ([queue count]) {
        [children addObject:element];
      }
      [queue addObject:element];
    }
  }
}

@end