def test_remind_reminds(self): server_mgr = self.fake_client_set.compute.servers instances.admin_client_set().AndReturn(self.fake_client_set) user_mgr = self.fake_client_set.identity_admin.users server = doubles.make(self.mox, doubles.Server, id='v2', name='instance-2', user_id='UID') user = doubles.make(self.mox, doubles.User, id='UID', name='user', email='*****@*****.**', fullname='The Test User Fullname') expires = datetime(2013, 1, 19, 11, 12, 13) instances.datetime.utcnow().AndReturn('UTCNOW') instances.InstanceDataDAO.remind_list('UTCNOW')\ .AndReturn([InstanceData(instance_id='v2', expires_at=expires)]) server_mgr.get('v2').AndReturn(server) user_mgr.get('UID').AndReturn(user) instances.send_instance_reminder(user.email, 'instance-2', 'v2', expires, greeting=user.fullname) instances.InstanceDataDAO.update('v2', remind_at=None) self.mox.ReplayAll() with self.app.test_request_context(): instances.remind_about_instances()
def test_remind_other_exception(self): server_mgr = self.fake_client_set.compute.servers expires = datetime(2013, 1, 19, 11, 12, 13) instances.admin_client_set().AndReturn(self.fake_client_set) instances.datetime.utcnow().AndReturn('UTCNOW') instances.InstanceDataDAO.remind_list('UTCNOW')\ .AndReturn([InstanceData(instance_id='v1', expires_at=expires)]) server_mgr.get('v1').AndRaise(RuntimeError('log me')) self.app.logger.exception(mox.IsA(basestring)) self.mox.ReplayAll() with self.app.test_request_context(): instances.remind_about_instances()
def test_remind_server_not_found(self): server_mgr = self.fake_client_set.compute.servers expires = datetime(2013, 1, 19, 11, 12, 13) instances.admin_client_set().AndReturn(self.fake_client_set) instances.datetime.utcnow().AndReturn('UTCNOW') instances.InstanceDataDAO.remind_list('UTCNOW')\ .AndReturn([InstanceData(instance_id='v1', expires_at=expires)]) server_mgr.get('v1').AndRaise(osc_exc.NotFound('deleted')) instances.InstanceDataDAO.delete('v1') self.mox.ReplayAll() with self.app.test_request_context(): instances.remind_about_instances()
def test_remind_user_not_found(self): server_mgr = self.fake_client_set.compute.servers user_mgr = self.fake_client_set.identity_admin.users server = doubles.make(self.mox, doubles.Server, id='v2', name='instance-2', user_id='UID') expires = datetime(2013, 1, 19, 11, 12, 13) instances.admin_client_set().AndReturn(self.fake_client_set) instances.datetime.utcnow().AndReturn('UTCNOW') instances.InstanceDataDAO.remind_list('UTCNOW')\ .AndReturn([InstanceData(instance_id='v2', expires_at=expires)]) server_mgr.get('v2').AndReturn(server) user_mgr.get('UID').AndRaise(osc_exc.NotFound('deleted')) instances.InstanceDataDAO.update('v2', remind_at=None) self.mox.ReplayAll() with self.app.test_request_context(): instances.remind_about_instances()