def test_profiler_init(self, users_credentials, expected_call_count, mock_profiler_get, mock_profiler_init): for user, credential in users_credentials: self.context.update({user: {"credential": credential}}) base_scenario.OpenStackScenario(self.context) self.assertEqual(expected_call_count, mock_profiler_init.call_count) self.assertEqual([mock.call()] * expected_call_count, mock_profiler_get.call_args_list)
def test_init_user_context(self): context = {"foo": "bar", "user": {"endpoint": mock.Mock()}} scenario = base_scenario.OpenStackScenario(context) self.assertEqual(context, scenario.context) self.osclients.mock.assert_called_once_with( context["user"]["endpoint"]) self.assertRaises(ValueError, base_scenario.OpenStackScenario, context, clients="foobar")
def test_init_user_context(self): self.context["user"] = {"credential": mock.Mock()} scenario = base_scenario.OpenStackScenario(self.context) self.assertEqual(self.context, scenario.context) self.osclients.mock.assert_called_once_with( self.context["user"]["credential"], {}) self.assertRaises(ValueError, base_scenario.OpenStackScenario, self.context, clients="foobar")
def test_init_user_context(self): user = {"credential": mock.Mock(), "tenant_id": "foo"} self.context["users"] = [user] self.context["tenants"] = {"foo": {"name": "bar"}} self.context["user_choice_method"] = "random" scenario = base_scenario.OpenStackScenario(self.context) self.assertEqual(user, scenario.context["user"]) self.assertEqual(self.context["tenants"]["foo"], scenario.context["tenant"]) self.osclients.mock.assert_called_once_with(user["credential"], {})
def test__choose_user_random(self): users = [{"credential": mock.Mock(), "tenant_id": "foo"} for _ in range(5)] self.context["users"] = users self.context["tenants"] = {"foo": {"name": "bar"}, "baz": {"name": "spam"}} self.context["user_choice_method"] = "random" scenario = base_scenario.OpenStackScenario() scenario._choose_user(self.context) self.assertIn("user", self.context) self.assertIn(self.context["user"], self.context["users"]) self.assertIn("tenant", self.context) tenant_id = self.context["user"]["tenant_id"] self.assertEqual(self.context["tenants"][tenant_id], self.context["tenant"])
def test__choose_user_round_robin(self, iteration, expected_user_id, expected_tenant_id): self.context["iteration"] = iteration self.context["user_choice_method"] = "round_robin" self.context["users"] = [] self.context["tenants"] = {} for tid in ("foo", "bar"): users = [{"id": str(i), "tenant_id": tid} for i in range(2)] self.context["users"] += users self.context["tenants"][tid] = {"name": tid, "users": users} scenario = base_scenario.OpenStackScenario() scenario._choose_user(self.context) self.assertIn("user", self.context) self.assertIn(self.context["user"], self.context["users"]) self.assertEqual(expected_user_id, self.context["user"]["id"]) self.assertIn("tenant", self.context) tenant_id = self.context["user"]["tenant_id"] self.assertEqual(self.context["tenants"][tenant_id], self.context["tenant"]) self.assertEqual(expected_tenant_id, tenant_id)
def test_init_user_clients(self): scenario = base_scenario.OpenStackScenario(self.context, clients="foobar") self.assertEqual(self.context, scenario.context) self.assertEqual("foobar", scenario._clients)
def test_init(self): scenario = base_scenario.OpenStackScenario(self.context) self.assertEqual(self.context, scenario.context)
def test_init_clients(self): scenario = base_scenario.OpenStackScenario(self.context, admin_clients="spam", clients="ham") self.assertEqual("spam", scenario._admin_clients) self.assertEqual("ham", scenario._clients)
def test_init(self): context = {"foo": "bar"} scenario = base_scenario.OpenStackScenario(context) self.assertEqual(context, scenario.context)