updated test

Author: chillaq

tests/api/test_splits_api.py

@@ -16,18 +16,18 @@ def test_fetch_split_changes(self, mocker):
         httpclient.get.return_value = client.HttpResponse(200, '{"prop1": "value1"}', {})
         split_api = splits.SplitsAPI(httpclient, 'some_api_key', SdkMetadata('1.0', 'some', '1.2.3.4'), mocker.Mock())
 
-        response = split_api.fetch_splits(123, FetchOptions())
+        response = split_api.fetch_splits(123, FetchOptions(False, None, 'set1,set2'))
         assert response['prop1'] == 'value1'
         assert httpclient.get.mock_calls == [mocker.call('sdk', 'splitChanges', 'some_api_key',
                                                          extra_headers={
                                                              'SplitSDKVersion': '1.0',
                                                              'SplitSDKMachineIP': '1.2.3.4',
                                                              'SplitSDKMachineName': 'some'
                                                          },
-                                                         query={'since': 123})]
+                                                         query={'since': 123, 'sets': 'set1,set2'})]
 
         httpclient.reset_mock()
-        response = split_api.fetch_splits(123, FetchOptions(True))
+        response = split_api.fetch_splits(123, FetchOptions(True, 123, 'set3'))
         assert response['prop1'] == 'value1'
         assert httpclient.get.mock_calls == [mocker.call('sdk', 'splitChanges', 'some_api_key',
                                                          extra_headers={
@@ -36,7 +36,7 @@ def test_fetch_split_changes(self, mocker):
                                                              'SplitSDKMachineName': 'some',
                                                              'Cache-Control': 'no-cache'
                                                          },
-                                                         query={'since': 123})]
+                                                         query={'since': 123, 'till': 123, 'sets': 'set3'})]
 
         httpclient.reset_mock()
         response = split_api.fetch_splits(123, FetchOptions(True, 123))
@@ -82,7 +82,7 @@ async def get(verb, url, key, query, extra_headers):
             return client.HttpResponse(200, '{"prop1": "value1"}', {})
         httpclient.get = get
 
-        response = await split_api.fetch_splits(123, FetchOptions())
+        response = await split_api.fetch_splits(123, FetchOptions(False, None, 'set1,set2'))
         assert response['prop1'] == 'value1'
         assert self.verb == 'sdk'
         assert self.url == 'splitChanges'
@@ -92,10 +92,10 @@ async def get(verb, url, key, query, extra_headers):
             'SplitSDKMachineIP': '1.2.3.4',
             'SplitSDKMachineName': 'some'
         }
-        assert self.query == {'since': 123}
+        assert self.query == {'since': 123, 'sets': 'set1,set2'}
 
         httpclient.reset_mock()
-        response = await split_api.fetch_splits(123, FetchOptions(True))
+        response = await split_api.fetch_splits(123, FetchOptions(True, 123, 'set3'))
         assert response['prop1'] == 'value1'
         assert self.verb == 'sdk'
         assert self.url == 'splitChanges'
@@ -106,7 +106,7 @@ async def get(verb, url, key, query, extra_headers):
             'SplitSDKMachineName': 'some',
             'Cache-Control': 'no-cache'
         }
-        assert self.query == {'since': 123}
+        assert self.query == {'since': 123, 'till': 123, 'sets': 'set3'}
 
         httpclient.reset_mock()
         response = await split_api.fetch_splits(123, FetchOptions(True, 123))