Merge pull request #60 from glebd/ctietze/addition-character-tests

Add tests to verify leading/trailing garbage doesn't pass during license verification
glebd · Dec 1, 2022 · e720d09 · e720d09
2 parents 72a8a15 + bcc4867
commit e720d09
Showing 1 changed file with 28 additions and 1 deletion.
diff --git a/swift5/CocoaFobTests/CocoaFobTests.swift b/swift5/CocoaFobTests/CocoaFobTests.swift
@@ -191,5 +191,32 @@ class CocoaFobTests: XCTestCase {
       XCTAssert(false, "\(error)")
     }
   }
-
+
+  func testVerifyAdditionalTrailingCharactersFail() {
+    let verifier = LicenseVerifier(publicKeyPEM: publicKeyPEM)
+    XCTAssertNotNil(verifier?.pubKey)
+    let name = "Joe Bloggs"
+    let regKey = "GAWQE-F9AQP-XJCCL-PAFAX-NU5XX-EUG6W-KLT3H-VTEB9-A9KHJ-8DZ5R-DL74G-TU4BN-7ATPY-3N4XB-V4V27-Qasdf"
+    let result = verifier?.verify(regKey, forName: name) ?? false
+    XCTAssertFalse(result)
+  }
+
+  func testVerifyAdditionalLeadingCharactersFail() {
+    let verifier = LicenseVerifier(publicKeyPEM: publicKeyPEM)
+    XCTAssertNotNil(verifier?.pubKey)
+    let name = "Joe Bloggs"
+    let regKey = "qwertGAWQE-F9AQP-XJCCL-PAFAX-NU5XX-EUG6W-KLT3H-VTEB9-A9KHJ-8DZ5R-DL74G-TU4BN-7ATPY-3N4XB-V4V27-Q"
+    let result = verifier?.verify(regKey, forName: name) ?? false
+    XCTAssertFalse(result)
+  }
+
+  func testVerifyWhitespaceInMiddleFails() {
+    let verifier = LicenseVerifier(publicKeyPEM: publicKeyPEM)
+    XCTAssertNotNil(verifier?.pubKey)
+    let name = "Joe Bloggs"
+    let regKey = "GAWQE-F9AQP- XJCCL-PAFAX-NU5XX - EUG6W-KLT3H-VTEB9\n-A9KHJ-8DZ5R- DL74G-\tTU4BN-7ATPY-3N4XB-V4V27-Q"
+    let result = verifier?.verify(regKey, forName: name) ?? false
+    XCTAssertFalse(result)
+  }
+
 }