def mock_get_cik_map(monkeypatch):
response_json = {"0": {"cik_str": "320193", "ticker": "AAPL", "title": "Apple Inc."},
"1": {"cik_str": "789019", "ticker": "MSFT", "title": "MICROSOFT CORP"},
"2": {"cik_str": "1018724", "ticker": "AMZN", "title": "AMAZON COM INC"},
"3": {"cik_str": "1326801", "ticker": "FB", "title": "Facebook Inc"},
"4": {"cik_str": "1652044", "ticker": "GOOGL", "title": "Alphabet Inc."},
"5": {"cik_str": "1652044", "ticker": "GOOG", "title": "Alphabet Inc."}}
response_json = json.dumps(response_json)
monkeypatch.setattr(requests, 'get', MockResponse(content=bytes(response_json, "utf-8")))
def mock_single_filing_page_good_response(monkeypatch):
monkeypatch.setattr(
requests, 'get', lambda *args, **kwargs: MockResponse(
datapath_args=['CIK', 'single_filing_page.html']))
def mock_multiple_cik_results_good_response(monkeypatch):
monkeypatch.setattr(
requests, 'get', lambda *args, **kwargs: MockResponse(
datapath_args=['CIK', 'cik_multiple_results.html']))
def mock_single_filing_type_good_response(monkeypatch):
"""Mock response with list of single filing type for single CIK."""
monkeypatch.setattr(
requests, 'get', lambda *args, **kwargs: MockResponse(
datapath_args=['CIK', 'single_cik_multiple_filings_10k.html']))
def mock_no_cik_found_bad_response(monkeypatch):
monkeypatch.setattr(
requests, 'get', lambda *args, **kwargs: MockResponse(
datapath_args=['CIK', 'cik_not_found.html']))
def mock_single_cik_not_found(monkeypatch):
monkeypatch.setattr(
NetworkClient, "get_response",
MockResponse(datapath_args=["CIK", "cik_not_found.html"]))
def mock_single_cik_multiple_results_response(monkeypatch):
monkeypatch.setattr(
NetworkClient, "get_response",
MockResponse(datapath_args=["CIK", "cik_multiple_results.html"]))
def mock_single_cik_lookup_response(monkeypatch):
monkeypatch.setattr(
NetworkClient, "get_response",
MockResponse(datapath_args=["CIK", "single_cik_search_result.html"]))
def _mock_single_cik_not_found(*args, **kwargs):
return MockResponse(datapath_args=["CIK", "cik_not_found.html"],
file_read_args="rb")
def _mock_single_cik_multiple_results_response(*args, **kwargs):
return MockResponse(datapath_args=["CIK", "cik_multiple_results.html"],
file_read_args="rb")
def _mock_single_cik_lookup_response(*args, **kwargs):
return MockResponse(
datapath_args=["CIK", "single_cik_search_result.html"],
file_read_args="rb")