def _data_setup(self): strategy1_name = "STRATEGY_ID_1" strategy2_name = "STRATEGY_ID_2" strategy3_name = "STRATEGY_ID_3" self.goal1 = utils.create_test_goal(id=1, uuid=w_utils.generate_uuid(), name="GOAL_ID", display_name="Goal") self.goal2 = utils.create_test_goal(id=2, uuid=w_utils.generate_uuid(), name="DUMMY", display_name="Dummy") with freezegun.freeze_time(self.FAKE_TODAY): self.strategy1 = utils.create_test_strategy( id=1, uuid=w_utils.generate_uuid(), name=strategy1_name, display_name="Strategy 1", goal_id=self.goal1.id) with freezegun.freeze_time(self.FAKE_OLD_DATE): self.strategy2 = utils.create_test_strategy( id=2, uuid=w_utils.generate_uuid(), name=strategy2_name, display_name="Strategy 2", goal_id=self.goal1.id) with freezegun.freeze_time(self.FAKE_OLDER_DATE): self.strategy3 = utils.create_test_strategy( id=3, uuid=w_utils.generate_uuid(), name=strategy3_name, display_name="Strategy 3", goal_id=self.goal2.id)
def setUp(self): super(TestAuditTemplateObject, self).setUp() self.fake_audit_template = utils.get_test_audit_template() self.fake_goal1 = utils.create_test_goal( id=1, uuid=w_utils.generate_uuid(), name="DUMMY") self.fake_goal2 = utils.create_test_goal( id=2, uuid=w_utils.generate_uuid(), name="BALANCE_LOAD")
def _data_setup(self): strategy1_name = "STRATEGY_ID_1" strategy2_name = "STRATEGY_ID_2" strategy3_name = "STRATEGY_ID_3" self.goal1 = utils.create_test_goal( id=1, uuid=w_utils.generate_uuid(), name="GOAL_ID", display_name="Goal") self.goal2 = utils.create_test_goal( id=2, uuid=w_utils.generate_uuid(), name="DUMMY", display_name="Dummy") with freezegun.freeze_time(self.FAKE_TODAY): self.strategy1 = utils.create_test_strategy( id=1, uuid=w_utils.generate_uuid(), name=strategy1_name, display_name="Strategy 1", goal_id=self.goal1.id) with freezegun.freeze_time(self.FAKE_OLD_DATE): self.strategy2 = utils.create_test_strategy( id=2, uuid=w_utils.generate_uuid(), name=strategy2_name, display_name="Strategy 2", goal_id=self.goal1.id) with freezegun.freeze_time(self.FAKE_OLDER_DATE): self.strategy3 = utils.create_test_strategy( id=3, uuid=w_utils.generate_uuid(), name=strategy3_name, display_name="Strategy 3", goal_id=self.goal2.id)
def setUp(self): super(TestAuditTemplateObject, self).setUp() self.fake_audit_template = utils.get_test_audit_template() self.fake_goal1 = utils.create_test_goal(id=1, uuid=w_utils.generate_uuid(), name="DUMMY") self.fake_goal2 = utils.create_test_goal(id=2, uuid=w_utils.generate_uuid(), name="BALANCE_LOAD")
def _data_setup(self): with freezegun.freeze_time(self.FAKE_TODAY): self.goal1 = utils.create_test_goal( id=1, uuid=w_utils.generate_uuid(), name="GOAL_1", display_name="Goal 1") with freezegun.freeze_time(self.FAKE_OLD_DATE): self.goal2 = utils.create_test_goal( id=2, uuid=w_utils.generate_uuid(), name="GOAL_2", display_name="Goal 2") with freezegun.freeze_time(self.FAKE_OLDER_DATE): self.goal3 = utils.create_test_goal( id=3, uuid=w_utils.generate_uuid(), name="GOAL_3", display_name="Goal 3")
def _data_setup(self): with freezegun.freeze_time(self.FAKE_TODAY): self.goal1 = utils.create_test_goal( id=1, uuid=w_utils.generate_uuid(), name="GOAL_1", display_name="Goal 1") with freezegun.freeze_time(self.FAKE_OLD_DATE): self.goal2 = utils.create_test_goal( id=2, uuid=w_utils.generate_uuid(), name="GOAL_2", display_name="Goal 2") with freezegun.freeze_time(self.FAKE_OLDER_DATE): self.goal3 = utils.create_test_goal( id=3, uuid=w_utils.generate_uuid(), name="GOAL_3", display_name="Goal 3")
def setUp(self): super(TestActionScheduling, self).setUp() self.goal = db_utils.create_test_goal(name="dummy") self.strategy = db_utils.create_test_strategy(name="dummy") self.audit = db_utils.create_test_audit(uuid=utils.generate_uuid(), strategy_id=self.strategy.id) self.planner = pbase.WeightPlanner( mock.Mock(weights={ 'turn_host_to_acpi_s3_state': 10, 'resize': 20, 'migrate': 30, 'sleep': 40, 'change_nova_service_state': 50, 'nop': 60, 'new_action_type': 70, }, parallelization={ 'turn_host_to_acpi_s3_state': 2, 'resize': 2, 'migrate': 2, 'sleep': 1, 'change_nova_service_state': 1, 'nop': 1, 'new_action_type': 70, }))
def setUp(self): super(TestActionScheduling, self).setUp() self.goal = db_utils.create_test_goal(name="server_consolidation") self.strategy = db_utils.create_test_strategy( name="node_resource_consolidation") self.audit = db_utils.create_test_audit(uuid=utils.generate_uuid(), strategy_id=self.strategy.id) self.planner = pbase.NodeResourceConsolidationPlanner(mock.Mock())
def setUp(self): super(TestActionScheduling, self).setUp() self.goal = db_utils.create_test_goal(name="dummy") self.strategy = db_utils.create_test_strategy(name="dummy") self.audit = db_utils.create_test_audit(uuid=utils.generate_uuid(), strategy_id=self.strategy.id) self.planner = pbase.WorkloadStabilizationPlanner(mock.Mock()) self.nova_helper = nova_helper.NovaHelper(mock.Mock())
def setUp(self): super(TestAuditObject, self).setUp() p_audit_notifications = mock.patch.object( notifications, 'audit', autospec=True) self.m_audit_notifications = p_audit_notifications.start() self.addCleanup(p_audit_notifications.stop) self.m_send_update = self.m_audit_notifications.send_update self.fake_goal = utils.create_test_goal(**self.goal_data)
def test_get_goal_by_uuid(self): efficacy_spec = [{"unit": "%", "name": "dummy", "schema": "Range(min=0, max=100, min_included=True, " "max_included=True, msg=None)", "description": "Dummy indicator"}] created_goal = utils.create_test_goal( efficacy_specification=efficacy_spec) goal = self.dbapi.get_goal_by_uuid(self.context, created_goal['uuid']) self.assertEqual(goal.uuid, created_goal['uuid'])
def test_get_audit_template_list_with_filters(self): goal = utils.create_test_goal(name='DUMMY') audit_template1 = utils.create_test_audit_template( id=1, uuid=w_utils.generate_uuid(), name='My Audit Template 1', description='Description of my audit template 1', goal_id=goal['id']) audit_template2 = utils.create_test_audit_template( id=2, uuid=w_utils.generate_uuid(), name='My Audit Template 2', description='Description of my audit template 2', goal_id=goal['id']) audit_template3 = utils.create_test_audit_template( id=3, uuid=w_utils.generate_uuid(), name='My Audit Template 3', description='Description of my audit template 3', goal_id=goal['id']) self.dbapi.soft_delete_audit_template(audit_template3['uuid']) res = self.dbapi.get_audit_template_list( self.context, filters={'name': 'My Audit Template 1'}) self.assertEqual([audit_template1['id']], [r.id for r in res]) res = self.dbapi.get_audit_template_list( self.context, filters={'name': 'Does not exist'}) self.assertEqual([], [r.id for r in res]) res = self.dbapi.get_audit_template_list( self.context, filters={'goal_name': 'DUMMY'}) self.assertEqual( sorted([audit_template1['id'], audit_template2['id']]), sorted([r.id for r in res])) temp_context = self.context temp_context.show_deleted = True res = self.dbapi.get_audit_template_list( temp_context, filters={'goal_name': 'DUMMY'}) self.assertEqual( sorted([audit_template1['id'], audit_template2['id'], audit_template3['id']]), sorted([r.id for r in res])) res = self.dbapi.get_audit_template_list( self.context, filters={'name': 'My Audit Template 2'}) self.assertEqual([audit_template2['id']], [r.id for r in res])
def test_get_audit_list_with_filters(self): goal = utils.create_test_goal(name='DUMMY') audit1 = utils.create_test_audit( id=1, audit_type=objects.audit.AuditType.ONESHOT.value, uuid=w_utils.generate_uuid(), name='My Audit {0}'.format(1), state=objects.audit.State.ONGOING, goal_id=goal['id']) audit2 = utils.create_test_audit( id=2, audit_type=objects.audit.AuditType.CONTINUOUS.value, uuid=w_utils.generate_uuid(), name='My Audit {0}'.format(2), state=objects.audit.State.PENDING, goal_id=goal['id']) audit3 = utils.create_test_audit( id=3, audit_type=objects.audit.AuditType.CONTINUOUS.value, uuid=w_utils.generate_uuid(), name='My Audit {0}'.format(3), state=objects.audit.State.ONGOING, goal_id=goal['id']) self.dbapi.soft_delete_audit(audit3['uuid']) res = self.dbapi.get_audit_list( self.context, filters={'audit_type': objects.audit.AuditType.ONESHOT.value}) self.assertEqual([audit1['id']], [r.id for r in res]) res = self.dbapi.get_audit_list(self.context, filters={'audit_type': 'bad-type'}) self.assertEqual([], [r.id for r in res]) res = self.dbapi.get_audit_list( self.context, filters={'state': objects.audit.State.ONGOING}) self.assertEqual([audit1['id']], [r.id for r in res]) res = self.dbapi.get_audit_list( self.context, filters={'state': objects.audit.State.PENDING}) self.assertEqual([audit2['id']], [r.id for r in res]) res = self.dbapi.get_audit_list(self.context, filters={'goal_name': 'DUMMY'}) self.assertEqual(sorted([audit1['id'], audit2['id']]), sorted([r.id for r in res])) temp_context = self.context temp_context.show_deleted = True res = self.dbapi.get_audit_list(temp_context, filters={'goal_name': 'DUMMY'}) self.assertEqual(sorted([audit1['id'], audit2['id'], audit3['id']]), sorted([r.id for r in res]))
def test_get_goal_list(self): uuids = [] for i in range(1, 6): goal = utils.create_test_goal(id=i, uuid=w_utils.generate_uuid(), name="GOAL_%s" % i, display_name='My Goal %s' % i) uuids.append(six.text_type(goal['uuid'])) res = self.dbapi.get_goal_list(self.context) res_uuids = [r.uuid for r in res] self.assertEqual(uuids.sort(), res_uuids.sort())
def test_get_goal_list(self): uuids = [] for i in range(1, 4): goal = utils.create_test_goal(id=i, uuid=w_utils.generate_uuid(), name="GOAL_%s" % i, display_name='My Goal %s' % i) uuids.append(six.text_type(goal['uuid'])) goals = self.dbapi.get_goal_list(self.context) goal_uuids = [g.uuid for g in goals] self.assertEqual(sorted(uuids), sorted(goal_uuids))
def test_get_goal_list_with_filters(self): goal1 = utils.create_test_goal( id=1, uuid=w_utils.generate_uuid(), name="GOAL_1", display_name='Goal 1', ) goal2 = utils.create_test_goal( id=2, uuid=w_utils.generate_uuid(), name="GOAL_2", display_name='Goal 2', ) goal3 = utils.create_test_goal( id=3, uuid=w_utils.generate_uuid(), name="GOAL_3", display_name='Goal 3', ) self.dbapi.soft_delete_goal(goal3['uuid']) res = self.dbapi.get_goal_list( self.context, filters={'display_name': 'Goal 1'}) self.assertEqual([goal1['uuid']], [r.uuid for r in res]) res = self.dbapi.get_goal_list( self.context, filters={'display_name': 'Goal 3'}) self.assertEqual([], [r.uuid for r in res]) res = self.dbapi.get_goal_list( self.context, filters={'name': 'GOAL_1'}) self.assertEqual([goal1['uuid']], [r.uuid for r in res]) res = self.dbapi.get_goal_list( self.context, filters={'display_name': 'Goal 2'}) self.assertEqual([goal2['uuid']], [r.uuid for r in res]) res = self.dbapi.get_goal_list( self.context, filters={'uuid': goal3['uuid']}) self.assertEqual([], [r.uuid for r in res])
def test_get_goal_list(self): uuids = [] for i in range(1, 6): goal = utils.create_test_goal( id=i, uuid=w_utils.generate_uuid(), name="GOAL_%s" % i, display_name='My Goal %s' % i) uuids.append(six.text_type(goal['uuid'])) res = self.dbapi.get_goal_list(self.context) res_uuids = [r.uuid for r in res] self.assertEqual(uuids.sort(), res_uuids.sort())
def setUp(self): super(TestCreateDeleteAuditObject, self).setUp() p_audit_notifications = mock.patch.object( notifications, 'audit', autospec=True) self.m_audit_notifications = p_audit_notifications.start() self.addCleanup(p_audit_notifications.stop) self.m_send_update = self.m_audit_notifications.send_update self.goal_id = 1 self.goal = utils.create_test_goal(id=self.goal_id, name="DUMMY") self.fake_audit = utils.get_test_audit( goal_id=self.goal_id, created_at=datetime.datetime.utcnow())
def test_create(self, mock_create_audit_template): goal = utils.create_test_goal() self.fake_audit_template['goal_id'] = goal.id mock_create_audit_template.return_value = self.fake_audit_template audit_template = objects.AuditTemplate(self.context, **self.fake_audit_template) audit_template.create() expected_audit_template = self.fake_audit_template.copy() expected_audit_template['created_at'] = expected_audit_template[ 'created_at'].replace(tzinfo=iso8601.UTC) mock_create_audit_template.assert_called_once_with( expected_audit_template) self.assertEqual(self.context, audit_template._context)
def setUp(self): super(TestAuditObjectSendNotifications, self).setUp() goal_id = 1 self.fake_goal = utils.create_test_goal(id=goal_id, name="DUMMY") self.fake_strategy = utils.create_test_strategy( id=goal_id, name="DUMMY") self.fake_audit = utils.get_test_audit( goal_id=goal_id, goal=utils.get_test_goal(id=goal_id), strategy_id=self.fake_strategy.id, strategy=self.fake_strategy) p_get_notifier = mock.patch.object(rpc, 'get_notifier') self.m_get_notifier = p_get_notifier.start() self.m_get_notifier.return_value = mock.Mock(name='m_notifier') self.m_notifier = self.m_get_notifier.return_value self.addCleanup(p_get_notifier.stop)
def setUp(self): super(TestStrategyObject, self).setUp() self.fake_goal = utils.create_test_goal(id=self.goal_id)
def test_create_goal_already_exists(self): goal_uuid = w_utils.generate_uuid() utils.create_test_goal(uuid=goal_uuid) self.assertRaises(exception.GoalAlreadyExists, utils.create_test_goal, uuid=goal_uuid)
def test_destroy_goal(self): goal = utils.create_test_goal() self.dbapi.destroy_goal(goal['uuid']) self.assertRaises(exception.GoalNotFound, self.dbapi.get_goal_by_uuid, self.context, goal['uuid'])
def test_update_goal_id(self): goal = utils.create_test_goal() self.assertRaises(exception.Invalid, self.dbapi.update_goal, goal['uuid'], {'uuid': 'NEW_GOAL'})
def test_update_goal(self): goal = utils.create_test_goal() res = self.dbapi.update_goal(goal['uuid'], {'display_name': 'updated-model'}) self.assertEqual('updated-model', res.display_name)
def setUp(self): super(TestAuditTemplateObject, self).setUp() self.fake_goal = utils.create_test_goal(**self.goal_data)
def setUp(self): super(TestCreateDeleteStrategyObject, self).setUp() self.fake_goal = utils.create_test_goal() self.fake_strategy = utils.get_test_strategy(goal_id=self.fake_goal.id)