Rename node_name to node_id

1 files changed, 8 insertions, 8 deletions

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