@@ -32,14 +32,40 @@ def test_update_info_returns_the_correct_response(
3232 assert update_response .object_id == test_collaboration .object_id
3333
3434
35+ @pytest .mark .parametrize ('data' , [
36+ {},
37+ {'role' : CollaborationRole .EDITOR , 'status' : CollaborationStatus .REJECTED },
38+ {'role' : CollaborationRole .EDITOR , 'status' : CollaborationStatus .ACCEPTED },
39+ {'role' : CollaborationRole .EDITOR , 'expires_at' : '2025-08-29T23:59:00-07:00' },
40+ {'role' : CollaborationRole .EDITOR , 'can_view_path' : True },
41+ ])
42+ def test_update_info_returns_the_correct_response_with_data_param (
43+ test_collaboration ,
44+ mock_box_session ,
45+ mock_collab_response ,
46+ data ):
47+ # pylint:disable=protected-access
48+ expected_url = test_collaboration .get_url ()
49+ mock_box_session .put .return_value = mock_collab_response
50+ update_response = test_collaboration .update_info (data = data )
51+ mock_box_session .put .assert_called_once_with (
52+ expected_url ,
53+ data = json .dumps (data ),
54+ headers = None ,
55+ params = None ,
56+ )
57+ assert isinstance (update_response , test_collaboration .__class__ )
58+ assert update_response .object_id == test_collaboration .object_id
59+
60+
3561def test_update_info_returns_204 (
3662 test_collaboration ,
3763 mock_box_session ):
3864 # pylint:disable=protected-access
3965 data = {'role' : CollaborationRole .OWNER , 'status' : CollaborationStatus .ACCEPTED }
4066 expected_url = test_collaboration .get_url ()
4167 mock_box_session .put .return_value .ok = True
42- is_success = test_collaboration .update_info (** data )
68+ is_success = test_collaboration .update_info (data = data )
4369 mock_box_session .put .assert_called_once_with (
4470 expected_url ,
4571 data = json .dumps (data ),
0 commit comments