diff options
Diffstat (limited to 'test.py')
| -rw-r--r-- | test.py | 16 |
1 files changed, 8 insertions, 8 deletions
@@ -19,7 +19,7 @@ def reset_time_mock(): class TestFemtoQueue(unittest.TestCase): def test_basic(self): dir = mkdtemp() - q = FemtoQueue(data_dir = dir, node_name = "node1") + q = FemtoQueue(data_dir = dir, node_id = "node1") # Add a JSON task some_data = json.dumps({ "foo": "bar" }) @@ -41,7 +41,7 @@ class TestFemtoQueue(unittest.TestCase): def test_aborted(self): dir = mkdtemp() - q = FemtoQueue(data_dir = dir, node_name = "node1") + q = FemtoQueue(data_dir = dir, node_id = "node1") # Add a JSON task some_data = json.dumps({ "foo": "bar" }) @@ -55,7 +55,7 @@ class TestFemtoQueue(unittest.TestCase): self.assertEqual(parsed_data["foo"], "bar") # Simulate a fault and start over - q = FemtoQueue(data_dir = dir, node_name = "node1") + q = FemtoQueue(data_dir = dir, node_id = "node1") # The task should still be assigned to this node task = q.pop() @@ -77,14 +77,14 @@ class TestFemtoQueue(unittest.TestCase): set_time_mock(0) # Node1 creates and claims the task - q1 = FemtoQueue(data_dir=dir, node_name="node1", timeout_stale_ms=timeout_stale_ms) + q1 = FemtoQueue(data_dir=dir, node_id="node1", timeout_stale_ms=timeout_stale_ms) q1.push(b"stuck") task = q1.pop() self.assertIsNotNone(task) task = cast(FemtoTask, task) # Assert that node2 can not see the task - q2 = FemtoQueue(data_dir=dir, node_name="node2", timeout_stale_ms=timeout_stale_ms) + q2 = FemtoQueue(data_dir=dir, node_id="node2", timeout_stale_ms=timeout_stale_ms) non_existant_task = q2.pop() self.assertIsNone(non_existant_task) @@ -101,7 +101,7 @@ class TestFemtoQueue(unittest.TestCase): def test_mark_task_failed(self): dir = mkdtemp() - q = FemtoQueue(data_dir=dir, node_name="node1") + q = FemtoQueue(data_dir=dir, node_id="node1") # Push and pop a task q.push(b"will fail") @@ -121,7 +121,7 @@ class TestFemtoQueue(unittest.TestCase): def test_mark_task_done(self): dir = mkdtemp() - q = FemtoQueue(data_dir=dir, node_name="node1") + q = FemtoQueue(data_dir=dir, node_id="node1") # Push and pop a task q.push(b"complete me") @@ -141,7 +141,7 @@ class TestFemtoQueue(unittest.TestCase): def test_fifo_order(self): dir = mkdtemp() - q = FemtoQueue(data_dir=dir, node_name="node1") + q = FemtoQueue(data_dir=dir, node_id="node1") num_tasks = 100 |