Robosats/robosats
1
0
Fork 0
mirror of https://github.com/RoboSats/robosats.git synced 2025-08-09 10:20:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix tests

Browse Source
This commit is contained in:
KoalaSat
2024-06-22 23:01:33 +02:00
parent eed8c11e63
commit cb927f2580
1 changed files with 45 additions and 32 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

77
tests/test_trade_pipeline.py
View File

@ -47,14 +47,6 @@ class TradeTest(BaseAPITestCase):
# Take the first node balances snapshot # Take the first node balances snapshot
compute_node_balance() compute_node_balance()
def get_notifications(self, headers):
"""
Get robot notifications
"""
response = self.client.get(reverse("notifications"), **headers)
self.assertResponse(response)
list(response.json())
def assert_order_logs(self, order_id): def assert_order_logs(self, order_id):
order = Order.objects.get(id=order_id) order = Order.objects.get(id=order_id)
order_admin = OrderAdmin(model=Order, admin_site=AdminSite()) order_admin = OrderAdmin(model=Order, admin_site=AdminSite())
@ -248,7 +240,9 @@ class TradeTest(BaseAPITestCase):
self.assert_order_logs(data["id"]) self.assert_order_logs(data["id"])
maker_headers = trade.get_robot_auth(trade.maker_index) maker_headers = trade.get_robot_auth(trade.maker_index)
notifications_data = self.get_notifications(maker_headers) response = self.client.get(reverse("notifications"), **maker_headers)
self.assertResponse(response)
notifications_data = list(response.json())
self.assertEqual( self.assertEqual(
len(notifications_data), len(notifications_data),
0, 0,
@ -364,7 +358,9 @@ class TradeTest(BaseAPITestCase):
self.assert_order_logs(data["id"]) self.assert_order_logs(data["id"])
maker_headers = trade.get_robot_auth(trade.maker_index) maker_headers = trade.get_robot_auth(trade.maker_index)
notifications_data = self.get_notifications(maker_headers) response = self.client.get(reverse("notifications"), **maker_headers)
self.assertResponse(response)
notifications_data = list(response.json())
self.assertEqual( self.assertEqual(
len(notifications_data), len(notifications_data),
1, 1,
@ -395,7 +391,9 @@ class TradeTest(BaseAPITestCase):
self.assertEqual(data["status_message"], Order.Status(Order.Status.PUB).label) self.assertEqual(data["status_message"], Order.Status(Order.Status.PUB).label)
maker_headers = trade.get_robot_auth(trade.maker_index) maker_headers = trade.get_robot_auth(trade.maker_index)
notifications_data = self.get_notifications(maker_headers) response = self.client.get(reverse("notifications"), **maker_headers)
self.assertResponse(response)
notifications_data = list(response.json())
self.assertEqual(notifications_data[0]["order_id"], trade.order_id) self.assertEqual(notifications_data[0]["order_id"], trade.order_id)
# Cancel order to avoid leaving pending HTLCs after a successful test # Cancel order to avoid leaving pending HTLCs after a successful test
@ -445,7 +443,9 @@ class TradeTest(BaseAPITestCase):
self.assert_order_logs(data["id"]) self.assert_order_logs(data["id"])
maker_headers = trade.get_robot_auth(trade.maker_index) maker_headers = trade.get_robot_auth(trade.maker_index)
notifications_data = self.get_notifications(maker_headers) response = self.client.get(reverse("notifications"), **maker_headers)
self.assertResponse(response)
notifications_data = list(response.json())
self.assertEqual(notifications_data[0]["order_id"], trade.order_id) self.assertEqual(notifications_data[0]["order_id"], trade.order_id)
def test_make_and_lock_contract(self): def test_make_and_lock_contract(self):
@ -471,7 +471,9 @@ class TradeTest(BaseAPITestCase):
self.assertFalse(data["escrow_locked"]) self.assertFalse(data["escrow_locked"])
maker_headers = trade.get_robot_auth(trade.maker_index) maker_headers = trade.get_robot_auth(trade.maker_index)
notifications_data = self.get_notifications(maker_headers) response = self.client.get(reverse("notifications"), **maker_headers)
self.assertResponse(response)
notifications_data = list(response.json())
notifications_data = list(trade.response.json()) notifications_data = list(trade.response.json())
self.assertEqual( self.assertEqual(
len(notifications_data), len(notifications_data),
@ -501,7 +503,9 @@ class TradeTest(BaseAPITestCase):
self.assertFalse(data["escrow_locked"]) self.assertFalse(data["escrow_locked"])
maker_headers = trade.get_robot_auth(trade.maker_index) maker_headers = trade.get_robot_auth(trade.maker_index)
notifications_data = self.get_notifications(maker_headers) response = self.client.get(reverse("notifications"), **maker_headers)
self.assertResponse(response)
notifications_data = list(response.json())
self.assertEqual( self.assertEqual(
len(notifications_data), len(notifications_data),
2, 2,
@ -536,7 +540,9 @@ class TradeTest(BaseAPITestCase):
self.assertTrue(data["escrow_locked"]) self.assertTrue(data["escrow_locked"])
maker_headers = trade.get_robot_auth(trade.maker_index) maker_headers = trade.get_robot_auth(trade.maker_index)
notifications_data = self.get_notifications(maker_headers) response = self.client.get(reverse("notifications"), **maker_headers)
self.assertResponse(response)
notifications_data = list(response.json())
self.assertEqual( self.assertEqual(
len(notifications_data), len(notifications_data),
3, 3,
@ -568,7 +574,9 @@ class TradeTest(BaseAPITestCase):
self.assertFalse(data["is_fiat_sent"]) self.assertFalse(data["is_fiat_sent"])
maker_headers = trade.get_robot_auth(trade.maker_index) maker_headers = trade.get_robot_auth(trade.maker_index)
notifications_data = self.get_notifications(maker_headers) response = self.client.get(reverse("notifications"), **maker_headers)
self.assertResponse(response)
notifications_data = list(response.json())
self.assertEqual( self.assertEqual(
len(notifications_data), len(notifications_data),
4, 4,
@ -603,7 +611,9 @@ class TradeTest(BaseAPITestCase):
self.assertFalse(data["is_fiat_sent"]) self.assertFalse(data["is_fiat_sent"])
maker_headers = trade.get_robot_auth(trade.maker_index) maker_headers = trade.get_robot_auth(trade.maker_index)
notifications_data = self.get_notifications(maker_headers) response = self.client.get(reverse("notifications"), **maker_headers)
self.assertResponse(response)
notifications_data = list(response.json())
self.assertEqual( self.assertEqual(
len(notifications_data), len(notifications_data),
4, 4,
@ -636,7 +646,9 @@ class TradeTest(BaseAPITestCase):
self.assertTrue(data["is_fiat_sent"]) self.assertTrue(data["is_fiat_sent"])
maker_headers = trade.get_robot_auth(trade.maker_index) maker_headers = trade.get_robot_auth(trade.maker_index)
notifications_data = self.get_notifications(maker_headers) response = self.client.get(reverse("notifications"), **maker_headers)
self.assertResponse(response)
notifications_data = list(response.json())
self.assertEqual( self.assertEqual(
len(notifications_data), len(notifications_data),
6, 6,
@ -684,7 +696,9 @@ class TradeTest(BaseAPITestCase):
self.assert_order_logs(data["id"]) self.assert_order_logs(data["id"])
maker_headers = trade.get_robot_auth(trade.maker_index) maker_headers = trade.get_robot_auth(trade.maker_index)
notifications_data = self.get_notifications(maker_headers) response = self.client.get(reverse("notifications"), **maker_headers)
self.assertResponse(response)
notifications_data = list(response.json())
self.assertEqual( self.assertEqual(
len(notifications_data), len(notifications_data),
7, 7,
@ -800,7 +814,9 @@ class TradeTest(BaseAPITestCase):
) )
maker_headers = trade.get_robot_auth(trade.maker_index) maker_headers = trade.get_robot_auth(trade.maker_index)
notifications_data = self.get_notifications(maker_headers) response = self.client.get(reverse("notifications"), **maker_headers)
self.assertResponse(response)
notifications_data = list(response.json())
self.assertEqual( self.assertEqual(
len(notifications_data), len(notifications_data),
6, 6,
@ -840,15 +856,6 @@ class TradeTest(BaseAPITestCase):
self.assert_order_logs(data["id"]) self.assert_order_logs(data["id"])
maker_headers = trade.get_robot_auth(trade.maker_index)
notifications_data = self.get_notifications(maker_headers)
self.assertEqual(
len(notifications_data),
6,
"User has a new order expired notification",
)
self.assertEqual(notifications_data[0]["order_id"], trade.order_id)
def test_public_order_expires(self): def test_public_order_expires(self):
""" """
Tests the expiration of a public order Tests the expiration of a public order
@ -883,7 +890,9 @@ class TradeTest(BaseAPITestCase):
self.assert_order_logs(data["id"]) self.assert_order_logs(data["id"])
maker_headers = trade.get_robot_auth(trade.maker_index) maker_headers = trade.get_robot_auth(trade.maker_index)
notifications_data = self.get_notifications(maker_headers) response = self.client.get(reverse("notifications"), **maker_headers)
self.assertResponse(response)
notifications_data = list(response.json())
self.assertEqual( self.assertEqual(
len(notifications_data), len(notifications_data),
6, 6,
@ -927,7 +936,9 @@ class TradeTest(BaseAPITestCase):
self.assert_order_logs(data["id"]) self.assert_order_logs(data["id"])
maker_headers = trade.get_robot_auth(trade.maker_index) maker_headers = trade.get_robot_auth(trade.maker_index)
notifications_data = self.get_notifications(maker_headers) response = self.client.get(reverse("notifications"), **maker_headers)
self.assertResponse(response)
notifications_data = list(response.json())
self.assertEqual( self.assertEqual(
len(notifications_data), len(notifications_data),
6, 6,
@ -1024,7 +1035,9 @@ class TradeTest(BaseAPITestCase):
self.assertEqual(response.json(), {}) # Nothing in the response self.assertEqual(response.json(), {}) # Nothing in the response
maker_headers = trade.get_robot_auth(trade.maker_index) maker_headers = trade.get_robot_auth(trade.maker_index)
notifications_data = self.get_notifications(maker_headers) response = self.client.get(reverse("notifications"), **maker_headers)
self.assertResponse(response)
notifications_data = list(response.json())
self.assertEqual( self.assertEqual(
len(notifications_data), len(notifications_data),
8, 8,