Add config flow for Coinbase (#45354)
Co-authored-by: Franck Nijhof <frenck@frenck.nl>
This commit is contained in:
parent
efee36a176
commit
fd1d110b80
16 changed files with 1299 additions and 96 deletions
84
tests/components/coinbase/common.py
Normal file
84
tests/components/coinbase/common.py
Normal file
|
@ -0,0 +1,84 @@
|
|||
"""Collection of helpers."""
|
||||
from homeassistant.components.coinbase.const import (
|
||||
CONF_CURRENCIES,
|
||||
CONF_EXCHANGE_RATES,
|
||||
DOMAIN,
|
||||
)
|
||||
from homeassistant.const import CONF_API_KEY, CONF_API_TOKEN
|
||||
|
||||
from .const import GOOD_EXCHNAGE_RATE, GOOD_EXCHNAGE_RATE_2, MOCK_ACCOUNTS_RESPONSE
|
||||
|
||||
from tests.common import MockConfigEntry
|
||||
|
||||
|
||||
class MockPagination:
|
||||
"""Mock pagination result."""
|
||||
|
||||
def __init__(self, value=None):
|
||||
"""Load simple pagination for tests."""
|
||||
self.next_starting_after = value
|
||||
|
||||
|
||||
class MockGetAccounts:
|
||||
"""Mock accounts with pagination."""
|
||||
|
||||
def __init__(self, starting_after=0):
|
||||
"""Init mocked object, forced to return two at a time."""
|
||||
if (target_end := starting_after + 2) >= (
|
||||
max_end := len(MOCK_ACCOUNTS_RESPONSE)
|
||||
):
|
||||
end = max_end
|
||||
self.pagination = MockPagination(value=None)
|
||||
else:
|
||||
end = target_end
|
||||
self.pagination = MockPagination(value=target_end)
|
||||
|
||||
self.accounts = {
|
||||
"data": MOCK_ACCOUNTS_RESPONSE[starting_after:end],
|
||||
}
|
||||
self.started_at = starting_after
|
||||
|
||||
def __getitem__(self, item):
|
||||
"""Handle subscript request."""
|
||||
return self.accounts[item]
|
||||
|
||||
|
||||
def mocked_get_accounts(_, **kwargs):
|
||||
"""Return simplied accounts using mock."""
|
||||
return MockGetAccounts(**kwargs)
|
||||
|
||||
|
||||
def mock_get_current_user():
|
||||
"""Return a simplified mock user."""
|
||||
return {
|
||||
"id": "123456-abcdef",
|
||||
"name": "Test User",
|
||||
}
|
||||
|
||||
|
||||
def mock_get_exchange_rates():
|
||||
"""Return a heavily reduced mock list of exchange rates for testing."""
|
||||
return {
|
||||
"currency": "USD",
|
||||
"rates": {GOOD_EXCHNAGE_RATE_2: "0.109", GOOD_EXCHNAGE_RATE: "0.00002"},
|
||||
}
|
||||
|
||||
|
||||
async def init_mock_coinbase(hass):
|
||||
"""Init Coinbase integration for testing."""
|
||||
config_entry = MockConfigEntry(
|
||||
domain=DOMAIN,
|
||||
unique_id="abcde12345",
|
||||
title="Test User",
|
||||
data={CONF_API_KEY: "123456", CONF_API_TOKEN: "AbCDeF"},
|
||||
options={
|
||||
CONF_CURRENCIES: [],
|
||||
CONF_EXCHANGE_RATES: [],
|
||||
},
|
||||
)
|
||||
config_entry.add_to_hass(hass)
|
||||
|
||||
await hass.config_entries.async_setup(config_entry.entry_id)
|
||||
await hass.async_block_till_done()
|
||||
|
||||
return config_entry
|
Loading…
Add table
Add a link
Reference in a new issue