ndb: fix source view counting

svinota · Dec 24, 2023 · f1c32e5 · f1c32e5
1 parent 020b793
commit f1c32e5
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/pyroute2/ndb/source.py b/pyroute2/ndb/source.py
@@ -180,6 +180,12 @@ def __init__(self, ndb, **spec):
         self.ndb.task_manager.db_add_nl_source(self.target, self.kind, spec)
         self.load_sql()
 
+    @classmethod
+    def _count(cls, view):
+        return view.ndb.task_manager.db_fetchone(
+            "SELECT count(*) FROM %s" % view.table
+        )
+
     @property
     def must_restart(self):
         if self.max_errors < 0 or self.errors_counter <= self.max_errors:

diff --git a/tests/test_linux/test_ndb/test_sources.py b/tests/test_linux/test_ndb/test_sources.py
@@ -38,6 +38,7 @@ def test_multiple_sources(context):
         assert len(list(ndb.neighbours.dump()))
         assert len(list(ndb.addresses.dump()))
         assert len(list(ndb.routes.dump()))
+        assert len(ndb.sources) == len(sources)
     # here NDB() gets closed
     #