def test_insert_or_update_transaction(self):
"""
Test the conditional insertion or update of a transaction in the database
"""
# Test insertion
self.database.insert_or_update_transaction(self.transaction1)
transactions = self.database.get_all_transactions()
self.assertEqual(len(transactions), 1)
# Test try to update with older timestamp
before_trans1 = Transaction(
self.transaction1.transaction_id, self.transaction1.assets,
self.transaction1.order_id, self.transaction1.partner_order_id,
Timestamp(int(self.transaction1.timestamp) - 1000))
self.database.insert_or_update_transaction(before_trans1)
transaction = self.database.get_transaction(
self.transaction1.transaction_id)
self.assertEqual(int(transaction.timestamp),
int(self.transaction1.timestamp))
# Test update with newer timestamp
after_trans1 = Transaction(
self.transaction1.transaction_id, self.transaction1.assets,
self.transaction1.order_id, self.transaction1.partner_order_id,
Timestamp(int(self.transaction1.timestamp) + 1000))
self.database.insert_or_update_transaction(after_trans1)
transaction = self.database.get_transaction(
self.transaction1.transaction_id)
self.assertEqual(int(transaction.timestamp),
int(after_trans1.timestamp))
def test_insert_or_update_transaction(db, transaction):
"""
Test the conditional insertion or update of a transaction in the database
"""
# Test insertion
db.insert_or_update_transaction(transaction)
transactions = db.get_all_transactions()
assert len(transactions) == 1
# Test try to update with older timestamp
before_trans1 = Transaction(transaction.transaction_id, transaction.assets,
transaction.order_id,
transaction.partner_order_id,
Timestamp(int(transaction.timestamp) - 1000))
db.insert_or_update_transaction(before_trans1)
transaction = db.get_transaction(transaction.transaction_id)
assert int(transaction.timestamp) == int(transaction.timestamp)
# Test update with newer timestamp
after_trans1 = Transaction(transaction.transaction_id, transaction.assets,
transaction.order_id,
transaction.partner_order_id,
Timestamp(int(transaction.timestamp) + 1000))
db.insert_or_update_transaction(after_trans1)
transaction = db.get_transaction(transaction.transaction_id)
assert int(transaction.timestamp) == int(after_trans1.timestamp)
def get_tx_done_block(ask_amount, bid_amount, traded_amount,
ask_total_traded, bid_total_traded):
ask_pair = AssetPair(AssetAmount(ask_amount, 'BTC'),
AssetAmount(ask_amount, 'MB'))
bid_pair = AssetPair(AssetAmount(bid_amount, 'BTC'),
AssetAmount(bid_amount, 'MB'))
ask = Order(OrderId(TraderId(b'0' * 20), OrderNumber(1)), ask_pair,
Timeout(3600), Timestamp.now(), True)
ask._traded_quantity = ask_total_traded
bid = Order(OrderId(TraderId(b'1' * 20), OrderNumber(1)), bid_pair,
Timeout(3600), Timestamp.now(), False)
bid._traded_quantity = bid_total_traded
tx = Transaction(
TransactionId(TraderId(b'0' * 20), TransactionNumber(1)),
AssetPair(AssetAmount(traded_amount, 'BTC'),
AssetAmount(traded_amount, 'MB')),
OrderId(TraderId(b'0' * 20), OrderNumber(1)),
OrderId(TraderId(b'1' * 20), OrderNumber(1)), Timestamp(0))
tx.transferred_assets.first += AssetAmount(traded_amount, 'BTC')
tx.transferred_assets.second += AssetAmount(traded_amount, 'MB')
tx_done_block = MarketBlock()
tx_done_block.type = b'tx_done'
tx_done_block.transaction = {
'ask': ask.to_status_dictionary(),
'bid': bid.to_status_dictionary(),
'tx': tx.to_dictionary(),
'version': MarketCommunity.PROTOCOL_VERSION
}
tx_done_block.transaction['ask']['address'], tx_done_block.transaction[
'ask']['port'] = "1.1.1.1", 1234
tx_done_block.transaction['bid']['address'], tx_done_block.transaction[
'bid']['port'] = "1.1.1.1", 1234
return tx_done_block
async def setUp(self):
super(TestDatabase, self).setUp()
path = os.path.join(self.getStateDir(), 'sqlite')
if not os.path.exists(path):
os.makedirs(path)
self.database = MarketDB(self.getStateDir(), 'market')
self.order_id1 = OrderId(TraderId(b'3' * 20), OrderNumber(4))
self.order_id2 = OrderId(TraderId(b'4' * 20), OrderNumber(5))
self.order1 = Order(
self.order_id1,
AssetPair(AssetAmount(5, 'BTC'), AssetAmount(6, 'EUR')),
Timeout(3600), Timestamp.now(), True)
self.order2 = Order(
self.order_id2,
AssetPair(AssetAmount(5, 'BTC'), AssetAmount(6, 'EUR')),
Timeout(3600), Timestamp.now(), False)
self.order2.reserve_quantity_for_tick(
OrderId(TraderId(b'3' * 20), OrderNumber(4)), 3)
self.transaction_id1 = TransactionId(b'a' * 32)
self.transaction1 = Transaction(
self.transaction_id1,
AssetPair(AssetAmount(100, 'BTC'), AssetAmount(30, 'MB')),
OrderId(TraderId(b'0' * 20), OrderNumber(1)),
OrderId(TraderId(b'1' * 20), OrderNumber(2)), Timestamp(20000))
self.payment1 = Payment(TraderId(b'0' * 20), self.transaction_id1,
AssetAmount(5, 'BTC'), WalletAddress('abc'),
WalletAddress('def'), PaymentId("abc"),
Timestamp(20000))
self.transaction1.add_payment(self.payment1)
def setUp(self):
# Object creation
self.transaction_id = TransactionId(b'a' * 32)
self.transaction = Transaction(
self.transaction_id,
AssetPair(AssetAmount(100, 'BTC'), AssetAmount(30, 'MC')),
OrderId(TraderId(b'0' * 20), OrderNumber(2)),
OrderId(TraderId(b'1' * 20), OrderNumber(1)), Timestamp(0))
self.proposed_trade = Trade.propose(
TraderId(b'0' * 20), OrderId(TraderId(b'0' * 20), OrderNumber(2)),
OrderId(TraderId(b'1' * 20), OrderNumber(3)),
AssetPair(AssetAmount(100, 'BTC'), AssetAmount(30, 'MC')),
Timestamp(0))
self.tick = Tick(
OrderId(TraderId(b'0' * 20), OrderNumber(1)),
AssetPair(AssetAmount(5, 'BTC'), AssetAmount(5, 'MC')), Timeout(0),
Timestamp(00), True)
self.tick2 = Tick(
OrderId(TraderId(b'0' * 20), OrderNumber(2)),
AssetPair(AssetAmount(500, 'BTC'), AssetAmount(5, 'MC')),
Timeout(0), Timestamp(0), True)
self.order_timestamp = Timestamp.now()
self.order = Order(
OrderId(TraderId(b'0' * 20), OrderNumber(3)),
AssetPair(AssetAmount(50, 'BTC'), AssetAmount(40, 'MC')),
Timeout(5000), self.order_timestamp, False)
self.order.set_verified()
self.order2 = Order(
OrderId(TraderId(b'0' * 20), OrderNumber(4)),
AssetPair(AssetAmount(50, 'BTC'), AssetAmount(10, 'MC')),
Timeout(5), Timestamp(int(time.time() * 1000) - 1000 * 1000), True)
self.order2.set_verified()
def transaction(payment):
transaction_id = TransactionId(b'a' * 32)
transaction = Transaction(
transaction_id,
AssetPair(AssetAmount(100, 'BTC'), AssetAmount(30, 'MB')),
OrderId(TraderId(b'0' * 20), OrderNumber(1)),
OrderId(TraderId(b'1' * 20), OrderNumber(2)), Timestamp(20000))
transaction.add_payment(payment)
return transaction
def setUp(self):
# Object creation
self.memory_transaction_repository = MemoryTransactionRepository(b'0' *
20)
self.transaction_id = TransactionId(b'a' * 32)
self.transaction = Transaction(
self.transaction_id,
AssetPair(AssetAmount(10, 'BTC'), AssetAmount(10, 'MB')),
OrderId(TraderId(b'0' * 20), OrderNumber(1)),
OrderId(TraderId(b'2' * 20), OrderNumber(2)), Timestamp(0))
def setUp(self):
BaseTestCase.setUp(self)
self.ask = Ask(
OrderId(TraderId(b'0' * 20), OrderNumber(1)),
AssetPair(AssetAmount(30, 'BTC'), AssetAmount(30, 'MB')),
Timeout(30), Timestamp(0), True)
self.bid = Ask(
OrderId(TraderId(b'1' * 20), OrderNumber(1)),
AssetPair(AssetAmount(30, 'BTC'), AssetAmount(30, 'MB')),
Timeout(30), Timestamp(0), False)
self.transaction = Transaction(
TransactionId(TraderId(b'0' * 20), TransactionNumber(1)),
AssetPair(AssetAmount(30, 'BTC'), AssetAmount(30, 'MB')),
OrderId(TraderId(b'0' * 20), OrderNumber(1)),
OrderId(TraderId(b'1' * 20), OrderNumber(1)), Timestamp(0))
ask_tx = self.ask.to_block_dict()
bid_tx = self.bid.to_block_dict()
self.tick_block = MarketBlock()
self.tick_block.type = b'ask'
self.tick_block.transaction = {'tick': ask_tx}
self.cancel_block = MarketBlock()
self.cancel_block.type = b'cancel_order'
self.cancel_block.transaction = {
'trader_id': 'a' * 20,
'order_number': 1
}
self.tx_block = MarketBlock()
self.tx_block.type = b'tx_init'
self.tx_block.transaction = {
'ask': ask_tx,
'bid': bid_tx,
'tx': self.transaction.to_dictionary()
}
payment = {
'trader_id': 'a' * 40,
'transaction_number': 3,
'transferred': {
'amount': 3,
'type': 'BTC'
},
'payment_id': 'a',
'address_from': 'a',
'address_to': 'b',
'timestamp': 1234,
'success': True
}
self.payment_block = MarketBlock()
self.payment_block.type = b'tx_payment'
self.payment_block.transaction = {'payment': payment}
def create_from_start_transaction(self, start_transaction):
"""
:type start_transaction: StartTransaction
:rtype: Transaction
"""
transaction = Transaction(start_transaction.transaction_id, start_transaction.assets,
start_transaction.recipient_order_id, start_transaction.order_id, Timestamp.now())
self.transaction_repository.add(transaction)
self._logger.info("Transaction created with id: %s, asset pair %s",
transaction.transaction_id, transaction.assets)
return transaction
def setUp(self):
# Object creation
self.transaction_id = TransactionId(TraderId(b'0' * 20),
TransactionNumber(1))
self.transaction = Transaction(
self.transaction_id,
AssetPair(AssetAmount(100, 'BTC'), AssetAmount(100, 'MB')),
OrderId(TraderId(b'3' * 20), OrderNumber(2)),
OrderId(TraderId(b'2' * 20), OrderNumber(1)), Timestamp(0))
self.proposed_trade = Trade.propose(
TraderId(b'0' * 20), OrderId(TraderId(b'0' * 20), OrderNumber(2)),
OrderId(TraderId(b'1' * 20), OrderNumber(3)),
AssetPair(AssetAmount(100, 'BTC'), AssetAmount(100, 'MB')),
Timestamp(0))
self.payment = Payment(
TraderId(b'0' * 20),
TransactionId(TraderId(b'2' * 20), TransactionNumber(2)),
AssetAmount(3, 'MB'), WalletAddress('a'), WalletAddress('b'),
PaymentId('aaa'), Timestamp(4), True)
