def test_list_for_worker(self, request):
resp = create_mock_json("tests/resources/task_router/reservations_list.json")
resp.status_code = 200
request.return_value = resp
uri = "{0}/Reservations".format(BASE_WORKER_URI)
list_resource = Reservations(BASE_WORKER_URI, AUTH)
list_resource.list()
request.assert_called_with("GET", uri, params={}, auth=AUTH, use_json_extension=False)
def test_get(self, request):
resp = create_mock_json("tests/resources/task_router/reservations_instance.json")
resp.status_code = 200
request.return_value = resp
uri = "{0}/Reservations/{1}".format(BASE_URI, RESERVATION_SID)
list_resource = Reservations(BASE_URI, AUTH)
list_resource.get(RESERVATION_SID)
request.assert_called_with("GET", uri, auth=AUTH, use_json_extension=False)
def test_update_list(self, request):
resp = create_mock_json("tests/resources/task_router/reservations_instance.json")
resp.status_code = 201
request.return_value = resp
uri = "{0}/Reservations/{1}".format(BASE_URI, RESERVATION_SID)
list_resource = Reservations(BASE_URI, AUTH)
list_resource.update(
RESERVATION_SID, worker_activity_sid="WAaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa", reservation_status="rejected"
)
exp_params = {"WorkerActivitySid": "WAaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa", "ReservationStatus": "rejected"}
request.assert_called_with("POST", uri, data=exp_params, auth=AUTH, use_json_extension=False)
def test_get(self, request):
resp = create_mock_json(
'tests/resources/task_router/reservations_instance.json')
resp.status_code = 200
request.return_value = resp
uri = "{0}/Reservations/{1}".format(BASE_URI, RESERVATION_SID)
list_resource = Reservations(BASE_URI, AUTH)
list_resource.get(RESERVATION_SID)
request.assert_called_with("GET",
uri,
auth=AUTH,
use_json_extension=False)
def test_list(self, request):
resp = create_mock_json(
'tests/resources/task_router/reservations_list.json')
resp.status_code = 200
request.return_value = resp
uri = "{0}/Reservations".format(BASE_URI)
list_resource = Reservations(BASE_URI, AUTH)
list_resource.list()
request.assert_called_with("GET",
uri,
params={},
auth=AUTH,
use_json_extension=False)
def test_update_list(self, request):
resp = create_mock_json(
'tests/resources/task_router/reservations_instance.json')
resp.status_code = 201
request.return_value = resp
uri = "{0}/Reservations/{1}".format(BASE_URI, RESERVATION_SID)
list_resource = Reservations(BASE_URI, AUTH)
list_resource.update(
RESERVATION_SID,
worker_activity_sid='WAaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa',
reservation_status='rejected')
exp_params = {
'WorkerActivitySid': "WAaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa",
'ReservationStatus': "rejected"
}
request.assert_called_with("POST",
uri,
data=exp_params,
auth=AUTH,
use_json_extension=False)
