-
Notifications
You must be signed in to change notification settings - Fork 214
/
test_legal_hold_assignment.py
49 lines (42 loc) · 2.06 KB
/
test_legal_hold_assignment.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
# coding: utf-8
from unittest.mock import Mock
import pytest
from boxsdk.object.legal_hold_policy_assignment import LegalHoldPolicyAssignment
from boxsdk.config import API
from boxsdk.network.default_network import DefaultNetworkResponse
def test_get(test_legal_hold_policy_assignment, mock_box_session):
expected_url = f'{API.BASE_API_URL}/legal_hold_policy_assignments/{test_legal_hold_policy_assignment.object_id}'
mock_box_session.get.return_value.json.return_value = {
'type': 'legal_hold_policy_assignment',
'id': test_legal_hold_policy_assignment.object_id,
'assigned_to': {
'type': 'user',
'id': '1234',
},
}
legal_hold_policy_assignment = test_legal_hold_policy_assignment.get()
mock_box_session.get.assert_called_once_with(expected_url, headers=None, params=None)
assert isinstance(legal_hold_policy_assignment, LegalHoldPolicyAssignment)
assert legal_hold_policy_assignment.type == 'legal_hold_policy_assignment'
assert legal_hold_policy_assignment.object_id == test_legal_hold_policy_assignment.object_id
assert legal_hold_policy_assignment['assigned_to']['type'] == 'user'
assert legal_hold_policy_assignment['assigned_to']['id'] == '1234'
@pytest.fixture(scope='module')
def delete_assignment_response():
# pylint:disable=redefined-outer-name
mock_network_response = Mock(DefaultNetworkResponse)
mock_network_response.ok = True
return mock_network_response
def test_delete_policy_return_the_correct_response(
test_legal_hold_policy_assignment,
mock_box_session,
delete_assignment_response,
):
# pylint:disable=redefined-outer-name
mock_box_session.delete.return_value = delete_assignment_response
response = test_legal_hold_policy_assignment.delete()
# pylint:disable=protected-access
expected_url = test_legal_hold_policy_assignment.get_url()
# pylint:enable = protected-access
mock_box_session.delete.assert_called_once_with(expected_url, params={}, expect_json_response=False, headers=None)
assert response is True