def test_get_subscription_not_found(self):
subscription = Subscription(
customer_id=1,
callback_url="http://0.0.0.0:80/post",
)
db.session.add(subscription)
db.session.commit()
with self.client:
response = self.client.get('/subscription/2/', )
self.assertEqual(response.status_code, 404)
def test_get_subscription(self):
subscription = Subscription(
customer_id=1,
callback_url="http://0.0.0.0:80/post",
)
db.session.add(subscription)
db.session.commit()
with self.client:
response = self.client.get('/subscription/1/', )
data = json.loads(response.data.decode())
self.assertTrue(data['callback_url'] == "http://0.0.0.0:80/post")
self.assertTrue(response.content_type == 'application/json')
self.assertTrue(data['customer_id'] == 1)
self.assertEqual(response.status_code, 200)
def test_patch_subscription_status(self):
"""Test for subscription update"""
subscription = Subscription(
customer_id=1,
callback_url="http://0.0.0.0:80/post",
)
db.session.add(subscription)
db.session.commit()
with self.client:
response = self.client.patch('/subscription/1',
data=json.dumps(dict(active=True, )),
content_type='application/json')
data = json.loads(response.data.decode())
self.assertTrue(data['active'] == True)
self.assertTrue(response.content_type == 'application/json')
self.assertEqual(response.status_code, 200)
def test_create_notification_no_title(self):
"""Test for notification creation"""
with self.client:
subscription = Subscription(
customer_id=1,
callback_url="http://0.0.0.0:80/post",
)
db.session.add(subscription)
db.session.commit()
response = self.client.post(
'/notification/',
data=json.dumps(
dict(message="Sample Message",
subscription_id=subscription.id)),
content_type='application/json')
data = json.loads(response.data.decode())
self.assertTrue(
data['title'] == ['Missing data for required field.'])
self.assertTrue(response.content_type == 'application/json')
self.assertEqual(response.status_code, 400)
def test_get_notifications(self):
subscription = Subscription(
customer_id=1,
callback_url="http://0.0.0.0:80/post",
)
db.session.add(subscription)
db.session.commit()
notification = Notification(message="Sample Message",
subscription_id=subscription.id,
title="Sample Title")
db.session.add(notification)
db.session.commit()
with self.client:
response = self.client.get(
'/notification/{}'.format(str(notification.id)), )
data = json.loads(response.data.decode())
self.assertTrue(data['message'] == "Sample Message")
self.assertTrue(data['subscription_id'] == subscription.id)
self.assertTrue(data['title'] == "Sample Title")
self.assertTrue(response.content_type == 'application/json')
self.assertEqual(response.status_code, 200)
def test_patch_notification_status(self):
"""Test for notification update"""
subscription = Subscription(
customer_id=1,
callback_url="http://0.0.0.0:80/post",
)
db.session.add(subscription)
db.session.commit()
notification = Notification(message="Sample Message",
subscription_id=subscription.id,
title="Sample Title")
db.session.add(notification)
db.session.commit()
with self.client:
response = self.client.patch(
'/notification/{}'.format(notification.id),
data=json.dumps(dict(delivery_status='cancelled', )),
content_type='application/json')
data = json.loads(response.data.decode())
self.assertTrue(data['delivery_status'] == 'cancelled')
self.assertTrue(response.content_type == 'application/json')
self.assertEqual(response.status_code, 200)
def save_object(self, data, **kwargs):
return Subscription(**data)