def mock_client_create_companies(requests_mock) -> ApiClient:
requests_mock.register_uri(
"POST",
"/company/v2/companies",
json=CREATE_COMPANIES_RESPONSE,
additional_matcher=match_payload,
)
return make_sandbox(requests_mock)
def test_client(requests_mock) -> ApiClient:
requests_mock.register_uri(
"GET", "/company/v2/companies/CompanyID9890", json=GET_COMPANY
)
requests_mock.register_uri(
"GET", "/company/v2/companies", json=GET_COMPANIES_1_PAGE_1
)
requests_mock.register_uri("POST", "/company/v2/companies", json=RESPONSES)
for x in COMPANIES:
requests_mock.register_uri(
"GET",
"/company/v2/companies/" + x["id"],
json=x,
)
return make_sandbox(requests_mock)
def categories_client(requests_mock) -> ApiClient:
request_headers = {
PARTNER_ID_HEADER: "sandbox-partner",
}
requests_mock.register_uri(
"GET",
"/banking/v2/categories",
json=GET_CATEGORIES_PAGED,
request_headers=request_headers,
)
requests_mock.register_uri(
"GET",
"/banking/v2/categories?pageId=2",
json=GET_CATEGORIES,
request_headers=request_headers,
)
return make_sandbox(requests_mock)
def merchants_client(requests_mock) -> ApiClient:
request_headers = {
PARTNER_ID_HEADER: "sandbox-partner",
}
requests_mock.register_uri(
"GET",
"/banking/v2/merchants",
json=GET_MERCHANTS_PAGED,
request_headers=request_headers,
)
requests_mock.register_uri(
"GET",
"/banking/v2/merchants?pageId=2",
json=GET_MERCHANTS,
request_headers=request_headers,
)
return make_sandbox(requests_mock)
def forecasts_client(requests_mock) -> ApiClient:
request_headers = {
COMPANY_ID_HEADER: "CompanyID1234",
PARTNER_ID_HEADER: "sandbox-partner",
}
requests_mock.register_uri(
"GET",
"/forecasting/v2/forecasts",
json=GET_FORECASTS_PAGED,
request_headers=request_headers,
)
requests_mock.register_uri(
"GET",
"/forecasting/v2/forecasts?pageId=2",
json=GET_FORECASTS,
request_headers=request_headers,
)
return make_sandbox(requests_mock)