traffic: assign context domain at start

Signed-off-by: Douglas Bagnall <douglas.bagnall@catalyst.net.nz>
Reviewed-by: Andrew Bartlett <abartlet@samba.org>

diff --git a/python/samba/emulate/traffic.py b/python/samba/emulate/traffic.py
index 1d4316586e6bae5efbc1c2a9c9f44c522a1ae822..dadc6c140b700b0599f363f3dc745727c38082df 100644 (file)
--- a/python/samba/emulate/traffic.py
+++ b/python/samba/emulate/traffic.py
@@ -359,9 +359,8 @@ class ReplayContext(object):
                  statsdir=None,
                  ou=None,
                  base_dn=None,
-                 domain=None,
+                 domain=os.environ.get("DOMAIN"),
                  domain_sid=None):
-
         self.server                   = server
         self.netlogon_connection      = None
         self.creds                    = creds
@@ -459,11 +458,6 @@ class ReplayContext(object):
         self.lp.set("state directory", self.tempdir)
         self.lp.set("tls verify peer", "no_check")
 
-        # If the domain was not specified, check for the environment
-        # variable.
-        if self.domain is None:
-            self.domain = os.environ["DOMAIN"]
-
         self.remoteAddress = "/root/ncalrpc_as_system"
         self.samlogon_dn   = ("cn=%s,%s" %
                               (self.netbios_name, self.ou))