def test_balance_save_frequency_check(data_dir, username):
msg_aggregator = MessagesAggregator()
data = DataHandler(data_dir, msg_aggregator)
data.unlock(username, '123', create_new=True)
now = int(time.time())
data_save_ts = now - 24 * 60 * 60 + 20
data.db.add_multiple_location_data([LocationData(
time=data_save_ts, location='kraken', usd_value='1500',
)])
assert not data.should_save_balances()
success, msg = data.set_settings({'balance_save_frequency': 5})
assert success
assert msg == '', f'set settings returned error: "{msg}"'
assert data.should_save_balances()
last_save_ts = data.db.get_last_balance_save_time()
assert last_save_ts == data_save_ts
def test_balance_save_frequency_check(data_dir, username):
data = DataHandler(data_dir)
data.unlock(username, '123', create_new=True)
now = int(time.time())
data_save_ts = now - 24 * 60 * 60 + 20
data.db.add_multiple_location_data([
LocationData(
time=data_save_ts,
location='kraken',
usd_value='1500',
)
])
assert not data.should_save_balances()
data.db.set_settings({'balance_save_frequency': 5})
assert data.should_save_balances()
last_save_ts = data.db.get_last_balance_save_time()
assert last_save_ts == data_save_ts
def test_get_latest_location_value_distribution(data_dir, username):
data = DataHandler(data_dir)
data.unlock(username, '123', create_new=True)
locations = [
LocationData(
time=Timestamp(1451606400),
location='kraken',
usd_value='100',
),
LocationData(
time=Timestamp(1451606400),
location='banks',
usd_value='1000',
),
LocationData(
time=Timestamp(1461606500),
location='poloniex',
usd_value='50',
),
LocationData(
time=Timestamp(1461606500),
location='kraken',
usd_value='200',
),
LocationData(
time=Timestamp(1461606500),
location='banks',
usd_value='50000',
),
LocationData(
time=Timestamp(1491607800),
location='poloniex',
usd_value='100',
),
LocationData(
time=Timestamp(1491607800),
location='kraken',
usd_value='2000',
),
LocationData(
time=Timestamp(1491607800),
location='banks',
usd_value='10000',
),
LocationData(
time=Timestamp(1491607800),
location='blockchain',
usd_value='200000',
),
]
data.db.add_multiple_location_data(locations)
distribution = data.db.get_latest_location_value_distribution()
assert len(distribution) == 4
assert all(entry.time == Timestamp(1491607800) for entry in distribution)
assert distribution[0].location == 'banks'
assert distribution[0].usd_value == '10000'
assert distribution[1].location == 'blockchain'
assert distribution[1].usd_value == '200000'
assert distribution[2].location == 'kraken'
assert distribution[2].usd_value == '2000'
assert distribution[3].location == 'poloniex'
assert distribution[3].usd_value == '100'
def add_starting_balances(datahandler) -> List[AssetBalance]:
"""Adds some starting balances and other data to a testing instance"""
balances = [
AssetBalance(
time=Timestamp(1488326400),
asset=A_BTC,
amount='1',
usd_value='1222.66',
),
AssetBalance(
time=Timestamp(1488326400),
asset=A_ETH,
amount='10',
usd_value='4517.4',
),
AssetBalance(
time=Timestamp(1488326400),
asset=A_EUR,
amount='100',
usd_value='61.5',
),
AssetBalance(
time=Timestamp(1488326400),
asset=A_XMR,
amount='5',
usd_value='135.6',
),
]
datahandler.db.add_multiple_balances(balances)
# Also add an unknown/invalid asset. This will generate a warning
cursor = datahandler.db.conn.cursor()
cursor.execute(
'INSERT INTO timed_balances('
' time, currency, amount, usd_value) '
' VALUES(?, ?, ?, ?)',
(1469326500, 'ADSADX', '10.1', '100.5'),
)
datahandler.db.conn.commit()
location_data = [
LocationData(
time=Timestamp(1451606400),
location='kraken',
usd_value='100',
),
LocationData(
time=Timestamp(1451606400),
location='banks',
usd_value='1000',
),
LocationData(
time=Timestamp(1461606500),
location='poloniex',
usd_value='50',
),
LocationData(
time=Timestamp(1461606500),
location='kraken',
usd_value='200',
),
LocationData(
time=Timestamp(1461606500),
location='banks',
usd_value='50000',
),
LocationData(
time=Timestamp(1491607800),
location='poloniex',
usd_value='100',
),
LocationData(
time=Timestamp(1491607800),
location='kraken',
usd_value='2000',
),
LocationData(
time=Timestamp(1491607800),
location='banks',
usd_value='10000',
),
LocationData(
time=Timestamp(1491607800),
location='blockchain',
usd_value='200000',
),
LocationData(
time=Timestamp(1451606400),
location='total',
usd_value='1500',
),
LocationData(
time=Timestamp(1461606500),
location='total',
usd_value='4500',
),
LocationData(
time=Timestamp(1491607800),
location='total',
usd_value='10700.5',
),
]
datahandler.db.add_multiple_location_data(location_data)
return balances