Libraries refactor and add some sinks #4
|
@ -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
|
||||
|
|
Loading…
Reference in New Issue