fix:tests

Author: cunla

scheduler/tests/test_mgmt_cmds.py

@@ -233,26 +233,28 @@ def test_import__should_schedule_job(self):
         call_command("import", filename=self.tmpfile.name)
         # assert
         self.assertEqual(1, Task.objects.filter(task_type=TaskType.ONCE).count())
+        self.assertEqual(1, Task.objects.filter(task_type=TaskType.REPEATABLE).count())
         db_job = Task.objects.filter(task_type=TaskType.ONCE).first()
         attrs = ["name", "queue", "callable", "enabled", "timeout"]
         for attr in attrs:
             self.assertEqual(getattr(jobs[0], attr), getattr(db_job, attr))
 
     def test_import__should_schedule_job_yaml(self):
-        jobs = list()
-        jobs.append(task_factory(ScheduledTask, enabled=True, instance_only=True))
-        jobs.append(task_factory(RepeatableTask, enabled=True, instance_only=True))
-        res = yaml.dump([j.to_dict() for j in jobs], default_flow_style=False)
+        tasks = list()
+        tasks.append(task_factory(ScheduledTask, enabled=True, instance_only=True))
+        tasks.append(task_factory(RepeatableTask, enabled=True, instance_only=True))
+        res = yaml.dump([j.to_dict() for j in tasks], default_flow_style=False)
         self.tmpfile.write(res)
         self.tmpfile.flush()
         # act
         call_command("import", filename=self.tmpfile.name, format="yaml")
         # assert
         self.assertEqual(1, Task.objects.filter(task_type=TaskType.ONCE).count())
-        db_job = Task.objects.filter(task_type=TaskType.ONCE).objects.first()
+        self.assertEqual(1, Task.objects.filter(task_type=TaskType.REPEATABLE).count())
+        db_job = Task.objects.filter(task_type=TaskType.ONCE).first()
         attrs = ["name", "queue", "callable", "enabled", "timeout"]
         for attr in attrs:
-            self.assertEqual(getattr(jobs[0], attr), getattr(db_job, attr))
+            self.assertEqual(getattr(tasks[0], attr), getattr(db_job, attr))
 
     def test_import__should_schedule_job_yaml_without_yaml_lib(self):
         jobs = list()
@@ -295,10 +297,10 @@ def test_import__should_schedule_job_reset(self):
             self.assertEqual(getattr(jobs[1], attr), getattr(db_job, attr))
 
     def test_import__should_schedule_job_update_existing(self):
-        jobs = list()
-        task_factory(ScheduledTask, enabled=True)
-        jobs.append(task_factory(ScheduledTask, enabled=True))
-        res = json.dumps([j.to_dict() for j in jobs])
+        tasks = list()
+        tasks.append(task_factory(ScheduledTask, enabled=True))
+        tasks.append(task_factory(ScheduledTask, enabled=True))
+        res = json.dumps([j.to_dict() for j in tasks])
         self.tmpfile.write(res)
         self.tmpfile.flush()
         # act
@@ -309,17 +311,16 @@ def test_import__should_schedule_job_update_existing(self):
         )
         # assert
         self.assertEqual(2, Task.objects.filter(task_type=TaskType.ONCE).count())
-        db_job = Task.objects.filter(task_type=TaskType.ONCE).get(name=jobs[0].name)
-        self.assertNotEqual(jobs[0].id, db_job.id)
+        db_job = Task.objects.filter(task_type=TaskType.ONCE).get(name=tasks[0].name)
         attrs = ["name", "queue", "callable", "enabled", "timeout"]
         for attr in attrs:
-            self.assertEqual(getattr(jobs[0], attr), getattr(db_job, attr))
+            self.assertEqual(getattr(tasks[0], attr), getattr(db_job, attr))
 
     def test_import__should_schedule_job_without_update_existing(self):
-        jobs = list()
-        task_factory(ScheduledTask, enabled=True)
-        jobs.append(task_factory(ScheduledTask, enabled=True))
-        res = json.dumps([j.to_dict() for j in jobs])
+        tasks = list()
+        tasks.append(task_factory(ScheduledTask, enabled=True))
+        tasks.append(task_factory(ScheduledTask, enabled=True))
+        res = json.dumps([j.to_dict() for j in tasks])
         self.tmpfile.write(res)
         self.tmpfile.flush()
         # act
@@ -329,7 +330,7 @@ def test_import__should_schedule_job_without_update_existing(self):
         )
         # assert
         self.assertEqual(2, Task.objects.filter(task_type=TaskType.ONCE).count())
-        db_job = Task.objects.get(name=jobs[0].name)
+        db_job = Task.objects.get(name=tasks[0].name)
         attrs = ["id", "name", "queue", "callable", "enabled", "timeout"]
         for attr in attrs:
-            self.assertEqual(getattr(jobs[0], attr), getattr(db_job, attr))
+            self.assertEqual(getattr(tasks[0], attr), getattr(db_job, attr))