XF/pluto
XF
/
pluto
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Write various transaction test cases

This commit is contained in:
Mark Veidemanis 2022-02-24 22:27:04 +00:00
parent 158b452167
commit ee93a91c04
Signed by: m
GPG Key ID: 5ACFCEED46C0904F
1 changed files with 111 additions and 36 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

147
handler/tests/test_transactions.py
View File

@ -40,51 +40,84 @@ class TestTransactions(TestCase):
transactions.r.keys = self.mock_keys
transactions.r.get = self.mock_get
# Mock some callbacks
self.transactions.irc = MagicMock()
self.transactions.irc.sendmsg = MagicMock()
self.transactions.release_funds = MagicMock()
self.transactions.notify = MagicMock()
self.transactions.notify.notify_complete_trade = MagicMock()
self.trade_1 = {
"id": "uuid1",
"buyer": "test_buyer_1",
"currency": "GBP",
"amount": "1",
"amount_xmr": "0.3",
"reference": "TEST-1",
# Mock the rates
self.transactions.money = MagicMock()
self.transactions.money.get_rates_all = MagicMock()
self.transactions.money.get_rates_all.return_value = {"GBP": 0.8}
self.trades = {
1: {
"id": "uuid1",
"buyer": "test_buyer_1",
"currency": "GBP",
"amount": "1",
"amount_xmr": "0.3",
"reference": "TEST-1",
},
2: {
"id": "uuid2",
"buyer": "test_buyer_2",
"currency": "GBP",
"amount": "1",
"amount_xmr": "0.3",
"reference": "TEST-2",
},
3: {
"id": "uuid3",
"buyer": "test_buyer_3",
"currency": "GBP",
"amount": "1000",
"amount_xmr": "3",
"reference": "TEST-3",
},
4: {
"id": "uuid4",
"buyer": "test_buyer_4",
"currency": "GBP",
"amount": "10",
"amount_xmr": "0.5",
"reference": "TEST-4",
},
}
self.trade_2 = {
"id": "uuid2",
"buyer": "test_buyer_2",
"currency": "GBP",
"amount": "1",
"amount_xmr": "0.3",
"reference": "TEST-2",
}
self.return_trades = [1, 2, 3]
@property
def test_data_copy(self):
return deepcopy(self.test_data)
def data_custom(self, amount, currency, reference):
test_data = self.test_data_copy
test_data["data"]["reference"] = reference
test_data["data"]["legs"][0]["amount"] = amount
test_data["data"]["legs"][0]["currency"] = currency
return test_data
def mock_hgetall(self, string):
if string == "trade.TEST-1":
return self.trade_1
elif string == "trade.TEST-2":
return self.trade_2
print("MOCK HGERALL CALLED WITH", string)
ref = string.split(".")[1]
for num, trade in self.trades.items():
if trade["reference"] == ref:
return trade
def mock_hmset(self, string, data):
print("HMSET", string, data)
def mock_keys(self, string):
return ["uuid1", "uuid2"]
return [v["id"] for k, v in self.trades.items() if k in self.return_trades]
def mock_get(self, string):
if string == "uuid1":
return "TEST-1"
elif string == "uuid2":
return "TEST-2"
print("MOCK GET", string)
for num, trade in self.trades.items():
if trade["id"] == string:
return trade["reference"]
def test_transaction(self):
self.transactions.transaction(self.test_data)
@ -98,38 +131,80 @@ class TestTransactions(TestCase):
def test_transaction_invalid(self):
invalid_data = self.test_data_copy
invalid_data["data"]["reference"] = "sss"
invalid_data["data"]["amount"] = 2000
invalid_data["data"]["currency"] = "SEK"
invalid_data = self.data_custom(2000, "SEK", "sss")
self.transactions.transaction(invalid_data)
self.transactions.release_funds.assert_not_called()
def test_transaction_malformed(self):
pass
malformed_data = self.test_data_copy
del malformed_data["data"]
self.transactions.transaction(malformed_data)
self.transactions.release_funds.assert_not_called()
malformed_data = self.test_data_copy
del malformed_data["data"]["type"]
self.transactions.transaction(malformed_data)
self.transactions.release_funds.assert_not_called()
def test_transaction_no_reference_fail(self):
pass
no_reference_fail = self.data_custom(1, "GBP", "none")
no_reference_fail["data"]["reference"] = "none"
self.transactions.transaction(no_reference_fail)
self.transactions.release_funds.assert_not_called()
def test_transaction_no_reference_pass(self):
pass
no_reference_pass = self.data_custom(1, "GBP", "none")
no_reference_pass["data"]["reference"] = "none"
self.return_trades = [1]
self.transactions.transaction(no_reference_pass)
self.transactions.release_funds.assert_called_with("uuid1", "TEST-1")
def test_transaction_large(self):
exceeds_max = self.data_custom(1000, "GBP", "TEST-3")
self.transactions.transaction(exceeds_max)
self.transactions.release_funds.assert_called_once_with("uuid3", "TEST-3")
def test_transaction_no_reference_exceeds_max(self):
pass
exceeds_max = self.data_custom(1000, "GBP", "noref")
self.transactions.transaction(exceeds_max)
self.transactions.release_funds.assert_not_called()
def test_transaction_wrong_currency(self):
pass
wrong_currency = self.data_custom(1, "EUR", "TEST-1")
self.transactions.transaction(wrong_currency)
self.transactions.release_funds.assert_not_called()
wrong_currency = self.data_custom(1, "EUR", "none")
self.transactions.transaction(wrong_currency)
self.transactions.release_funds.assert_not_called()
def test_transaction_wrong_amount(self):
pass
self.transactions.money.get_acceptable_margins = MagicMock()
self.transactions.money.get_acceptable_margins.return_value = (0.8, 1.8)
wrong_amount = self.data_custom(10, "GBP", "TEST-1")
self.transactions.transaction(wrong_amount)
self.transactions.release_funds.assert_not_called()
wrong_amount = self.data_custom(10, "GBP", "none")
self.transactions.transaction(wrong_amount)
self.transactions.release_funds.assert_not_called()
def test_transaction_pending(self):
pass
pending_tx = self.test_data_copy
pending_tx["data"]["state"] = "pending"
self.transactions.transaction(pending_tx)
self.transactions.release_funds.assert_not_called()
def test_transaction_too_low(self):
pass
too_low = self.data_custom(5, "GBP", "TEST-1")
self.transactions.transaction(too_low)
self.transactions.release_funds.assert_not_called()
def test_transaction_too_high(self):
pass
too_high = self.data_custom(15, "GBP", "TEST-1")
self.transactions.transaction(too_high)
self.transactions.release_funds.assert_not_called()
# def test_transaction_pending_then_completed(self):
# pass