Example #1
0
def test_saveing_multiple_channel(ms_database: Database):
    ms_database.conn.execute(
        "INSERT INTO token_network (address, settle_timeout) VALUES (?, ?)",
        [
            to_checksum_address(DEFAULT_TOKEN_NETWORK_ADDRESS),
            DEFAULT_TOKEN_NETWORK_SETTLE_TIMEOUT
        ],
    )
    tn_address2 = make_token_network_address()
    ms_database.conn.execute(
        "INSERT INTO token_network (address, settle_timeout) VALUES (?, ?)",
        [
            to_checksum_address(tn_address2),
            DEFAULT_TOKEN_NETWORK_SETTLE_TIMEOUT
        ],
    )

    channel1 = create_channel()
    channel2 = create_channel()
    channel2.token_network_address = tn_address2

    ms_database.upsert_channel(channel1)
    loaded_channel1 = ms_database.get_channel(
        token_network_address=channel1.token_network_address,
        channel_id=channel1.identifier)
    assert loaded_channel1 == channel1
    assert ms_database.channel_count() == 1

    ms_database.upsert_channel(channel2)
    loaded_channel2 = ms_database.get_channel(
        token_network_address=channel2.token_network_address,
        channel_id=channel2.identifier)
    assert loaded_channel2 == channel2
    assert ms_database.channel_count() == 2
Example #2
0
def test_save_and_load_channel(ms_database: Database):
    ms_database.conn.execute(
        "INSERT INTO token_network (address) VALUES (?)",
        [to_checksum_address(DEFAULT_TOKEN_NETWORK_ADDRESS)],
    )
    for update_status in [
            None,
            OnChainUpdateStatus(update_sender_address=Address(bytes([1] * 20)),
                                nonce=random.randint(0, UINT256_MAX)),
    ]:
        channel = create_channel(update_status)
        ms_database.upsert_channel(channel)
        loaded_channel = ms_database.get_channel(
            token_network_address=channel.token_network_address,
            channel_id=channel.identifier)
        assert loaded_channel == channel