From b56495cd0d4f6bb310ca4e85a32df66d4813fea6 Mon Sep 17 00:00:00 2001 From: Jan Tuomi Date: Thu, 15 May 2025 09:54:08 +0300 Subject: Rename node_name to node_id --- test.py | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) (limited to 'test.py') diff --git a/test.py b/test.py index e1fc086..30d9a95 100644 --- a/test.py +++ b/test.py @@ -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 -- cgit v1.3