def test_sdr_feeding(self, name, mod=None, side_effect=None, err_type=None, err_msg=None): sdr = deepcopy(BASIC_SDR) if mod is not None: mod(self, sdr) if side_effect is not None: side_effect(self) sdr_mng = sdr_manager.SDRManager() error = None try: sdr_mng.validate_sdr(sdr) except Exception as e: error = e if err_type is None: self.assertTrue(error is None) sdr_manager.Organization.objects.filter.assert_called_once_with(name='test_user') sdr_manager.Order.objects.get.assert_called_once_with(order_id='1') self._order.get_product_contract.assert_called_once_with('2') self.assertEquals(self._contract, sdr_mng._contract) self.assertEquals(self._order, sdr_mng._order) self.assertEquals(self._timestamp, sdr_mng._time_stamp) sdr_manager.User.objects.get.assert_called_once_with(username='******') else: self.assertTrue(isinstance(error, err_type)) self.assertEquals(unicode(e), err_msg)
def test_update_usage(self): sdr_mng = sdr_manager.SDRManager() sdr_mng._contract = self._contract sdr_mng._order = self._order sdr_mng._time_stamp = self._timestamp sdr_mng.update_usage() self.assertEquals(2, self._contract.correlation_number) self.assertEquals(self._timestamp, self._contract.last_usage) self._order.save.assert_called_once_with()