update EntityStateTests

This commit is contained in:
Igor Kravchenko 2020-09-30 16:59:34 +03:00
parent 53d2c27082
commit efa06d5a27
1 changed files with 3 additions and 0 deletions

View File

@ -169,6 +169,7 @@ class EntityStateTests: XCTestCase {
func testAddWithNoQualifierCreatesTypeProvider() {
state.add(MockComponent.self)
let provider = state.providers[MockComponent.identifier]
XCTAssertNotNil(provider)
XCTAssertTrue(provider is ComponentTypeProvider?)
XCTAssertTrue(provider?.getComponent() is MockComponent?)
}
@ -176,6 +177,7 @@ class EntityStateTests: XCTestCase {
func testAddWithTypeQualifierCreatesTypeProvider() {
state.add(MockComponent.self).withType(MockComponent2.self)
let provider = state.providers[MockComponent.identifier]
XCTAssertNotNil(provider)
XCTAssertTrue(provider is ComponentTypeProvider?)
XCTAssertTrue(provider?.getComponent() is MockComponent2?)
}
@ -202,6 +204,7 @@ class EntityStateTests: XCTestCase {
state.add(MockComponent.self).withMethod(dynamickProvider)
let provider = state.providers[MockComponent.identifier]
XCTAssertNotNil(provider)
XCTAssertTrue(provider is DynamicComponentProvider?)
XCTAssertTrue(provider?.getComponent() is MockComponent)
}