def test_env_get(self):
env1 = db.env_create("name", "STATUS42", "descr", {}, {}, {}, [])
env2 = db.env_create("name2", "STATUS42", "descr", {}, {}, {}, [])
self.assertEqual(env1, db.env_get(env1["uuid"]))
self.assertEqual(env1, db.env_get(env1["name"]))
self.assertEqual(env2, db.env_get(env2["uuid"]))
self.assertEqual(env2, db.env_get(env2["name"]))
def test_env_get(self):
env1 = db.env_create("name", "STATUS42", "descr", {}, {}, {}, [])
env2 = db.env_create("name2", "STATUS42", "descr", {}, {}, {}, [])
self.assertEqual(env1, db.env_get(env1["uuid"]))
self.assertEqual(env1, db.env_get(env1["name"]))
self.assertEqual(env2, db.env_get(env2["uuid"]))
self.assertEqual(env2, db.env_get(env2["name"]))
def test_evn_set_status(self):
env = db.env_create("name", "status", "descr", {}, {}, {}, [])
self.assertRaises(exceptions.DBConflict, db.env_set_status,
env["uuid"], "wrong_old_status", "new_status")
env = db.env_get(env["uuid"])
self.assertEqual("status", env["status"])
self.assertTrue(db.env_set_status(env["uuid"], "status", "new_status"))
env = db.env_get(env["uuid"])
self.assertEqual("new_status", env["status"])
def test_evn_set_status(self):
env = db.env_create("name", "status", "descr", {}, {}, {}, [])
self.assertRaises(
exceptions.DBConflict,
db.env_set_status, env["uuid"], "wrong_old_status", "new_status")
env = db.env_get(env["uuid"])
self.assertEqual("status", env["status"])
self.assertTrue(
db.env_set_status(env["uuid"], "status", "new_status"))
env = db.env_get(env["uuid"])
self.assertEqual("new_status", env["status"])
def get(cls, uuid_or_name):
"""Get the instance of EnvManager by uuid or name.
:param uuid_or_name: Returns record that has uuid or name equal to it.
:returns: Instance of rally.env.env_mgr.EnvManager
"""
return cls(db.env_get(uuid_or_name))
def get(cls, uuid_or_name):
"""Get the instance of EnvManager by uuid or name.
:param uuid_or_name: Returns record that has uuid or name equal to it.
:returns: Instance of rally.env.env_mgr.EnvManager
"""
return cls(db.env_get(uuid_or_name))
def test_env_update(self):
env = db.env_create("name", "status", "descr", {}, {}, {}, [])
self.assertTrue(db.env_update(env["uuid"]))
self.assertTrue(
db.env_update(env["uuid"], "another_descr", {"e": 123}, {"c": 1}))
env = db.env_get(env["uuid"])
self.assertEqual("another_descr", env["description"])
self.assertEqual({"e": 123}, env["extras"])
self.assertEqual({"c": 1}, env["config"])
def test_env_update(self):
env = db.env_create("name", "status", "descr", {}, {}, {}, [])
self.assertTrue(db.env_update(env["uuid"]))
self.assertTrue(
db.env_update(env["uuid"], "another_descr", {"e": 123}, {"c": 1}))
env = db.env_get(env["uuid"])
self.assertEqual("another_descr", env["description"])
self.assertEqual({"e": 123}, env["extras"])
self.assertEqual({"c": 1}, env["config"])
def to_dict(self):
db_task = self.task
deployment_name = db.env_get(self.task["env_uuid"])["name"]
db_task["deployment_uuid"] = db_task["env_uuid"]
db_task["deployment_name"] = deployment_name
self._serialize_dt(db_task)
for subtask in db_task.get("subtasks", []):
self._serialize_dt(subtask)
for workload in subtask["workloads"]:
self._serialize_dt(workload)
return db_task
def to_dict(self):
db_task = self.task
if self.task.get("env_uuid"):
env_name = db.env_get(self.task["env_uuid"])["name"]
db_task["env_name"] = env_name
db_task["deployment_name"] = env_name
db_task["deployment_uuid"] = db_task["env_uuid"]
self._serialize_dt(db_task)
for subtask in db_task.get("subtasks", []):
self._serialize_dt(subtask)
for workload in subtask["workloads"]:
self._serialize_dt(workload)
return db_task
def test_env_create(self):
env = db.env_create("name", "status", "descr", {"extra": "test"},
{"conf": "c1"}, {"spec": "spec"}, [])
self.assertIsInstance(env, dict)
self.assertIsNotNone(env["uuid"])
self.assertEqual(env, db.env_get(env["uuid"]))
self.assertEqual("name", env["name"])
self.assertEqual("status", env["status"])
self.assertEqual("descr", env["description"])
self.assertEqual({"conf": "c1"}, env["config"])
self.assertEqual({"extra": "test"}, env["extras"])
self.assertEqual({"spec": "spec"}, env["spec"])
def test_env_create(self):
env = db.env_create(
"name", "status", "descr",
{"extra": "test"}, {"conf": "c1"}, {"spec": "spec"}, [])
self.assertIsInstance(env, dict)
self.assertIsNotNone(env["uuid"])
self.assertEqual(env, db.env_get(env["uuid"]))
self.assertEqual("name", env["name"])
self.assertEqual("status", env["status"])
self.assertEqual("descr", env["description"])
self.assertEqual({"conf": "c1"}, env["config"])
self.assertEqual({"extra": "test"}, env["extras"])
self.assertEqual({"spec": "spec"}, env["spec"])
def data(self):
"""Returns full information about env including platforms."""
self._env = db.env_get(self.uuid)
return {
"id": self._env["id"],
"uuid": self._env["uuid"],
"created_at": self._env["created_at"],
"updated_at": self._env["updated_at"],
"name": self._env["name"],
"description": self._env["description"],
"status": self._env["status"],
"spec": copy.deepcopy(self._env["spec"]),
"extras": copy.deepcopy(self._env["extras"]),
"platforms": db.platforms_list(self.uuid)
}
def data(self):
"""Returns full information about env including platforms."""
self._env = db.env_get(self.uuid)
self._env["platforms"] = db.platforms_list(self.uuid)
return self.cached_data
def test_env_rename(self):
env = db.env_create("name", "status", "descr", {"extra": "test"},
{"spec": "spec"}, {}, [])
self.assertTrue(db.env_rename(env["uuid"], env["name"], "name2"))
self.assertEqual("name2", db.env_get(env["uuid"])["name"])
def test_env_rename(self):
env = db.env_create("name", "status", "descr",
{"extra": "test"}, {"spec": "spec"}, {}, [])
self.assertTrue(db.env_rename(env["uuid"], env["name"], "name2"))
self.assertEqual("name2", db.env_get(env["uuid"])["name"])
def data(self):
"""Returns full information about env including platforms."""
self._env = db.env_get(self.uuid)
self._env["platforms"] = db.platforms_list(self.uuid)
return self.cached_data
