mirror of
https://github.com/saymrwulf/BraiinsRatchet.git
synced 2026-05-14 20:37:52 +00:00
41 lines
1.6 KiB
Python
41 lines
1.6 KiB
Python
from decimal import Decimal
|
|
import sqlite3
|
|
import unittest
|
|
|
|
from braiins_ratchet.models import MarketSnapshot
|
|
from braiins_ratchet.storage import init_db, latest_market_snapshot, save_market_snapshot
|
|
|
|
|
|
class StorageTests(unittest.TestCase):
|
|
def test_market_snapshot_round_trip_extended_fields(self) -> None:
|
|
conn = sqlite3.connect(":memory:")
|
|
init_db(conn)
|
|
save_market_snapshot(
|
|
conn,
|
|
MarketSnapshot(
|
|
timestamp_utc="2026-04-25T12:00:00+00:00",
|
|
best_price_btc_per_eh_day=Decimal("0.30"),
|
|
best_bid_btc_per_eh_day=Decimal("0.29"),
|
|
best_ask_btc_per_eh_day=Decimal("0.30"),
|
|
last_price_btc_per_eh_day=Decimal("0.31"),
|
|
total_hashrate_eh_s=Decimal("0.25"),
|
|
available_hashrate_eh_s=Decimal("0.21"),
|
|
status="SPOT_INSTRUMENT_STATUS_ACTIVE",
|
|
source="test",
|
|
),
|
|
)
|
|
|
|
snapshot = latest_market_snapshot(conn)
|
|
self.assertIsNotNone(snapshot)
|
|
assert snapshot is not None
|
|
self.assertEqual(snapshot.best_price_btc_per_eh_day, Decimal("0.30"))
|
|
self.assertEqual(snapshot.best_bid_btc_per_eh_day, Decimal("0.29"))
|
|
self.assertEqual(snapshot.best_ask_btc_per_eh_day, Decimal("0.30"))
|
|
self.assertEqual(snapshot.last_price_btc_per_eh_day, Decimal("0.31"))
|
|
self.assertEqual(snapshot.total_hashrate_eh_s, Decimal("0.25"))
|
|
self.assertEqual(snapshot.available_hashrate_eh_s, Decimal("0.21"))
|
|
self.assertEqual(snapshot.status, "SPOT_INSTRUMENT_STATUS_ACTIVE")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|