def test_assert_requests_post_is_called_once():
token = 'my-test-token-123'
aut = Automator(token)
stashed = aut.requests.post
droplet_id = 212611563
full_url = 'https://api.digitalocean.com/v2/droplets/%s/actions' % droplet_id
data = {"type": "shutdown"}
json_headers = {'Content-Type': 'application/json'}
# mock response w/ success status
responses.reset()
shutdown_droplet_success_response(droplet_id)
mocked_response = requests.post(full_url)
stashed = aut._Automator__check_action_status
stashed2 = aut.requests.post
aut._Automator__check_action_status = MagicMock(return_value='completed')
aut.requests.post = MagicMock(return_value=mocked_response)
aut.turnoff_droplet(droplet_id)
aut.requests.post.assert_called_once_with(full_url,
json=data,
headers=json_headers)
aut._Automator__check_action_status.assert_called_once_with(
mocked_response.json()['action']['id'])
aut._Automator__check_action_status = stashed
aut.requests.post = stashed2
def test_assert_request_post_is_called_with_args():
floating_ip = '124.52.234.6'
droplet_id = 42312552
full_url = 'https://api.digitalocean.com/v2/floating_ips/%s/actions' % floating_ip
data = {"type": "assign", "droplet_id": 8219222}
json_headers = {'Content-Type': 'application/json'}
token = 'my-test-token-123'
aut = Automator(token)
stashed = aut.requests.post
stashed2 = aut._Automator__check_action_status
# mock response w/ success status
responses.reset()
assign_floating_ip_success_response(floating_ip, droplet_id)
mocked_response = requests.post(full_url)
aut._Automator__check_action_status = MagicMock(return_value='completed')
aut.requests.post = MagicMock(return_value=mocked_response)
aut.assign_floating_ip_to_droplet('124.52.234.6', 8219222)
aut.requests.post.assert_called_once_with(full_url,
json=data,
headers=json_headers)
aut.requests.post = stashed
aut._Automator__check_action_status = stashed2
def test_if_response_status_201_return_completed():
token = 'my-test-token-123'
aut = Automator(token)
droplet_id = 212611563
# response mock w/ 201 response
shutdown_droplet_success_response(droplet_id)
stashed = aut._Automator__check_action_status
aut._Automator__check_action_status = MagicMock(return_value='completed')
result = aut.turnoff_droplet(droplet_id)
assert result == 'completed'
aut._Automator__check_action_status = stashed
def test_if_response_status_201_return_completed():
token = 'my-test-token-123'
aut = Automator(token)
floating_ip = '135.155.152.2'
droplet_id = 42143152
# response mock w/ 201 response
assign_floating_ip_success_response(floating_ip, droplet_id)
stashed = aut._Automator__check_action_status
aut._Automator__check_action_status = MagicMock(return_value='completed')
result = aut.assign_floating_ip_to_droplet(floating_ip, droplet_id)
assert result == 'completed'
aut._Automator__check_action_status = stashed
def test_assert_requests_post_is_called_once():
token = 'my-test-token-123'
aut = Automator(token)
stashed = aut.requests.post
floating_ip = '124.52.234.6'
droplet_id = 42312552
full_url = 'https://api.digitalocean.com/v2/floating_ips/%s/actions' % floating_ip
# mock response w/ success status
responses.reset()
assign_floating_ip_success_response(floating_ip, droplet_id)
mocked_response = requests.post(full_url)
stashed = aut._Automator__check_action_status
stashed2 = aut.requests.post
aut._Automator__check_action_status = MagicMock(return_value='completed')
aut.requests.post = MagicMock(return_value=mocked_response)
aut.assign_floating_ip_to_droplet(floating_ip, droplet_id)
aut.requests.post.assert_called_once()
aut._Automator__check_action_status = stashed
aut.requests.post = stashed2