def test_process_usage_for_updates_finish_resize_end_later_update(self): notification = self._create_mock_notification() raw = self.mox.CreateMockAnything() raw.event = 'compute.instance.finish_resize.end' usage = self.mox.CreateMockAnything() usage.launched_at = utils.decimal_utc(LATER_DUMMY_TIME) usage.instance_type_id = INSTANCE_TYPE_ID_2 usage.instance_flavor_id = INSTANCE_FLAVOR_ID_2 views.STACKDB.get_or_create_instance_usage(instance=INSTANCE_ID_1, request_id=REQUEST_ID_1) \ .AndReturn((usage, True)) views.STACKDB.save(usage) self.mox.ReplayAll() views._process_usage_for_updates(raw, notification) self.assertEqual(usage.instance_type_id, INSTANCE_TYPE_ID_1) self.assertEqual(usage.instance_flavor_id, INSTANCE_FLAVOR_ID_1) self.assertEqual(usage.launched_at, utils.decimal_utc(LATER_DUMMY_TIME)) self.assertEquals(usage.tenant, TENANT_ID_1) self.assertEquals(usage.os_architecture, OS_ARCH_1) self.assertEquals(usage.os_version, OS_VERSION_1) self.assertEquals(usage.os_distro, OS_DISTRO_1) self.assertEquals(usage.rax_options, RAX_OPTIONS_1) self.mox.VerifyAll()
def test_process_usage_for_updates_revert_end(self): when_time = datetime.datetime.utcnow() when_decimal = utils.decimal_utc(when_time) notif = utils.create_nova_notif(request_id=REQUEST_ID_1, launched=str(when_time)) json_str = json.dumps(notif) event = 'compute.instance.resize.revert.end' raw = utils.create_raw(self.mox, when_decimal, event=event, json_str=json_str) usage = self.mox.CreateMockAnything() usage.instance = INSTANCE_ID_1 usage.request_id = REQUEST_ID_1 usage.instance_type_id = '1' views.STACKDB.get_or_create_instance_usage(instance=INSTANCE_ID_1, request_id=REQUEST_ID_1)\ .AndReturn((usage, True)) views.STACKDB.save(usage) self.mox.ReplayAll() views._process_usage_for_updates(raw, notif[1]) self.assertEqual(usage.instance, INSTANCE_ID_1) self.assertEqual(usage.request_id, REQUEST_ID_1) self.assertEqual(usage.instance_type_id, '1') self.assertEqual(usage.launched_at, when_decimal) self.mox.VerifyAll()
def test_process_usage_for_updates_finish_resize_end(self): notification = self._create_mock_notification() raw = self.mox.CreateMockAnything() raw.event = 'compute.instance.finish_resize.end' usage = self.mox.CreateMockAnything() usage.launched_at = None usage.instance_type_id = INSTANCE_TYPE_ID_2 usage.instance_flavor_id = INSTANCE_FLAVOR_ID_2 views.STACKDB.get_or_create_instance_usage(instance=INSTANCE_ID_1, request_id=REQUEST_ID_1) \ .AndReturn((usage, True)) views.STACKDB.save(usage) self.mox.ReplayAll() views._process_usage_for_updates(raw, notification) self.assertEqual(usage.instance_type_id, INSTANCE_TYPE_ID_1) self.assertEqual(usage.instance_flavor_id, INSTANCE_FLAVOR_ID_1) self.assertEquals(usage.tenant, TENANT_ID_1) self.assertEquals(usage.os_architecture, OS_ARCH_1) self.assertEquals(usage.os_version, OS_VERSION_1) self.assertEquals(usage.os_distro, OS_DISTRO_1) self.assertEquals(usage.rax_options, RAX_OPTIONS_1) self.mox.VerifyAll()
def test_process_usage_for_updates_prep_end(self): notification = self.mox.CreateMockAnything() notification.launched_at = str(DUMMY_TIME) notification.tenant = TENANT_ID_1 notification.rax_options = RAX_OPTIONS_1 notification.os_architecture = OS_ARCH_1 notification.os_version = OS_VERSION_1 notification.os_distro = OS_DISTRO_1 notification.instance = INSTANCE_ID_1 notification.request_id = REQUEST_ID_1 notification.new_instance_type_id = INSTANCE_TYPE_ID_2 notification.message = None raw = self.mox.CreateMockAnything() raw.event = 'compute.instance.resize.prep.end' usage = self.mox.CreateMockAnything() usage.launched_at = None views.STACKDB.get_or_create_instance_usage(instance=INSTANCE_ID_1, request_id=REQUEST_ID_1) \ .AndReturn((usage, True)) views.STACKDB.save(usage) self.mox.ReplayAll() views._process_usage_for_updates(raw, notification) self.assertEqual(usage.instance_type_id, INSTANCE_TYPE_ID_2) self.assertEquals(usage.tenant, TENANT_ID_1) self.assertEquals(usage.os_architecture, OS_ARCH_1) self.assertEquals(usage.os_version, OS_VERSION_1) self.assertEquals(usage.os_distro, OS_DISTRO_1) self.assertEquals(usage.rax_options, RAX_OPTIONS_1) self.mox.VerifyAll()
def test_process_usage_for_updates_create_end_error_message(self): notification = self.mox.CreateMockAnything() notification.message = 'Error' raw = self.mox.CreateMockAnything() raw.event = 'compute.instance.create.end' self.mox.ReplayAll() views._process_usage_for_updates(raw, notification) self.mox.VerifyAll()
def test_process_usage_for_updates_prep_end(self): kwargs = {'launched': str(DUMMY_TIME), 'new_type_id': INSTANCE_TYPE_ID_2, 'tenant_id': TENANT_ID_1} notification = utils.create_nova_notif(request_id=REQUEST_ID_1, **kwargs) event = 'compute.instance.resize.prep.end' raw, usage = self._setup_process_usage_mocks(event, notification) views._process_usage_for_updates(raw, notification[1]) self.assertEqual(usage.instance_type_id, INSTANCE_TYPE_ID_2) self.assertEquals(usage.tenant, TENANT_ID_1) self.mox.VerifyAll()
def test_process_usage_for_updates_create_end(self): kwargs = {'launched': str(DUMMY_TIME), 'tenant_id': TENANT_ID_1} notification = utils.create_nova_notif(request_id=REQUEST_ID_1, **kwargs) event = 'compute.instance.create.end' raw, usage = self._setup_process_usage_mocks(event, notification) views._process_usage_for_updates(raw, notification[1]) self.assertEqual(usage.launched_at, utils.decimal_utc(DUMMY_TIME)) self.assertEqual(usage.tenant, TENANT_ID_1) self.mox.VerifyAll()
def test_process_usage_for_updates_create_end_error_message(self): kwargs = {'launched': str(DUMMY_TIME), 'tenant_id': TENANT_ID_1} notification = utils.create_nova_notif(request_id=REQUEST_ID_1, **kwargs) notification[1]['payload']['message'] = "Error" event = 'compute.instance.create.end' when_time = DUMMY_TIME when_decimal = utils.decimal_utc(when_time) json_str = json.dumps(notification) raw = utils.create_raw(self.mox, when_decimal, event=event, json_str=json_str) self.mox.ReplayAll() views._process_usage_for_updates(raw, notification[1]) self.mox.VerifyAll()
def test_process_usage_for_updates_create_end(self): kwargs = {'launched': str(DUMMY_TIME), 'tenant_id': TENANT_ID_1, 'rax_options': RAX_OPTIONS_1, 'os_architecture': OS_ARCH_1, 'os_version': OS_VERSION_1, 'os_distro': OS_DISTRO_1 } notification = utils.create_nova_notif(request_id=REQUEST_ID_1, **kwargs) event = 'compute.instance.create.end' raw, usage = self._setup_process_usage_mocks(event, notification) views._process_usage_for_updates(raw, notification[1]) self.assertEqual(usage.launched_at, utils.decimal_utc(DUMMY_TIME)) self.assertEqual(usage.tenant, TENANT_ID_1) self.assertEquals(usage.os_architecture, OS_ARCH_1) self.assertEquals(usage.os_version, OS_VERSION_1) self.assertEquals(usage.os_distro, OS_DISTRO_1) self.assertEquals(usage.rax_options, RAX_OPTIONS_1) self.mox.VerifyAll()
def test_process_usage_for_updates_create_end_success_message(self): notification = self._create_mock_notification() notification.message = 'Success' raw = self.mox.CreateMockAnything() raw.event = 'compute.instance.create.end' usage = self.mox.CreateMockAnything() usage.launched_at = None views.STACKDB.get_or_create_instance_usage(instance=INSTANCE_ID_1, request_id=REQUEST_ID_1) \ .AndReturn((usage, True)) views.STACKDB.save(usage) self.mox.ReplayAll() views._process_usage_for_updates(raw, notification) self.assertEqual(usage.launched_at, utils.decimal_utc(DUMMY_TIME)) self.assertEqual(usage.tenant, TENANT_ID_1) self.assertEquals(usage.os_architecture, OS_ARCH_1) self.assertEquals(usage.os_version, OS_VERSION_1) self.assertEquals(usage.os_distro, OS_DISTRO_1) self.assertEquals(usage.rax_options, RAX_OPTIONS_1) self.mox.VerifyAll()