def test_sanitize_supply_cost_raise_ValueError_if_cost_not_found(self): with self.assertRaises(ValueError): helpers.sanitize_supply_cost(self.model, 'some_cost', 'value')
def test_sanitize_supply_cost_set_cost_as_default(self): self.model.cost_names.append('other_cost') helpers.sanitize_supply_cost(self.model, None, 'value') actual = self.model.default_cost self.assertEqual(actual, 'cost')