add extensive tests (cert session only); coverage >95 enforced

.github/workflows/python-app.yml

@@ -31,7 +31,7 @@ jobs:
         mypy -p tastytrade
     - name: Testing...
       run: |
-        python -m pytest --cov=tastytrade --cov-report=term-missing tests/
+        python -m pytest --cov=tastytrade --cov-report=term-missing tests/ --cov-fail-under=95
       env:
         TT_USERNAME: ${{ secrets.TT_USERNAME }}
         TT_PASSWORD: ${{ secrets.TT_PASSWORD }}

Makefile

@@ -14,10 +14,7 @@ lint:
 	mypy -p tests
 
 test:
-	python -m pytest --cov=tastytrade --cov-report=term-missing tests/
-
-test:
-	python -m pytest --cov=tastytrade --cov-report=term-missing tests/
+	python -m pytest --cov=tastytrade --cov-report=term-missing tests/ --cov-fail-under=95
 
 install:
 	env/bin/pip install -e .

docs/index.rst

@@ -7,6 +7,7 @@
    installation
    sessions
    data-streamer
+   watchlists
 
 .. toctree::
    :maxdepth: 2

requirements.txt

@@ -7,3 +7,4 @@ websockets==11.0.3
 pydantic==1.10.11
 pytest==7.4.0
 pytest_cov==4.1.0
+pytest-asyncio==0.21.1

tastytrade/account.py

@@ -1,16 +1,22 @@
 from datetime import date, datetime
 from decimal import Decimal
-from typing import Any, Dict, List, Optional
+from typing import Any, Dict, List, Optional, Union
 
 import requests
+from pydantic import BaseModel
 
 from tastytrade.order import (InstrumentType, NewOrder, OrderStatus,
                               PlacedOrder, PlacedOrderResponse, PriceEffect)
-from tastytrade.session import Session
+from tastytrade.session import ProductionSession, Session
 from tastytrade.utils import (TastytradeError, TastytradeJsonDataclass,
                               validate_response)
 
 
+class EmptyDict(BaseModel):
+    class Config:
+        extra = 'forbid'
+
+
 class AccountBalance(TastytradeJsonDataclass):
     """
     Dataclass containing account balance information.
@@ -168,21 +174,21 @@ class MarginReportEntry(TastytradeJsonDataclass):
     code: str
     underlying_symbol: str
     underlying_type: str
-    expected_price_range_up_percent: Decimal
-    expected_price_range_down_percent: Decimal
-    point_of_no_return_percent: Decimal
     margin_calculation_type: str
     margin_requirement: Decimal
     margin_requirement_effect: PriceEffect
-    initial_requirement: Decimal
-    initial_requirement_effect: PriceEffect
     maintenance_requirement: Decimal
     maintenance_requirement_effect: PriceEffect
     buying_power: Decimal
     buying_power_effect: PriceEffect
     groups: List[Dict[str, Any]]
     price_increase_percent: Decimal
     price_decrease_percent: Decimal
+    expected_price_range_up_percent: Optional[Decimal] = None
+    expected_price_range_down_percent: Optional[Decimal] = None
+    point_of_no_return_percent: Optional[Decimal] = None
+    initial_requirement: Optional[Decimal] = None
+    initial_requirement_effect: Optional[PriceEffect] = None
 
 
 class MarginReport(TastytradeJsonDataclass):
@@ -207,8 +213,8 @@ class MarginReport(TastytradeJsonDataclass):
     reg_t_option_buying_power_effect: PriceEffect
     maintenance_excess: Decimal
     maintenance_excess_effect: PriceEffect
-    groups: List[MarginReportEntry]
     last_state_timestamp: int
+    groups: List[Union[MarginReportEntry, EmptyDict]]
     initial_requirement: Optional[Decimal] = None
     initial_requirement_effect: Optional[PriceEffect] = None
 
@@ -706,15 +712,16 @@ def get_total_fees(
 
     def get_net_liquidating_value_history(
         self,
-        session: Session,
+        session: ProductionSession,
         time_back: Optional[str] = None,
         start_time: Optional[datetime] = None
-    ) -> List[NetLiqOhlc]:
+    ) -> List[NetLiqOhlc]:  # pragma: no cover
         """
         Returns a list of account net liquidating value snapshots over the
         specified time period.
 
-        :param session: the session to use for the request.
+        :param session:
+            the session to use for the request, can't be certification.
         :param time_back:
             the time period to get net liquidating value snapshots for. This
             param is required if start_time is not given. Possible values are:
