fix: insert new node providers into node_provider_lookup instead of s…

…ubscribers table
aviate-labs · Nov 14, 2023 · 6d3d7ce · 6d3d7ce
1 parent b0d3310
commit 6d3d7ce
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git a/node_monitor/node_monitor.py b/node_monitor/node_monitor.py
@@ -206,9 +206,7 @@ def update_node_provider_lookup_if_new(
 
         if node_providers_new:
             for node_provider in node_providers_new:
-                params = (node_provider.principal_id, False, False, 
-                          False, node_provider.display_name, False)
-                self.node_provider_db.execute_write(INSERT_SUBSCRIBER, params)
+                self.node_provider_db.insert_node_provider(node_provider)
 
 
 

diff --git a/tests/test_node_monitor.py b/tests/test_node_monitor.py
@@ -232,7 +232,7 @@ def test_no_new_node_provider():
                      mock_slack_bot, mock_telegram_bot)
 
     nm.update_node_provider_lookup_if_new(cached['node_provider_control'])
-    assert mock_node_provider_db.execute_write.call_count == 0
+    assert mock_node_provider_db.insert_node_provider.call_count == 0
 
 
 
@@ -244,5 +244,5 @@ def test_one_new_node_provider():
                      mock_slack_bot, mock_telegram_bot)
 
     nm.update_node_provider_lookup_if_new(cached['new_node_providers'])
-    assert mock_node_provider_db.execute_write.call_count == 2
+    assert mock_node_provider_db.insert_node_provider.call_count == 2