Trades tests (#136)

* trades tests

* trial error

* trial error

* style

Author: mcdayoub

tests/mocks.py

@@ -39,6 +39,18 @@
         "/v3/reference/tickers/types",
         '{"results":[{"code":"CS","description":"Common Stock","asset_class":"stocks","locale":"us"},{"code":"PFD","description":"Preferred Stock","asset_class":"stocks","locale":"us"},{"code":"WARRANT","description":"Warrant","asset_class":"stocks","locale":"us"},{"code":"RIGHT","description":"Rights","asset_class":"stocks","locale":"us"},{"code":"BOND","description":"Corporate Bond","asset_class":"stocks","locale":"us"},{"code":"ETF","description":"Exchange Traded Fund","asset_class":"stocks","locale":"us"},{"code":"ETN","description":"Exchange Traded Note","asset_class":"stocks","locale":"us"},{"code":"SP","description":"Structured Product","asset_class":"stocks","locale":"us"},{"code":"ADRC","description":"American Depository Receipt Common","asset_class":"stocks","locale":"us"},{"code":"ADRW","description":"American Depository Receipt Warrants","asset_class":"stocks","locale":"us"},{"code":"ADRR","description":"American Depository Receipt Rights","asset_class":"stocks","locale":"us"},{"code":"FUND","description":"Fund","asset_class":"stocks","locale":"us"},{"code":"BASKET","description":"Basket","asset_class":"stocks","locale":"us"},{"code":"UNIT","description":"Unit","asset_class":"stocks","locale":"us"},{"code":"LT","description":"Liquidating Trust","asset_class":"stocks","locale":"us"}],"status":"OK","request_id":"efbfc7c2304bba6c2f19a2567f568134","count":15}',
     ),
+    (
+        "/v2/last/trade/AAPL",
+        '{"results":{"c":[12,37],"i":"237688","p":166.25,"s":2,"x":4,"r":202,"z":3,"T":"AAPL","t":1651179319310617300,"y":1651179319308000000,"f":1651179319310588400,"q":7084210},"status":"OK","request_id":"d4bafa50e72cf9ed19ac538ae1a3185a"}',
+    ),
+    (
+        "/v1/last/crypto/BTC/USD",
+        '{"last":{"conditions":[2],"exchange":2,"price":39976.89682331,"size":0.005,"timestamp":1651180409688},"request_id":"d67c9bfe1fa0c29db9177d78b3ab713c","status":"success","symbol":"BTC-USD"}',
+    ),
+    (
+        "/v3/trades/AAPL",
+        '{"results":[{"conditions":[12,37],"correction":1,"exchange":11,"id":"183276","participant_timestamp":1651181822461636600,"price":156.43,"sequence_number":7179341,"sip_timestamp":1651181822461979400,"size":10,"tape":3,"trf_id":3,"trf_timestamp":1651181557090806500},{"conditions":[12,37],"correction":1,"exchange":12,"id":"183276","participant_timestamp":1651181822461636600,"price":157.43,"sequence_number":7179341,"sip_timestamp":1651181822461979400,"size":10,"tape":3,"trf_id":3,"trf_timestamp":1651181557090806500}],"status":"OK","request_id":"756f9910624b35a47eb07f21a7a373bb"}',
+    ),
     (
         "/v1/marketstatus/upcoming",
         '[{"exchange":"NYSE","name":"Memorial Day","date":"2022-05-30","status":"closed"},{"exchange":"NASDAQ","name":"Memorial Day","date":"2022-05-30","status":"closed"},{"exchange":"NASDAQ","name":"Juneteenth","date":"2022-06-20","status":"closed"},{"exchange":"NYSE","name":"Juneteenth","date":"2022-06-20","status":"closed"},{"exchange":"NYSE","name":"Independence Day","date":"2022-07-04","status":"closed"},{"exchange":"NASDAQ","name":"Independence Day","date":"2022-07-04","status":"closed"},{"exchange":"NYSE","name":"Labor Day","date":"2022-09-05","status":"closed"},{"exchange":"NASDAQ","name":"Labor Day","date":"2022-09-05","status":"closed"},{"exchange":"NYSE","name":"Thanksgiving","date":"2022-11-24","status":"closed"},{"exchange":"NASDAQ","name":"Thanksgiving","date":"2022-11-24","status":"closed"},{"exchange":"NYSE","name":"Thanksgiving","date":"2022-11-25","status":"early-close","open":"2022-11-25T14:30:00.000Z","close":"2022-11-25T18:00:00.000Z"},{"exchange":"NASDAQ","name":"Thanksgiving","date":"2022-11-25","status":"early-close","open":"2022-11-25T14:30:00.000Z","close":"2022-11-25T18:00:00.000Z"},{"exchange":"NYSE","name":"Christmas","date":"2022-12-26","status":"closed"},{"exchange":"NASDAQ","name":"Christmas","date":"2022-12-26","status":"closed"}]',

tests/test_trades.py

@@ -0,0 +1,83 @@
+from mocks import BaseTest
+from polygon.rest.models import (
+    Trade,
+    LastTrade,
+    Last,
+    LastTradeCrypto,
+)
+
+
+class TradesTest(BaseTest):
+    def test_get_last_trade(self):
+        last_trade = self.c.get_last_trade("AAPL")
+        expected = [
+            LastTrade(
+                ticker="AAPL",
+                trf_timestamp=1651179319310588400,
+                sequence_number=7084210,
+                sip_timestamp=1651179319310617300,
+                participant_timestamp=1651179319308000000,
+                conditions=[12, 37],
+                correction=None,
+                id="237688",
+                price=166.25,
+                trf_id=202,
+                size=2,
+                exchange=4,
+                tape=3,
+            )
+        ]
+        self.assertEqual(last_trade, expected)
+
+    def test_get_last_trade_crypto(self):
+        last_trade_crypto = self.c.get_last_trade_crypto("BTC", "USD")
+        expected = [
+            LastTradeCrypto(
+                last={
+                    "conditions": [2],
+                    "exchange": 2,
+                    "price": 39976.89682331,
+                    "size": 0.005,
+                    "timestamp": 1651180409688,
+                },
+                ticker="BTC-USD",
+                status="success",
+                request_id="d67c9bfe1fa0c29db9177d78b3ab713c",
+            )
+        ]
+        self.assertEqual(last_trade_crypto, expected)
+
+    def test_trades(self):
+        trades = [t for t in self.c.list_trades(ticker="AAPL", limit=2)]
+        print(trades)
+        expected = [
+            Trade(
+                conditions=[12, 37],
+                correction=1,
+                exchange=11,
+                id="183276",
+                participant_timestamp=1651181822461636600,
+                price=156.43,
+                sequence_number=7179341,
+                sip_timestamp=1651181822461979400,
+                size=10,
+                tape=3,
+                trf_id=3,
+                trf_timestamp=1651181557090806500,
+            ),
+            Trade(
+                conditions=[12, 37],
+                correction=1,
+                exchange=12,
+                id="183276",
+                participant_timestamp=1651181822461636600,
+                price=157.43,
+                sequence_number=7179341,
+                sip_timestamp=1651181822461979400,
+                size=10,
+                tape=3,
+                trf_id=3,
+                trf_timestamp=1651181557090806500,
+            ),
+        ]
+        self.assertEqual(trades, expected)