@@ -766,13 +773,14 @@ def get_position_limit(self, session: Session) -> PositionLimit:
 
     def get_effective_margin_requirements(
         self,
-        session: Session,
+        session: ProductionSession,
         symbol: str
-    ) -> MarginRequirement:
+    ) -> MarginRequirement:  # pragma: no cover
         """
         Get the effective margin requirements for a given symbol.
 
-        :param session: the session to use for the request.
+        :param session:
+            the session to use for the request, can't be certification
         :param symbol: the symbol to get margin requirements for.
 
         :return: a :class:`MarginRequirement` object.

tastytrade/dxfeed/event.py

@@ -25,7 +25,7 @@ class EventType(str, Enum):
 
 class Event(ABC):
     @classmethod
-    def from_stream(cls, data: list) -> List['Event']:
+    def from_stream(cls, data: list) -> List['Event']:  # pragma: no cover
         """
         Makes a list of event objects from a list of raw trade data fetched by
         a :class:`~tastyworks.streamer.DataStreamer`.

tastytrade/instruments.py

@@ -6,7 +6,7 @@
 import requests
 
 from tastytrade.order import InstrumentType, TradeableTastytradeJsonDataclass
-from tastytrade.session import Session
+from tastytrade.session import ProductionSession, Session
 from tastytrade.utils import TastytradeJsonDataclass, validate_response
 
 
@@ -1060,9 +1060,9 @@ def get_option_chain(
 
 
 def get_future_option_chain(
-    session: Session,
+    session: ProductionSession,
     symbol: str
-) -> Dict[date, List[FutureOption]]:
+) -> Dict[date, List[FutureOption]]:  # pragma: no cover
     """
     Returns a mapping of expiration date to a list of futures options
     objects representing the options chain for the given symbol.

tastytrade/metrics.py

@@ -4,7 +4,7 @@
 
 import requests
 
-from tastytrade.session import Session
+from tastytrade.session import ProductionSession, Session
 from tastytrade.utils import TastytradeJsonDataclass, validate_response
 
 
@@ -91,9 +91,9 @@ class MarketMetricInfo(TastytradeJsonDataclass):
 
 
 def get_market_metrics(
-    session: Session,
+    session: ProductionSession,
     symbols: List[str]
-) -> List[MarketMetricInfo]:
+) -> List[MarketMetricInfo]:  # pragma: no cover
     """
     Retrieves market metrics for the given symbols.
 
