def test_get_or_create_sailthru_list(self, mock_sailthru_client): """ Test the task the create sailthru lists. """ mock_sailthru_client.api_get.return_value = SailthruResponse(JsonResponse({'lists': []})) _get_or_create_user_list(mock_sailthru_client, 'test1_user_list') mock_sailthru_client.api_get.assert_called_with("list", {}) mock_sailthru_client.api_post.assert_called_with( "list", {'list': 'test1_user_list', 'primary': 0, 'public_name': 'test1_user_list'} ) # test existing user list mock_sailthru_client.api_get.return_value = \ SailthruResponse(JsonResponse({'lists': [{'name': 'test1_user_list'}]})) _get_or_create_user_list(mock_sailthru_client, 'test2_user_list') mock_sailthru_client.api_get.assert_called_with("list", {}) mock_sailthru_client.api_post.assert_called_with( "list", {'list': 'test2_user_list', 'primary': 0, 'public_name': 'test2_user_list'} ) # test get error from Sailthru mock_sailthru_client.api_get.return_value = \ SailthruResponse(JsonResponse({'error': 43, 'errormsg': 'Got an error'})) self.assertEqual(_get_or_create_user_list( mock_sailthru_client, 'test1_user_list'), None ) # test post error from Sailthru mock_sailthru_client.api_post.return_value = \ SailthruResponse(JsonResponse({'error': 43, 'errormsg': 'Got an error'})) mock_sailthru_client.api_get.return_value = SailthruResponse(JsonResponse({'lists': []})) self.assertEqual(_get_or_create_user_list(mock_sailthru_client, 'test2_user_list'), None)
def test_get_or_create_sailthru_list(self, mock_sailthru_client): """ Test the task the create sailthru lists. """ mock_sailthru_client.api_get.return_value = SailthruResponse(JsonResponse({"lists": []})) _get_or_create_user_list(mock_sailthru_client, "test1_user_list") mock_sailthru_client.api_get.assert_called_with("list", {}) mock_sailthru_client.api_post.assert_called_with( "list", {"list": "test1_user_list", "primary": 0, "public_name": "test1_user_list"} ) # test existing user list mock_sailthru_client.api_get.return_value = SailthruResponse( JsonResponse({"lists": [{"name": "test1_user_list"}]}) ) _get_or_create_user_list(mock_sailthru_client, "test2_user_list") mock_sailthru_client.api_get.assert_called_with("list", {}) mock_sailthru_client.api_post.assert_called_with( "list", {"list": "test2_user_list", "primary": 0, "public_name": "test2_user_list"} ) # test get error from Sailthru mock_sailthru_client.api_get.return_value = SailthruResponse( JsonResponse({"error": 43, "errormsg": "Got an error"}) ) self.assertEqual(_get_or_create_user_list(mock_sailthru_client, "test1_user_list"), None) # test post error from Sailthru mock_sailthru_client.api_post.return_value = SailthruResponse( JsonResponse({"error": 43, "errormsg": "Got an error"}) ) mock_sailthru_client.api_get.return_value = SailthruResponse(JsonResponse({"lists": []})) self.assertEqual(_get_or_create_user_list(mock_sailthru_client, "test2_user_list"), None)