def test_create(self): sot = segment.Segment(**SEGMENT) self.assertEqual(SEGMENT["id"], sot.id) self.assertEqual(SEGMENT["uuid"], sot.uuid) self.assertEqual(SEGMENT["created_at"], sot.created_at) self.assertEqual(SEGMENT["updated_at"], sot.updated_at) self.assertEqual(SEGMENT["name"], sot.name) self.assertEqual(SEGMENT["description"], sot.description) self.assertEqual(SEGMENT["recovery_method"], sot.recovery_method) self.assertEqual(SEGMENT["service_type"], sot.service_type)
def test_basic(self): sot = segment.Segment(SEGMENT) self.assertEqual("segment", sot.resource_key) self.assertEqual("segments", sot.resources_key) self.assertEqual("/segments", sot.base_path) self.assertEqual("ha", sot.service.service_type) self.assertTrue(sot.allow_list) self.assertTrue(sot.allow_fetch) self.assertTrue(sot.allow_create) self.assertTrue(sot.allow_commit) self.assertTrue(sot.allow_delete) self.assertDictEqual({"limit": "limit", "marker": "marker", "recovery_method": "recovery_method", "service_type": "service_type", "sort_dir": "sort_dir", "sort_key": "sort_key"}, sot._query_mapping._mapping)
def data(TEST): TEST.masakari_segment = test_data_utils.TestDataContainer() segment1 = segment.Segment( uuid=uuidsentinel.segment1, name='test', recovery_method='auto', service_type='service', description='demo', is_enabled=True, ) segment2 = segment.Segment( uuid=uuidsentinel.segment2, name='test2', recovery_method='auto', service_type='service', description='demo', is_enabled=False, ) segment3 = segment.Segment( uuid=uuidsentinel.segment3, name='test3', recovery_method='auto', service_type='service', description='demo', is_enabled=True, ) TEST.masakari_segment.add(segment1) TEST.masakari_segment.add(segment2) TEST.masakari_segment.add(segment3) TEST.masakari_host = test_data_utils.TestDataContainer() host1 = host.Host(uuid=uuidsentinel.host1, name="test", reserved='True', type='service', control_attributes='test', failover_segment_id=uuidsentinel.segment1, on_maintenance='False') TEST.masakari_host.add(host1) TEST.hypervisors = test_data_utils.TestDataContainer() hypervisor1 = Hypervisor(HypervisorManager, { 'id': '1', 'hypervisor_hostname': "test" }) TEST.hypervisors.add(hypervisor1) TEST.masakari_notification = test_data_utils.TestDataContainer() notification1 = notification.Notification( notification_uuid=uuidsentinel.notification1, status='new', generated_time=(NOW - datetime.timedelta(seconds=2)), payload='test', type='type1', source_host_uuid=uuidsentinel.host1) notification2 = notification.Notification( notification_uuid=uuidsentinel.notification2, status='running', generated_time=(NOW - datetime.timedelta(seconds=3)), payload='test', type='type2', source_host_uuid=uuidsentinel.host2) notification3 = notification.Notification( notification_uuid=uuidsentinel.notification3, status='error', generated_time=(NOW - datetime.timedelta(seconds=4)), payload='test', type='type3', source_host_uuid=uuidsentinel.host3) TEST.masakari_notification.add(notification1) TEST.masakari_notification.add(notification2) TEST.masakari_notification.add(notification3)