Fix tests for libexpat >=2.6.0 (or with CVE-2023-52425 fixes backport…

…ed) (cea-hpc#556) * tests/TreeGatewayTest.py: Close the XML parser when done * tests/TreeGatewayTest.py: Call flush for Expat >=2.6.0
thiell · Sep 9, 2024 · e49f842 · e49f842
1 parent e58d8bc
commit e49f842
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/TreeGatewayTest.py b/tests/TreeGatewayTest.py
@@ -131,9 +131,12 @@ def _recvxml(self):
         while not self.xml_reader.msg_available():
             xml_msg = self.gateway.recv()
             if len(xml_msg) == 0:
-                return None
+                self.parser.close()
+                break
             self.assertTrue(type(xml_msg) is bytes)
             self.parser.feed(xml_msg)
+            if hasattr(self.parser, 'flush'):  # >=3.13 and backports
+                self.parser.flush()
 
         return self.xml_reader.pop_msg()