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)
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)
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', )
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', )