Skip to content

Commit 34d917a

Browse files
authored
Fix flaky test (#619)
1 parent 2228ad0 commit 34d917a

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

test/unit/client/test_client.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44
from __future__ import unicode_literals
55
import json
66

7-
from mock import Mock
7+
from mock import Mock, ANY
88
import pytest
99
from six import text_type, BytesIO, int2byte, PY2
1010

@@ -647,7 +647,8 @@ def test_create_legal_hold_policy_returns_the_correct_policy_object(mock_client,
647647
create_policy_response.json.return_value.update(params)
648648
mock_box_session.post.return_value = create_policy_response
649649
new_policy = mock_client.create_legal_hold_policy(test_policy_name, **params)
650-
mock_box_session.post.assert_called_once_with(expected_url, data=value)
650+
mock_box_session.post.assert_called_once_with(expected_url, data=ANY)
651+
assert dict(json.loads(mock_box_session.post.call_args[1]['data'])) == dict(json.loads(value))
651652
assert isinstance(new_policy, LegalHoldPolicy)
652653
assert new_policy.policy_name == test_policy_name
653654
for param in params:

0 commit comments

Comments
 (0)