react-native/Examples/UIExplorer/UIExplorerUnitTests/RCTSparseArrayTests.m

// Copyright 2004-present Facebook. All Rights Reserved.

#import <XCTest/XCTest.h>

#import "RCTSparseArray.h"
#import "UIView+React.h"

@interface RCTSparseArrayTests : XCTestCase

@end

@implementation RCTSparseArrayTests

- (void)testDictionary
{
  NSObject<RCTViewNodeProtocol> *myView = [[UIView alloc] init];
  myView.reactTag = @4;

  NSObject<RCTViewNodeProtocol> *myOtherView = [[UIView alloc] init];
  myOtherView.reactTag = @5;

  RCTSparseArray *registry = [[RCTSparseArray alloc] init];
  XCTAssertNil(registry[@4], @"how did you have a view when none are registered?");
  XCTAssertNil(registry[@5], @"how did you have a view when none are registered?");

  registry[myView.reactTag] = myView;
  XCTAssertEqual(registry[@4], myView);
  XCTAssertNil(registry[@5], @"didn't register other view yet");

  registry[myOtherView.reactTag] = myOtherView;
  XCTAssertEqual(registry[@4], myView);
  XCTAssertEqual(registry[@5], myOtherView);

  registry[myView.reactTag] = nil;
  XCTAssertNil(registry[@4]);
  XCTAssertEqual(registry[@5], myOtherView);

  registry[myOtherView.reactTag] = nil;
  XCTAssertNil(registry[@4], @"how did you have a view when none are registered?");
  XCTAssertNil(registry[@5], @"how did you have a view when none are registered?");
}

@end