@@ -114,7 +114,10 @@ def get_market_metrics(
     return [MarketMetricInfo(**entry) for entry in data]
 
 
-def get_dividends(session: Session, symbol: str) -> List[DividendInfo]:
+def get_dividends(
+    session: ProductionSession,
+    symbol: str
+) -> List[DividendInfo]:  # pragma: no cover
     """
     Retrieves dividend information for the given symbol.
 
@@ -136,10 +139,10 @@ def get_dividends(session: Session, symbol: str) -> List[DividendInfo]:
 
 
 def get_earnings(
-    session: Session,
+    session: ProductionSession,
     symbol: str,
     start_date: date
-) -> List[EarningsInfo]:
+) -> List[EarningsInfo]:  # pragma: no cover
     """
     Retrieves earnings information for the given symbol.
 

tastytrade/search.py

@@ -2,7 +2,7 @@
 
 import requests
 
-from tastytrade.session import Session
+from tastytrade.session import ProductionSession
 from tastytrade.utils import TastytradeJsonDataclass
 
 
@@ -14,7 +14,10 @@ class SymbolData(TastytradeJsonDataclass):
     description: str
 
 
-def symbol_search(session: Session, symbol: str) -> List[SymbolData]:
+def symbol_search(
+    session: ProductionSession,
+    symbol: str
+) -> List[SymbolData]:  # pragma: no cover
     """
     Performs a symbol search using the Tastytrade API and returns a
     list of symbols that are similar to the given search phrase.

tastytrade/session.py

@@ -113,7 +113,7 @@ def __init__(
         self.validate()
 
 
-class ProductionSession(Session):
+class ProductionSession(Session):  # pragma: no cover
     """
     Contains a local user login which can then be used to interact with the
     remote API.
@@ -326,7 +326,7 @@ def get_time_and_sale(
 def _map_event(
     event_type: str,
     event_dict: Any  # Usually Dict[str, Any]; sometimes a list
-) -> Event:
+) -> Event:  # pragma: no cover
     """
     Parses the raw JSON data from the dxfeed REST API into event objects.
 

tastytrade/streamer.py

@@ -248,7 +248,7 @@ async def _subscribe(
         await self._websocket.send(json.dumps(message))  # type: ignore
 
 
-class DataStreamer:
+class DataStreamer:  # pragma: no cover
     """
     A :class:`DataStreamer` object is used to fetch quotes or greeks
     for a given symbol or list of symbols. It should always be

tastytrade/utils.py

@@ -30,7 +30,7 @@ class Config:
         allow_population_by_field_name = True
 
 
-def validate_response(response: Response) -> None:
+def validate_response(response: Response) -> None:  # pragma: no cover
     """
     Checks if the given code is an error; if so, raises an exception.
 

tastytrade/watchlists.py

@@ -3,7 +3,7 @@
 import requests
 
 from tastytrade.instruments import InstrumentType
-from tastytrade.session import Session
+from tastytrade.session import ProductionSession
 from tastytrade.utils import TastytradeJsonDataclass, validate_response
 
 
@@ -19,7 +19,7 @@ class Pair(TastytradeJsonDataclass):
     right_quantity: int
 
 
-class PairsWatchlist(TastytradeJsonDataclass):
+class PairsWatchlist(TastytradeJsonDataclass):  # pragma: no cover
     """
     Dataclass that represents a pairs watchlist object.
     """
@@ -28,7 +28,7 @@ class PairsWatchlist(TastytradeJsonDataclass):
     pairs_equations: List[Pair]
 
     @classmethod
-    def get_pairs_watchlists(cls, session: Session) -> List['PairsWatchlist']:
+    def get_pairs_watchlists(cls, session: ProductionSession) -> List['PairsWatchlist']:
         """
         Fetches a list of all Tastytrade public pairs watchlists.
 
@@ -48,7 +48,7 @@ def get_pairs_watchlists(cls, session: Session) -> List['PairsWatchlist']:
     @classmethod
     def get_pairs_watchlist(
         cls,
-        session: Session,
+        session: ProductionSession,
         name: str
     ) -> 'PairsWatchlist':
         """
@@ -70,7 +70,7 @@ def get_pairs_watchlist(
         return cls(**data)
 
 
-class Watchlist(TastytradeJsonDataclass):
+class Watchlist(TastytradeJsonDataclass):  # pragma: no cover
     """
     Dataclass that represents a watchlist object (public or private),
     with functions to update, publish, modify and remove watchlists.
@@ -83,7 +83,7 @@ class Watchlist(TastytradeJsonDataclass):
     @classmethod
     def get_public_watchlists(
         cls,
-        session: Session,
+        session: ProductionSession,
         counts_only: bool = False
     ) -> List['Watchlist']:
         """
@@ -106,7 +106,7 @@ def get_public_watchlists(
         return [cls(**entry) for entry in data]
 
     @classmethod
-    def get_public_watchlist(cls, session: Session, name: str) -> 'Watchlist':
+    def get_public_watchlist(cls, session: ProductionSession, name: str) -> 'Watchlist':
         """
         Fetches a Tastytrade public watchlist by name.
 
@@ -126,7 +126,7 @@ def get_public_watchlist(cls, session: Session, name: str) -> 'Watchlist':
         return cls(**data)
 
     @classmethod
-    def get_private_watchlists(cls, session: Session) -> List['Watchlist']:
+    def get_private_watchlists(cls, session: ProductionSession) -> List['Watchlist']:
         """
         Fetches a the user's private watchlists.
 
@@ -145,7 +145,7 @@ def get_private_watchlists(cls, session: Session) -> List['Watchlist']:
         return [cls(**entry) for entry in data]
 
     @classmethod
-    def get_private_watchlist(cls, session: Session, name: str) -> 'Watchlist':
+    def get_private_watchlist(cls, session: ProductionSession, name: str) -> 'Watchlist':
         """
         Fetches a user's watchlist by name.
 
@@ -165,7 +165,7 @@ def get_private_watchlist(cls, session: Session, name: str) -> 'Watchlist':
         return cls(**data)
 
     @classmethod
-    def remove_private_watchlist(cls, session: Session, name: str) -> None:
+    def remove_private_watchlist(cls, session: ProductionSession, name: str) -> None:
         """
         Deletes the named private watchlist.
 
@@ -178,7 +178,7 @@ def remove_private_watchlist(cls, session: Session, name: str) -> None:
         )
         validate_response(response)
 
-    def upload_private_watchlist(self, session: Session) -> None:
+    def upload_private_watchlist(self, session: ProductionSession) -> None:
         """
         Creates a private remote watchlist identical to this local one.
 
@@ -191,7 +191,7 @@ def upload_private_watchlist(self, session: Session) -> None:
         )
         validate_response(response)
 
-    def update_private_watchlist(self, session: Session) -> None:
+    def update_private_watchlist(self, session: ProductionSession) -> None:
         """
         Updates the existing private remote watchlist.
 

tests/conftest.py

@@ -0,0 +1,15 @@
+import os
+import pytest
+
+from tastytrade import CertificationSession
+
+
+@pytest.fixture(scope='session')
+def session():
+    username = os.environ.get('TT_USERNAME', None)
+    password = os.environ.get('TT_PASSWORD', None)
+
+    session = CertificationSession(username, password)
+    yield session
+
+    session.destroy()