nomy-data-models 0.2.4

Data model definitions for Nomy wallet analysis data processing
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156

"""Tests for the raw_trade module."""

import uuid
from datetime import datetime
from decimal import Decimal

import pytest

from nomy_data_models.models.raw_trade import RawTrade
from nomy_data_models.models.trade_base import MarketType


@pytest.fixture
def raw_trade():
    """Fixture providing a RawTrade instance with predefined values."""
    trade = RawTrade()
    trade.id = uuid.uuid4()
    trade.event_at = datetime.now()
    trade.txn_id = "0x1234567890123456789012345678901234567890123456789012345678901234"
    trade.wallet_address = "0x1234567890123456789012345678901234567890"
    trade.chain_id = 1
    trade.exchange = "uniswap"
    trade.is_buy = True
    trade.token_price = Decimal("1000.0")
    trade.token_symbol_pair = "ETH/USDT"
    trade.token_address_pair = "0x1234567890123456789012345678901234567890/0x0987654321098765432109876543210987654321"
    trade.base_token_symbol = "ETH"
    trade.quote_token_symbol = "USDT"
    trade.base_token_address = "0x1234567890123456789012345678901234567890"
    trade.quote_token_address = "0x0987654321098765432109876543210987654321"
    trade.base_amount = Decimal("1.0")
    trade.quote_amount = Decimal("1000.0")
    trade.usd_amount = Decimal("1000.0")
    trade.market_type = MarketType.SPOT
    return trade


@pytest.fixture
def minimal_raw_trade():
    """Fixture providing a RawTrade instance with minimal required attributes."""
    trade = RawTrade()
    trade.id = uuid.uuid4()
    trade.wallet_address = "0x1234567890123456789012345678901234567890"
    trade.token_symbol_pair = "ETH/USDT"
    trade.is_buy = True
    return trade


class TestRawTrade:
    """Test cases for RawTrade model."""

    def test_raw_trade_creation(self, raw_trade):
        """Test creating a RawTrade instance with all attributes.

        Args:
            raw_trade: Fixture providing a fully populated RawTrade instance
        """
        # Verify the instance was created correctly
        assert raw_trade.id is not None, "ID should be set"
        assert raw_trade.event_at is not None, "Event timestamp should be set"
        assert (
            raw_trade.txn_id
            == "0x1234567890123456789012345678901234567890123456789012345678901234"
        ), "Transaction ID should match"
        assert (
            raw_trade.wallet_address == "0x1234567890123456789012345678901234567890"
        ), "Wallet address should match"
        assert raw_trade.chain_id == 1, "Chain ID should be 1"
        assert raw_trade.exchange == "uniswap", "Exchange should be 'uniswap'"
        assert raw_trade.is_buy is True, "is_buy should be True"
        assert raw_trade.token_price == Decimal(
            "1000.0"
        ), "Token price should be 1000.0"
        assert (
            raw_trade.token_symbol_pair == "ETH/USDT"
        ), "Token symbol pair should be 'ETH/USDT'"
        assert (
            raw_trade.token_address_pair
            == "0x1234567890123456789012345678901234567890/0x0987654321098765432109876543210987654321"
        ), "Token address pair should match"
        assert raw_trade.base_token_symbol == "ETH", "Base token symbol should be 'ETH'"
        assert (
            raw_trade.quote_token_symbol == "USDT"
        ), "Quote token symbol should be 'USDT'"
        assert (
            raw_trade.base_token_address == "0x1234567890123456789012345678901234567890"
        ), "Base token address should match"
        assert (
            raw_trade.quote_token_address
            == "0x0987654321098765432109876543210987654321"
        ), "Quote token address should match"
        assert raw_trade.base_amount == Decimal("1.0"), "Base amount should be 1.0"
        assert raw_trade.quote_amount == Decimal(
            "1000.0"
        ), "Quote amount should be 1000.0"
        assert raw_trade.usd_amount == Decimal("1000.0"), "USD amount should be 1000.0"
        assert raw_trade.market_type == MarketType.SPOT, "Market type should be SPOT"

    def test_raw_trade_repr(self, minimal_raw_trade):
        """Test the string representation of a RawTrade.

        Args:
            minimal_raw_trade: Fixture providing a RawTrade with minimal attributes
        """
        # Verify the string representation
        expected_repr = f"<RawTrade(id={minimal_raw_trade.id}, wallet=0x123456..., pair=ETH/USDT, is_buy=True)"
        assert (
            str(minimal_raw_trade) == expected_repr
        ), "String representation should match expected format"

    def test_raw_trade_db_integration(self, session):
        """Test database integration with the RawTrade model.

        Args:
            session: SQLAlchemy session fixture
        """
        # Create a new RawTrade instance
        trade = RawTrade()
        trade.event_at = datetime.now()
        trade.txn_id = (
            "0x1234567890123456789012345678901234567890123456789012345678901234"
        )
        trade.wallet_address = "0x1234567890123456789012345678901234567890"
        trade.chain_id = 1
        trade.exchange = "uniswap"
        trade.is_buy = True
        trade.token_price = Decimal("1000.0")
        trade.token_symbol_pair = "ETH/USDT"
        trade.token_address_pair = "0x1234567890123456789012345678901234567890/0x0987654321098765432109876543210987654321"
        trade.base_token_symbol = "ETH"
        trade.quote_token_symbol = "USDT"
        trade.base_token_address = "0x1234567890123456789012345678901234567890"
        trade.quote_token_address = "0x0987654321098765432109876543210987654321"
        trade.base_amount = Decimal("1.0")
        trade.quote_amount = Decimal("1000.0")
        trade.usd_amount = Decimal("1000.0")
        trade.market_type = MarketType.SPOT

        # Add to session and commit
        session.add(trade)
        session.commit()

        # Verify the trade was saved to the database
        retrieved = session.query(RawTrade).filter_by(id=trade.id).first()
        assert retrieved is not None, "Trade should be retrievable from database"
        assert retrieved.id == trade.id, "Retrieved trade ID should match"
        assert (
            retrieved.wallet_address == "0x1234567890123456789012345678901234567890"
        ), "Retrieved wallet address should match"
        assert (
            retrieved.token_symbol_pair == "ETH/USDT"
        ), "Retrieved token symbol pair should match"


if __name__ == "__main__":
    pytest.main(["-v", __file__])