def test_get_subscriptions__subscriptions_exist_and_are_returned_as_list(
test_client, generate_subscription, test_user):
subscriptions = [
generate_subscription(user=test_user),
generate_subscription(user=test_user)
]
url = f'{BASE_PATH}/subscriptions/'
response = test_client.get(url, headers=basic_auth_header(test_user))
assert 200 == response.status_code
response_data = json.loads(response.data)
assert isinstance(response_data, list)
assert [t.id for t in subscriptions] == [d['id'] for d in response_data]
assert [t.queue
for t in subscriptions] == [d['queue'] for d in response_data]
assert [t.topic.name for t in subscriptions
] == [d['topic']['name'] for d in response_data]
assert [t.durable
for t in subscriptions] == [d['durable'] for d in response_data]
assert [t.active
for t in subscriptions] == [d['active'] for d in response_data]
assert [t.qos.value
for t in subscriptions] == [d['qos'] for d in response_data]
def test_delete_subscription__subscription_is_deleted_and_returns_204(
mock_broker_client, test_client, generate_subscription, test_user):
mock_broker_client.delete_queue_binding = mock.Mock(return_value=None)
subscription = generate_subscription(user=test_user,
with_broker_queue=True)
url = f'{BASE_PATH}/subscriptions/{subscription.id}'
response = test_client.delete(url, headers=basic_auth_header(test_user))
assert 204 == response.status_code
# check that the subscription has been deleted from db
response = test_client.get(url, headers=basic_auth_header(test_user))
assert 404 == response.status_code
def test_get_subscription__subscription_does_not_exist__returns_404(
test_client, test_user):
url = f'{BASE_PATH}/subscriptions/123456'
response = test_client.get(url, headers=basic_auth_header(test_user))
assert 404 == response.status_code
def test_put_subscription__subscription_is_updated_in_db_and_broker(
mock_broker_client, test_client, generate_subscription, test_user):
mock_broker_client.bind_queue_to_topic = mock.Mock(return_value=None)
mock_broker_client.delete_queue_binding = mock.Mock(return_value=None)
subscription = generate_subscription(user=test_user,
with_broker_queue=True)
subscription_data = {'active': not subscription.active}
url = f'{BASE_PATH}/subscriptions/{subscription.id}'
response = test_client.put(url,
data=json.dumps(subscription_data),
content_type='application/json',
headers=basic_auth_header(test_user))
assert 200 == response.status_code
response_data = json.loads(response.data)
assert subscription_data['active'] == response_data['active']
db_subscription = get_subscription_by_id(subscription.id)
assert subscription_data['active'] == db_subscription.active
# remove the queue from the broker
broker.delete_queue(subscription.queue)
def test_get_subscription__user_tries_to_get_subscription_of_another_user__returns_404(
test_client, generate_user, generate_subscription):
user1 = generate_user('username1', DEFAULT_LOGIN_PASS)
user2 = generate_user('username2', DEFAULT_LOGIN_PASS)
subscription = generate_subscription(user=user1)
url = f'{BASE_PATH}/subscriptions/{subscription.id}'
response = test_client.get(url, headers=basic_auth_header(user2))
assert 404 == response.status_code
def test_get_subscriptions__admin_user_can_get_all_existing_subscriptions(
test_client, test_admin_user, generate_user, generate_subscription):
generate_subscription()
generate_subscription()
url = f'{BASE_PATH}/subscriptions/'
response = test_client.get(url, headers=basic_auth_header(test_admin_user))
assert 200 == response.status_code
response_data = json.loads(response.data)
assert 2 == len(response_data)
def test_get_subscriptions__no_subscription_exists_for_user__empty_list_is_returned(
test_client, generate_user, generate_subscription):
user1 = generate_user('username1', DEFAULT_LOGIN_PASS)
user2 = generate_user('username2', DEFAULT_LOGIN_PASS)
generate_subscription(user=user1)
url = f'{BASE_PATH}/subscriptions/'
response = test_client.get(url, headers=basic_auth_header(user2))
assert 200 == response.status_code
response_data = json.loads(response.data)
assert [] == response_data
def test_get_subscription__admin_user_can_get_subscription_of_another_user(
test_client, test_user, test_admin_user, generate_subscription):
subscription = generate_subscription(user=test_user)
url = f'{BASE_PATH}/subscriptions/{subscription.id}'
response = test_client.get(url, headers=basic_auth_header(test_admin_user))
assert 200 == response.status_code
response_data = json.loads(response.data)
assert subscription.topic.name == response_data['topic']['name']
assert subscription.queue == response_data['queue']
assert subscription.active == response_data['active']
assert subscription.qos.value == response_data['qos']
assert subscription.durable == response_data['durable']
assert subscription.id == response_data['id']
def test_put_subscription__subscription_does_not_exist__returns_404(
test_client, test_user):
subscription_data = {'active': False}
url = f'{BASE_PATH}/subscriptions/1234'
response = test_client.put(url,
data=json.dumps(subscription_data),
content_type='application/json',
headers=basic_auth_header(test_user))
assert 404 == response.status_code
response_data = json.loads(response.data)
assert "Subscription with id 1234 does not exist" == response_data[
'detail']
def test_get_subscriptions__filter_by_queue(test_client, generate_subscription,
test_user):
subscriptions = [
generate_subscription(user=test_user),
generate_subscription(user=test_user)
]
url = f'{BASE_PATH}/subscriptions/?queue={subscriptions[0].queue}'
response = test_client.get(url, headers=basic_auth_header(test_user))
assert 200 == response.status_code
response_data = json.loads(response.data)
assert isinstance(response_data, list)
assert 1 == len(response_data)
assert subscriptions[0].queue == response_data[0]['queue']
def test_get_subscription__subscription_exists_and_its_data_is_returned(
test_client, generate_subscription, test_user):
subscription = generate_subscription(user=test_user)
url = f'{BASE_PATH}/subscriptions/{subscription.id}'
response = test_client.get(url, headers=basic_auth_header(test_user))
assert 200 == response.status_code
response_data = json.loads(response.data)
assert subscription.topic.name == response_data['topic']['name']
assert subscription.queue == response_data['queue']
assert subscription.active == response_data['active']
assert subscription.qos.value == response_data['qos']
assert subscription.durable == response_data['durable']
assert subscription.id == response_data['id']
def test_put_subscription__db_error__returns_500(mock_update_subscription,
mock_bind_queue_to_topic,
mock_delete_queue_binding,
test_client,
generate_subscription,
test_user):
subscription = generate_subscription(user=test_user)
subscription_data = {'active': not subscription.active}
url = f'{BASE_PATH}/subscriptions/{subscription.id}'
response = test_client.put(url,
data=json.dumps(subscription_data),
content_type='application/json',
headers=basic_auth_header(test_user))
assert 500 == response.status_code
def test_post_subscription__invalid_topic_id__returns_400(
test_client, test_user):
subscription_data = {
'topic_id': 1234,
}
url = f'{BASE_PATH}/subscriptions/'
response = test_client.post(url,
data=json.dumps(subscription_data),
content_type='application/json',
headers=basic_auth_header(test_user))
assert 400 == response.status_code
response_data = json.loads(response.data)
assert "{'topic_id': ['there is no topic with id 1234']}" == response_data[
'detail']
def test_post_subscription__missing_topic_id__returns_400(
test_client, test_user):
subscription_data = {
'active': True,
'qos': QOS.EXACTLY_ONCE.value,
'durable': True
}
url = f'{BASE_PATH}/subscriptions/'
response = test_client.post(url,
data=json.dumps(subscription_data),
content_type='application/json',
headers=basic_auth_header(test_user))
assert 400 == response.status_code
response_data = json.loads(response.data)
assert "'topic_id' is a required property" == response_data['detail']
def test_put_subscription__broker_error__returns_502(mock_bind_queue_to_topic,
mock_delete_queue_binding,
test_client,
generate_subscription,
test_user):
subscription = generate_subscription(user=test_user)
subscription_data = {'active': not subscription.active}
url = f'{BASE_PATH}/subscriptions/{subscription.id}'
response = test_client.put(url,
data=json.dumps(subscription_data),
content_type='application/json',
headers=basic_auth_header(test_user))
assert 502 == response.status_code
response_data = json.loads(response.data)
assert "Error while accessing broker: error" == response_data['detail']
def test_post_subscription__db_error__returns_500(mock_create_subscription,
test_client, generate_topic,
test_user):
topic = generate_topic('test_topic')
subscription_data = {
'topic_id': topic.id,
'active': True,
'qos': QOS.EXACTLY_ONCE.value,
'durable': True
}
url = f'{BASE_PATH}/subscriptions/'
response = test_client.post(url,
data=json.dumps(subscription_data),
content_type='application/json',
headers=basic_auth_header(test_user))
assert 500 == response.status_code
def test_post_subscription__subscription_is_saved_in_db(
mock_broker_client, test_client, generate_topic, test_user):
mock_broker_client.create_queue_for_topic = mock.Mock(return_value=None)
topic = generate_topic('test_topic')
subscription_data = {
'topic_id': topic.id,
'active': True,
'qos': QOS.EXACTLY_ONCE.value,
'durable': True
}
url = f'{BASE_PATH}/subscriptions/'
response = test_client.post(url,
data=json.dumps(subscription_data),
content_type='application/json',
headers=basic_auth_header(test_user))
assert 201 == response.status_code
response_data = json.loads(response.data)
assert 'id' in response_data
assert isinstance(response_data['id'], int)
assert 'queue' in response_data
assert isinstance(response_data['queue'], str)
assert subscription_data['durable'] == response_data['durable']
assert subscription_data['active'] == response_data['active']
assert subscription_data['qos'] == response_data['qos']
assert subscription_data['topic_id'] == response_data['topic']['id']
db_subscription = get_subscription_by_id(response_data['id'])
assert db_subscription is not None
assert subscription_data['qos'] == db_subscription.qos.value
assert subscription_data['active'] == db_subscription.active
assert subscription_data['durable'] == db_subscription.durable
assert subscription_data['topic_id'] == db_subscription.topic.id
broker.delete_queue(response_data['queue'])
def test_post_subscription__invalid_qos__returns_400(test_client,
generate_topic,
test_user):
topic = generate_topic('test_topic')
subscription_data = {
'topic_id': topic.id,
'active': True,
'qos': 'invalid',
'durable': True
}
url = f'{BASE_PATH}/subscriptions/'
response = test_client.post(url,
data=json.dumps(subscription_data),
content_type='application/json',
headers=basic_auth_header(test_user))
assert 400 == response.status_code
response_data = json.loads(response.data)
assert f"'invalid' is not one of {QOS.all()}" == response_data['detail']
def test_post_subscription__broker_error__returns_502(mock_queue_for_topic,
test_client,
generate_topic,
test_user):
topic = generate_topic('test_topic')
subscription_data = {
'topic_id': topic.id,
'active': True,
'qos': QOS.EXACTLY_ONCE.value,
'durable': True
}
url = f'{BASE_PATH}/subscriptions/'
response = test_client.post(url,
data=json.dumps(subscription_data),
content_type='application/json',
headers=basic_auth_header(test_user))
assert 502 == response.status_code
response_data = json.loads(response.data)
assert "Error while accessing the broker: error" == response_data['detail']