Esempio n. 1
0
 def test_get_usage_type_cost_with_exception(self, plugin_run_mock):
     for exc in (KeyError(), AttributeError()):
         plugin_run_mock.side_effect = exc
         result = ServicePlugin._get_usage_type_cost(
             start=datetime.date(2013, 10, 10),
             end=datetime.date(2013, 10, 20),
             usage_type=self.usage_type,
             forecast=True,
             ventures=self.ventures,
         )
         self.assertEquals(result, 0)
Esempio n. 2
0
 def test_get_usage_type_cost_with_exception(self, plugin_run_mock):
     for exc in (KeyError(), AttributeError()):
         plugin_run_mock.side_effect = exc
         result = ServicePlugin._get_usage_type_cost(
             start=datetime.date(2013, 10, 10),
             end=datetime.date(2013, 10, 20),
             usage_type=self.usage_type,
             forecast=True,
             ventures=self.ventures,
         )
         self.assertEquals(result, 0)
Esempio n. 3
0
 def test_get_usage_type_cost(self, plugin_run_mock):
     plugin_run_mock.return_value = 100
     result = ServicePlugin._get_usage_type_cost(
         start=datetime.date(2013, 10, 10),
         end=datetime.date(2013, 10, 20),
         usage_type=self.usage_type,
         forecast=True,
         ventures=self.ventures,
     )
     self.assertEquals(result, 100)
     plugin_run_mock.assert_called_with(
         'reports',
         self.usage_type.get_plugin_name(),
         start=datetime.date(2013, 10, 10),
         end=datetime.date(2013, 10, 20),
         usage_type=self.usage_type,
         forecast=True,
         ventures=self.ventures,
         type='total_cost',
     )
Esempio n. 4
0
 def test_get_usage_type_cost(self, plugin_run_mock):
     plugin_run_mock.return_value = 100
     result = ServicePlugin._get_usage_type_cost(
         start=datetime.date(2013, 10, 10),
         end=datetime.date(2013, 10, 20),
         usage_type=self.usage_type,
         forecast=True,
         ventures=self.ventures,
     )
     self.assertEquals(result, 100)
     plugin_run_mock.assert_called_with(
         'reports',
         self.usage_type.get_plugin_name(),
         start=datetime.date(2013, 10, 10),
         end=datetime.date(2013, 10, 20),
         usage_type=self.usage_type,
         forecast=True,
         ventures=self.ventures,
         type='total_cost',
     )