@@ -72,30 +72,29 @@ class PrivacyIconLogicTests: XCTestCase {
7272
7373 func testPrivacyIconIsShieldWithDotForHTTP( ) {
7474 let url = PrivacyIconLogicTests . insecurePageURL
75- let entity = Entity ( displayName: " E " , domains: [ ] , prevalence: TrackerInfo . Constants . majorNetworkPrevalence - 1.0 )
75+ let entity = Entity ( displayName: " E " , domains: [ ] , prevalence: 1.0 )
7676 let protectionStatus = ProtectionStatus ( unprotectedTemporary: false , enabledFeatures: [ ] , allowlisted: false , denylisted: false )
7777 let privacyInfo = PrivacyInfo ( url: url, parentEntity: entity, protectionStatus: protectionStatus)
7878
7979 let icon = PrivacyIconLogic . privacyIcon ( for: privacyInfo)
8080
8181 XCTAssertTrue ( url. isHttp)
8282 XCTAssertFalse ( privacyInfo. https)
83- XCTAssertFalse ( privacyInfo. isMajorTrackerNetwork)
8483 XCTAssertEqual ( icon, . shieldWithDot)
8584 }
8685
87- func testPrivacyIconIsShieldWithDotForMajorTrackerNetwork ( ) {
86+ func testPrivacyIconIsShieldWithoutDotForMajorTrackerNetwork ( ) {
8887 let url = PrivacyIconLogicTests . pageURL
89- let entity = Entity ( displayName: " E " , domains: [ ] , prevalence: TrackerInfo . Constants. majorNetworkPrevalence + 1.0 )
88+ // We don't have constants for major tracker network now now so just use a huge, unlikely prevalence
89+ let entity = Entity ( displayName: " E " , domains: [ ] , prevalence: 100.0 )
9090 let protectionStatus = ProtectionStatus ( unprotectedTemporary: false , enabledFeatures: [ ] , allowlisted: false , denylisted: false )
9191 let privacyInfo = PrivacyInfo ( url: url, parentEntity: entity, protectionStatus: protectionStatus)
9292
9393 let icon = PrivacyIconLogic . privacyIcon ( for: privacyInfo)
9494
9595 XCTAssertTrue ( url. isHttps)
9696 XCTAssertTrue ( privacyInfo. https)
97- XCTAssertTrue ( privacyInfo. isMajorTrackerNetwork)
98- XCTAssertEqual ( icon, . shieldWithDot)
97+ XCTAssertEqual ( icon, . shield)
9998 }
10099
101100}
0 commit comments