material-components_materia.../components/FlexibleHeader/tests/unit/FlexibleHeaderViewControllerTraitCollectionTests.m
Cody Weaver 1161a9db98
[FlexibleHeader] Add traitCollectionDidChange block (#7917)
The Flexible needs an API so clients can hook-in to trait collection changes. This additionally passes the flexibleHeader as a parameter so clients can modify the flexible header within the block.
2019-07-15 15:41:22 -07:00

71 lines
2.6 KiB
Objective-C

// Copyright 2019-present the Material Components for iOS authors. All Rights Reserved.
//
// 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 <XCTest/XCTest.h>
#import "MaterialFlexibleHeader.h"
@interface FlexibleHeaderViewControllerTraitCollectionTests : XCTestCase
@end
@implementation FlexibleHeaderViewControllerTraitCollectionTests
- (void)testTraitCollectionDidChangeBlockCalledWhenTraitCollectionChanges {
// Given
MDCFlexibleHeaderViewController *flexibleHeaderController =
[[MDCFlexibleHeaderViewController alloc] init];
XCTestExpectation *expectation =
[self expectationWithDescription:@"Called traitCollectionDidChange"];
flexibleHeaderController.traitCollectionDidChangeBlock =
^(MDCFlexibleHeaderViewController *_Nonnull flexibleHeaderViewController,
UITraitCollection *_Nullable previousTraitCollection) {
[expectation fulfill];
};
// When
[flexibleHeaderController traitCollectionDidChange:nil];
// Then
[self waitForExpectations:@[ expectation ] timeout:1];
}
- (void)testTraitCollectionDidChangeBlockCalledWithExpectedParameters {
// Given
MDCFlexibleHeaderViewController *flexibleHeaderController =
[[MDCFlexibleHeaderViewController alloc] init];
XCTestExpectation *expectation =
[self expectationWithDescription:@"Called traitCollectionDidChange"];
__block UITraitCollection *passedTraitCollection;
__block MDCFlexibleHeaderViewController *passedFlexibleHeader;
flexibleHeaderController.traitCollectionDidChangeBlock =
^(MDCFlexibleHeaderViewController *_Nonnull flexibleHeaderViewController,
UITraitCollection *_Nullable previousTraitCollection) {
passedTraitCollection = previousTraitCollection;
passedFlexibleHeader = flexibleHeaderViewController;
[expectation fulfill];
};
// When
UITraitCollection *testCollection = [UITraitCollection traitCollectionWithDisplayScale:77];
[flexibleHeaderController traitCollectionDidChange:testCollection];
// Then
[self waitForExpectations:@[ expectation ] timeout:1];
XCTAssertEqual(passedTraitCollection, testCollection);
XCTAssertEqual(passedFlexibleHeader, flexibleHeaderController);
}
@end