def test_privat_mock(self, ):
main.update_from_privat_api()
rate = XRate.get_or_none(from_currency="USD", to_currency="UAH")
self.assertIsNotNone(rate)
self.assertEqual(rate.rate, 30)
rates_count = XRate.select().count()
self.assertEqual(rates_count, 1)
def test_privat_class_mock(self, PrivatMock):
PrivatMock().get_rates.return_value = get_privat_response()
main.update_from_privat_api()
rate = XRate.get_or_none(from_currency="USD", to_currency="UAH")
self.assertIsNotNone(rate)
self.assertEqual(rate.rate, 30)
self.assertTrue(PrivatMock.called)
def _test_privat_timeout(self):
# self.assertRaises(requests.exceptions.ReadTimeout, main.update_from_privat_api)
main.update_from_privat_api()
for from_currency, to_currency in (("USD", "UAH"), ("RUR", "UAH"),
("EUR", "UAH")):
rate = XRate.get_or_none(from_currency=from_currency,
to_currency=to_currency)
self.assertIsNotNone(rate)
self.assertNotEqual(rate.rate, 1.0)