material-components_materia.../components/FlexibleHeader/tests/unit/FlexibleHeaderViewTraitCollectionTests.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

69 lines
2.5 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 FlexibleHeaderViewTraitCollectionTests : XCTestCase
@end
@implementation FlexibleHeaderViewTraitCollectionTests
- (void)testTraitCollectionDidChangeBlockCalledWhenTraitCollectionChanges {
// Given
MDCFlexibleHeaderView *flexibleHeader = [[MDCFlexibleHeaderView alloc] init];
XCTestExpectation *expectation =
[self expectationWithDescription:@"Called traitCollectionDidChange"];
flexibleHeader.traitCollectionDidChangeBlock =
^(MDCFlexibleHeaderView *_Nonnull flexibleHeaderView,
UITraitCollection *_Nullable previousTraitCollection) {
[expectation fulfill];
};
// When
[flexibleHeader traitCollectionDidChange:nil];
// Then
[self waitForExpectations:@[ expectation ] timeout:1];
}
- (void)testTraitCollectionDidChangeBlockCalledWithExpectedParameters {
// Given
MDCFlexibleHeaderView *flexibleHeader = [[MDCFlexibleHeaderView alloc] init];
XCTestExpectation *expectation =
[self expectationWithDescription:@"Called traitCollectionDidChange"];
__block UITraitCollection *passedTraitCollection;
__block MDCFlexibleHeaderView *passedFlexibleHeader;
flexibleHeader.traitCollectionDidChangeBlock =
^(MDCFlexibleHeaderView *_Nonnull flexibleHeaderView,
UITraitCollection *_Nullable previousTraitCollection) {
passedTraitCollection = previousTraitCollection;
passedFlexibleHeader = flexibleHeaderView;
[expectation fulfill];
};
// When
UITraitCollection *testCollection = [UITraitCollection traitCollectionWithDisplayScale:77];
[flexibleHeader traitCollectionDidChange:testCollection];
// Then
[self waitForExpectations:@[ expectation ] timeout:1];
XCTAssertEqual(passedTraitCollection, testCollection);
XCTAssertEqual(passedFlexibleHeader, flexibleHeader);
}
@end