Add test for embedded phishing dataset updates

duckduckgo · Oct 22, 2024 · 1549f59 · 1549f59
1 parent 8c43f96
commit 1549f59
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/Tests/PhishingDetectionTests/PhishingDetectionDataStoreTests.swift b/Tests/PhishingDetectionTests/PhishingDetectionDataStoreTests.swift
@@ -60,6 +60,22 @@ class PhishingDetectionDataStoreTests: XCTestCase {
         XCTAssertEqual(actualFilterSet, expectedFilerSet)
         XCTAssertEqual(actualHashPrefix, expectedHashPrefix)
     }
+
+    func testWhenEmbeddedRevisionNewerThanOnDisk_ThenLoadEmbedded() async {
+        mockDataProvider.embeddedRevision = 5
+        let expectedFilerSet = Set([Filter(hashValue: "some", regex: "some")])
+        let expectedHashPrefix = Set(["sassa"])
+        mockDataProvider.shouldReturnFilterSet(set: expectedFilerSet)
+        mockDataProvider.shouldReturnHashPrefixes(set: expectedHashPrefix)
+
+        let actualRevision = dataStore.currentRevision
+        let actualFilterSet = dataStore.filterSet
+        let actualHashPrefix = dataStore.hashPrefixes
+
+        XCTAssertEqual(actualFilterSet, expectedFilerSet)
+        XCTAssertEqual(actualHashPrefix, expectedHashPrefix)
+        XCTAssertEqual(actualRevision, 5)
+    }
 
     func testWriteAndLoadData() async {
         // Get and write data