Skip to content

Commit 436dd55

Browse files
committed
fix remaining integration tests
1 parent 5f3e28f commit 436dd55

File tree

1 file changed

+4
-8
lines changed

1 file changed

+4
-8
lines changed

tests/integration/test_client_e2e.py

Lines changed: 4 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -140,7 +140,7 @@ def test_get_treatments(self):
140140
result = client.get_treatments('invalidKey', ['invalid_feature'])
141141
assert len(result) == 1
142142
assert result['invalid_feature'] == 'control'
143-
self._validate_last_impressions(client, ('invalid_feature', 'invalidKey', 'control'))
143+
self._validate_last_impressions(client)
144144

145145
# testing a killed feature. No matter what the key, must return default treatment
146146
result = client.get_treatments('invalidKey', ['killed_feature'])
@@ -170,7 +170,6 @@ def test_get_treatments(self):
170170
client,
171171
('all_feature', 'invalidKey', 'on'),
172172
('killed_feature', 'invalidKey', 'defTreatment'),
173-
('invalid_feature', 'invalidKey', 'control'),
174173
('sample_feature', 'invalidKey', 'off')
175174
)
176175

@@ -191,7 +190,7 @@ def test_get_treatments_with_config(self):
191190
result = client.get_treatments_with_config('invalidKey', ['invalid_feature'])
192191
assert len(result) == 1
193192
assert result['invalid_feature'] == ('control', None)
194-
self._validate_last_impressions(client, ('invalid_feature', 'invalidKey', 'control'))
193+
self._validate_last_impressions(client)
195194

196195
# testing a killed feature. No matter what the key, must return default treatment
197196
result = client.get_treatments_with_config('invalidKey', ['killed_feature'])
@@ -221,7 +220,6 @@ def test_get_treatments_with_config(self):
221220
client,
222221
('all_feature', 'invalidKey', 'on'),
223222
('killed_feature', 'invalidKey', 'defTreatment'),
224-
('invalid_feature', 'invalidKey', 'control'),
225223
('sample_feature', 'invalidKey', 'off'),
226224
)
227225

@@ -395,7 +393,7 @@ def test_get_treatments(self):
395393
result = client.get_treatments('invalidKey', ['invalid_feature'])
396394
assert len(result) == 1
397395
assert result['invalid_feature'] == 'control'
398-
self._validate_last_impressions(client, ('invalid_feature', 'invalidKey', 'control'))
396+
self._validate_last_impressions(client)
399397

400398
# testing a killed feature. No matter what the key, must return default treatment
401399
result = client.get_treatments('invalidKey', ['killed_feature'])
@@ -425,7 +423,6 @@ def test_get_treatments(self):
425423
client,
426424
('all_feature', 'invalidKey', 'on'),
427425
('killed_feature', 'invalidKey', 'defTreatment'),
428-
('invalid_feature', 'invalidKey', 'control'),
429426
('sample_feature', 'invalidKey', 'off')
430427
)
431428

@@ -446,7 +443,7 @@ def test_get_treatments_with_config(self):
446443
result = client.get_treatments_with_config('invalidKey', ['invalid_feature'])
447444
assert len(result) == 1
448445
assert result['invalid_feature'] == ('control', None)
449-
self._validate_last_impressions(client, ('invalid_feature', 'invalidKey', 'control'))
446+
self._validate_last_impressions(client)
450447

451448
# testing a killed feature. No matter what the key, must return default treatment
452449
result = client.get_treatments_with_config('invalidKey', ['killed_feature'])
@@ -476,7 +473,6 @@ def test_get_treatments_with_config(self):
476473
client,
477474
('all_feature', 'invalidKey', 'on'),
478475
('killed_feature', 'invalidKey', 'defTreatment'),
479-
('invalid_feature', 'invalidKey', 'control'),
480476
('sample_feature', 'invalidKey', 'off'),
481477
)
482478

0 commit comments

Comments
 (0)