tests/auth_log: Simplify isRemote()

Signed-off-by: Joseph Sutton <josephsutton@catalyst.net.nz>
Reviewed-by: Andrew Bartlett <abartlet@samba.org>

diff --git a/python/samba/tests/auth_log_base.py b/python/samba/tests/auth_log_base.py
index 2f9b13e4f445fa3bc08b4b6b6e1473699a211aac..ebe9c3a124ab310267160db8bc6753df1e55a4eb 100644 (file)
--- a/python/samba/tests/auth_log_base.py
+++ b/python/samba/tests/auth_log_base.py
@@ -98,11 +98,13 @@ class AuthLogTestBase(samba.tests.TestCase):
             if self.remoteAddress is None:
                 return True
 
-            remote = None
-            if message["type"] == "Authorization":
-                remote = message["Authorization"]["remoteAddress"]
-            elif message["type"] == "Authentication":
-                remote = message["Authentication"]["remoteAddress"]
+            supported_types = {
+                "Authentication",
+                "Authorization",
+            }
+            message_type = message["type"]
+            if message_type in supported_types:
+                remote = message[message_type]["remoteAddress"]
             else:
                 return False