python/tests/gensec: make it possible to add knownfail tests for gensec.update()

BUG: https://bugzilla.samba.org/show_bug.cgi?id=14106

Signed-off-by: Stefan Metzmacher <metze@samba.org>
Reviewed-by: Andreas Schneider <asn@samba.org>

diff --git a/python/samba/tests/gensec.py b/python/samba/tests/gensec.py
index b5ce51de7560c25fa3d414c2fe1b184816d40d56..c9056ef9681c32751389a581066959672573d691 100644 (file)
--- a/python/samba/tests/gensec.py
+++ b/python/samba/tests/gensec.py
@@ -79,10 +79,16 @@ class GensecTests(samba.tests.TestCase):
         while True:
             if not client_finished:
                 print("running client gensec_update")
-                (client_finished, client_to_server) = self.gensec_client.update(server_to_client)
+                try:
+                    (client_finished, client_to_server) = self.gensec_client.update(server_to_client)
+                except samba.NTSTATUSError as nt:
+                    raise AssertionError(nt)
             if not server_finished:
                 print("running server gensec_update")
-                (server_finished, server_to_client) = self.gensec_server.update(client_to_server)
+                try:
+                    (server_finished, server_to_client) = self.gensec_server.update(client_to_server)
+                except samba.NTSTATUSError as nt:
+                    raise AssertionError(nt)
 
             if client_finished and server_finished:
                 break