diff --git a/SnapTests/SnapTests.swift b/SnapTests/SnapTests.swift index bebb1b7..5a82a13 100644 --- a/SnapTests/SnapTests.swift +++ b/SnapTests/SnapTests.swift @@ -35,11 +35,16 @@ class SnapTests: XCTestCase { make.left.equalTo(v2.snp_top).offset(50) return } + + XCTAssertEqual(self.container.constraints().count, 2, "Should have 2 constraints installed") + v2.snp_makeConstraints { (make) -> Void in make.edges.equalTo(v1) return } + XCTAssertEqual(self.container.constraints().count, 6, "Should have 6 constraints installed") + } func testUpdateConstraints() { @@ -53,11 +58,16 @@ class SnapTests: XCTestCase { make.left.equalTo(v2.snp_top).offset(50) return } + + XCTAssertEqual(self.container.constraints().count, 2, "Should have 2 constraints installed") + v1.snp_updateConstraints { (make) -> Void in make.top.equalTo(v2.snp_top).offset(15) return } + XCTAssertEqual(self.container.constraints().count, 2, "Should still have 2 constraints installed") + } func testRemakeConstraints() { @@ -71,11 +81,16 @@ class SnapTests: XCTestCase { make.left.equalTo(v2.snp_top).offset(50) return } + + XCTAssertEqual(self.container.constraints().count, 2, "Should have 2 constraints installed") + v1.snp_remakeConstraints { (make) -> Void in make.edges.equalTo(v2) return } + XCTAssertEqual(self.container.constraints().count, 4, "Should have 4 constraints installed") + } func testRemoveConstraints() { @@ -89,8 +104,40 @@ class SnapTests: XCTestCase { make.left.equalTo(v2.snp_top).offset(50) return } + + XCTAssertEqual(self.container.constraints().count, 2, "Should have 2 constraints installed") + v1.snp_removeConstraints() + XCTAssertEqual(self.container.constraints().count, 0, "Should have 0 constraints installed") + + } + + func testPrepareConstraints() { + let v1 = UIView() + let v2 = UIView() + self.container.addSubview(v1) + self.container.addSubview(v2) + + let constraints = v1.snp_prepareConstraints { (make) -> Void in + make.edges.equalTo(v2) + return + } + + XCTAssertEqual(self.container.constraints().count, 0, "Should have 0 constraints installed") + + for constraint in constraints { + constraint.install() + } + + XCTAssertEqual(self.container.constraints().count, 4, "Should have 4 constraints installed") + + for constraint in constraints { + constraint.uninstall() + } + + XCTAssertEqual(self.container.constraints().count, 0, "Should have 0 constraints installed") + } }