def test_set_usages(self): """ Hamster usages Test Case """ # fake setting need to run plugin settings.HAMSTER_API_URL = "/" with mock.patch( 'ralph_pricing.plugins.hamster.get_venture_capacity' ) as get_venture_capacity: get_venture_capacity.side_effect = mock_get_venture_capacity status, message, args = hamster_runner( today=datetime.datetime.today() ) self.assertTrue(status) usages = DailyUsage.objects.all() self.assertEqual(len(usages), 2) usage_venture1 = DailyUsage.objects.get( pricing_venture=self.venture_1 ) usage_venture2 = DailyUsage.objects.get( pricing_venture=self.venture_2 ) self.assertEqual( usage_venture1.value, 2131231233.0 / (1024 * 1024) ) self.assertEqual( usage_venture2.value, 4234233423.0 / (1024 * 1024) )
def test_fail_plugin(self): """ Testing not configured plugin """ with mock.patch('ralph_pricing.plugins.hamster.get_venture_capacity' ) as get_venture_capacity: get_venture_capacity.side_effect = mock_get_venture_capacity status, message, args = hamster_runner( today=datetime.datetime.today()) self.assertFalse(status)
def test_fail_plugin(self): """ Testing not configured plugin """ with mock.patch( 'ralph_pricing.plugins.hamster.get_venture_capacity' ) as get_venture_capacity: get_venture_capacity.side_effect = mock_get_venture_capacity status, message, args = hamster_runner( today=datetime.datetime.today() ) self.assertFalse(status)
def test_set_usages(self): """ Hamster usages Test Case """ # fake setting need to run plugin settings.HAMSTER_API_URL = "/" with mock.patch('ralph_pricing.plugins.hamster.get_venture_capacity' ) as get_venture_capacity: get_venture_capacity.side_effect = mock_get_venture_capacity status, message, args = hamster_runner( today=datetime.datetime.today()) self.assertTrue(status) usages = DailyUsage.objects.all() self.assertEqual(len(usages), 2) usage_venture1 = DailyUsage.objects.get( pricing_venture=self.venture_1) usage_venture2 = DailyUsage.objects.get( pricing_venture=self.venture_2) self.assertEqual(usage_venture1.value, 2131231233.0 / (1024 * 1024)) self.assertEqual(usage_venture2.value, 4234233423.0 / (1024 * 1024))