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)