[FME-6993] - Pausing - E2E Test 5

Split/Storage/Splits/SplitsStorage.swift

@@ -169,7 +169,6 @@ class DefaultSplitsStorage: SplitsStorage {
                     segmentsInUse += 1
                 } else if inMemorySplits.value(forKey: splitName) != nil && !active { // If known Split and archived
                     segmentsInUse -= 1
-
                 }
             }
 

SplitTests/Integration/streaming/MySegmentUpdateTest.swift

@@ -385,6 +385,97 @@ class MySegmentUpdateTest: XCTestCase {
             destroy(client)
         }
     }
+
+    func testSdkRestartMembershipsSyncIfNewFlag() throws {
+
+        var sdkReadyFired = false
+        var cacheReadyFired = true
+        let sdkReady = XCTestExpectation(description: "SDK should be ready")
+        let cacheReadyExp = XCTestExpectation(description: "Cache should be ready")
+        let segmentsHit = XCTestExpectation(description: "/memberships should be hit at least once")
+        var membershipsHit = 0
+
+        var json = IntegrationHelper.loadSplitChangeFileJson(name: "splitschanges_no_segments", sourceClass: IntegrationHelper()) // no Segments
+
+        // 1. Configure dispatcher
+        let dispatcher: HttpClientTestDispatcher = { request in
+            if request.url.absoluteString.contains("/splitChanges") {
+                return TestDispatcherResponse(code: 200, data: Data(json!.utf8))
+            }
+
+            if request.url.absoluteString.contains("/memberships") {
+                segmentsHit.fulfill()
+                membershipsHit += 1
+                return TestDispatcherResponse(code: 200, data: Data(IntegrationHelper.emptyMySegments.utf8))
+            }
+
+            return TestDispatcherResponse(code: 200)
+        }
+
+        // 2. Setup Factory, Network & Client
+        let splitConfig: SplitClientConfig = SplitClientConfig()
+        splitConfig.featuresRefreshRate = 5
+        splitConfig.segmentsRefreshRate = 5
+        splitConfig.impressionRefreshRate = 30
+        splitConfig.sdkReadyTimeOut = 60000
+        splitConfig.eventsPerPush = 10
+        splitConfig.streamingEnabled = false
+        splitConfig.eventsQueueSize = 100
+        splitConfig.eventsPushRate = 999999
+        splitConfig.eventsFirstPushWindow = 999
+        splitConfig.impressionsMode = "DEBUG"
+        splitConfig.serviceEndpoints = ServiceEndpoints.builder()
+        .set(sdkEndpoint: "localhost").set(eventsEndpoint: "localhost").build()
+
+        let splitDatabase = TestingHelper.createTestDatabase(name: "ready_from_cache_test")
+        splitDatabase.generalInfoDao.update(info: .flagsSpec, stringValue: "1.3")
+
+        let userKey = "test-user-key"
+        let key: Key = Key(matchingKey: userKey, bucketingKey: nil)
+        let session = HttpSessionMock()
+        let reqManager = HttpRequestManagerTestDispatcher(dispatcher: dispatcher, streamingHandler: buildStreamingHandler())
+        httpClient = DefaultHttpClient(session: session, requestManager: reqManager)
+        let builder = DefaultSplitFactoryBuilder()
+
+        _ = builder.setTestDatabase(splitDatabase)
+        _ = builder.setHttpClient(httpClient)
+        var factory = builder.setApiKey(apiKey).setKey(key).setConfig(splitConfig).build()
+        let client = factory?.client
+
+        client?.on(event: .sdkReady) {
+            sdkReadyFired = true
+            sdkReady.fulfill()
+        }
+
+        client?.on(event: .sdkReadyFromCache) {
+            cacheReadyExp.fulfill()
+            cacheReadyFired = true
+        }
+
+        wait(for: [segmentsHit], timeout: 3)
+        XCTAssertEqual(sdkReadyFired, false)
+
+        wait(for: [cacheReadyExp, sdkReady], timeout: 4)
+
+        // MARK: Key part
+        var waitExp = XCTestExpectation(description: "Just waiting")
+        waitExp.isInverted = true // Inverted expectation
+        wait(for: [waitExp], timeout: 10)
+        XCTAssertEqual(membershipsHit, 1, "After some time, if segments are not used, SDK shouldn't hit /memberships")
+
+        // MARK: Key part 2
+        json = IntegrationHelper.loadSplitChangeFileJson(name: "splitchanges_1", sourceClass: IntegrationHelper()) // splitChanges, now WITH Segments
+
+        waitExp = XCTestExpectation(description: "Just waiting")
+        waitExp.isInverted = true // Inverted expectation
+        wait(for: [waitExp], timeout: 15)
+        XCTAssertGreaterThan(membershipsHit, 2, "If new flags with segments arrive, the mechanism should be restarted and SDK should hit /memberships many times again")
+
+        // Cleanup
+        if let client = client {
+            destroy(client)
+        }
+    }
 
     func testMySegmentsUpdateBounded() throws {
         try mySegmentsUpdateBoundedTest(type: .mySegmentsUpdate)