66 lines
1.5 KiB
Python
66 lines
1.5 KiB
Python
|
# standard imports
|
||
|
|
import os
|
||
|
|
|
||
|
|
# third-party imports
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
# local imports
|
||
|
|
from cic_eth.queue.tx import create as queue_create
|
||
|
|
from cic_eth.db.models.lock import Lock
|
||
|
|
from cic_eth.db.enum import LockEnum
|
||
|
|
from cic_eth.error import LockedError
|
||
|
|
|
||
|
|
|
||
|
|
def test_queue_lock(
|
||
|
|
init_database,
|
||
|
|
default_chain_spec,
|
||
|
|
):
|
||
|
|
|
||
|
|
chain_str = str(default_chain_spec)
|
||
|
|
|
||
|
|
address = '0x' + os.urandom(20).hex()
|
||
|
|
tx_hash = '0x' + os.urandom(32).hex()
|
||
|
|
tx_raw = '0x' + os.urandom(128).hex()
|
||
|
|
|
||
|
|
Lock.set(chain_str, LockEnum.QUEUE)
|
||
|
|
with pytest.raises(LockedError):
|
||
|
|
queue_create(
|
||
|
|
0,
|
||
|
|
address,
|
||
|
|
tx_hash,
|
||
|
|
tx_raw,
|
||
|
|
chain_str
|
||
|
|
)
|
||
|
|
|
||
|
|
Lock.set(chain_str, LockEnum.QUEUE, address=address)
|
||
|
|
with pytest.raises(LockedError):
|
||
|
|
queue_create(
|
||
|
|
0,
|
||
|
|
address,
|
||
|
|
tx_hash,
|
||
|
|
tx_raw,
|
||
|
|
chain_str
|
||
|
|
)
|
||
|
|
|
||
|
|
Lock.reset(chain_str, LockEnum.QUEUE)
|
||
|
|
with pytest.raises(LockedError):
|
||
|
|
queue_create(
|
||
|
|
0,
|
||
|
|
address,
|
||
|
|
tx_hash,
|
||
|
|
tx_raw,
|
||
|
|
chain_str
|
||
|
|
)
|
||
|
|
|
||
|
|
Lock.set(chain_str, LockEnum.QUEUE, address=address, tx_hash=tx_hash)
|
||
|
|
with pytest.raises(LockedError):
|
||
|
|
queue_create(
|
||
|
|
0,
|
||
|
|
address,
|
||
|
|
tx_hash,
|
||
|
|
tx_raw,
|
||
|
|
chain_str
|
||
|
|
)
|
||
|
|
|
||
|
|
|