PostgresNode::source_walsender is updated [revision]

Author: dmitry-lipetsk

testgres/node.py

@@ -431,9 +431,11 @@ def source_walsender(self):
             where application_name = %s
         """
 
-        if not self.master:
+        if self.master is None:
             raise TestgresException("Node doesn't have a master")
 
+        assert type(self.master) == PostgresNode
+
         # master should be on the same host
         assert self.master.host == self.host
 

tests/test_testgres_common.py

@@ -348,6 +348,7 @@ def LOCAL__check_auxiliary_pids__multiple_attempts(
                 assert (con.pid > 0)
 
             with master.replicate().start() as replica:
+                assert type(replica) == PostgresNode
 
                 # test __str__ method
                 str(master.child_processes[0])