Index: client/tests/kvm/kvm_scheduler.py |
diff --git a/client/tests/kvm/kvm_scheduler.py b/client/tests/kvm/kvm_scheduler.py |
index f1adb39ab9f781be36bccfaf66d9e19787ea0625..95282e447785ecf3361d261ed571ae61b78a282b 100644 |
--- a/client/tests/kvm/kvm_scheduler.py |
+++ b/client/tests/kvm/kvm_scheduler.py |
@@ -74,13 +74,13 @@ class scheduler: |
# The scheduler wants this worker to free its used resources |
elif cmd[0] == "cleanup": |
env_filename = os.path.join(self.bindir, self_dict["env"]) |
- env = kvm_utils.load_env(env_filename, {}) |
+ env = kvm_utils.Env(env_filename) |
for obj in env.values(): |
if isinstance(obj, kvm_vm.VM): |
obj.destroy() |
- elif isinstance(obj, kvm_subprocess.kvm_spawn): |
+ elif isinstance(obj, kvm_subprocess.Spawn): |
obj.close() |
- kvm_utils.dump_env(env, env_filename) |
+ env.save() |
w.write("cleanup_done\n") |
w.write("ready\n") |