test_reader.py

from open_interest import LocaleDAO, OnlineReader, next_expiry_date
import expiry
import underlying

import datetime as datetime

invalid_product = {"name": "something", "productGroupId": "7", "productId": "b"}
invalid_expiry = {"month": "d", "year": "a", "date": "20230317"}


def test_read_all_byexpiry_date():
    parameter = {"product": invalid_product, "type": "muhh", "expiry_date": invalid_expiry, "bus_date": "20230217"}
    
    local_dao = LocaleDAO()
    assert local_dao != None

    results = local_dao.read_all_by_expiry_date(parameter)
    assert len(list(results))  == 0

    parameter = {}
    results = local_dao.read_all_by_expiry_date(parameter)
    assert len(list(results))  == 0

def test_online_reader():
    online_reader = OnlineReader()
    assert online_reader != None

    parameter = {"product": invalid_product, "type": "muhh", "expiry_date": invalid_expiry, "bus_date": "20230217"}
    online_reader.request_data(parameter) == None

def test_next_expiry_date():
    result = next_expiry_date()
    assert result == {"month": 4, "year": 2023, "date": "20230421"}