diff options
Diffstat (limited to 'source4/scripting/python')
| -rw-r--r-- | source4/scripting/python/samba/getopt.py | 10 | ||||
| -rw-r--r-- | source4/scripting/python/samba/tests/__init__.py | 15 | ||||
| -rw-r--r-- | source4/scripting/python/samba/tests/dcerpc/registry.py | 9 | 
3 files changed, 20 insertions, 14 deletions
| diff --git a/source4/scripting/python/samba/getopt.py b/source4/scripting/python/samba/getopt.py index 82cb004b62..7ec684a9d6 100644 --- a/source4/scripting/python/samba/getopt.py +++ b/source4/scripting/python/samba/getopt.py @@ -35,12 +35,14 @@ class SambaOptions(optparse.OptionGroup):          self._configfile = arg      def get_loadparm(self): -        import param +        import os, param          lp = param.LoadParm() -        if self._configfile is None: -            lp.load_default() -        else: +        if self._configfile is not None:              lp.load(self._configfile) +        elif os.getenv("SMB_CONF_PATH") is not None: +            lp.load(os.getenv("SMB_CONF_PATH")) +        else: +            lp.load_default()          return lp  class VersionOptions(optparse.OptionGroup): diff --git a/source4/scripting/python/samba/tests/__init__.py b/source4/scripting/python/samba/tests/__init__.py index c8673d3fae..e29b4a87d5 100644 --- a/source4/scripting/python/samba/tests/__init__.py +++ b/source4/scripting/python/samba/tests/__init__.py @@ -85,8 +85,13 @@ class LdbExtensionTests(TestCaseInTempDir):              os.unlink(path) -def get_loadparm(): -    import param -    lp = param.LoadParm() -    lp.load(os.getenv("SMB_CONF_PATH")) -    return lp +cmdline_loadparm = None +cmdline_credentials = None + +class RpcInterfaceTestCase(unittest.TestCase): +    def get_loadparm(self): +        assert cmdline_loadparm is not None +        return cmdline_loadparm + +    def get_credentials(self): +        return cmdline_credentials diff --git a/source4/scripting/python/samba/tests/dcerpc/registry.py b/source4/scripting/python/samba/tests/dcerpc/registry.py index 147acc5098..05ac7c66ee 100644 --- a/source4/scripting/python/samba/tests/dcerpc/registry.py +++ b/source4/scripting/python/samba/tests/dcerpc/registry.py @@ -18,14 +18,13 @@  #  import winreg -from param import LoadParm  import unittest -from samba.tests import get_loadparm +from samba.tests import RpcInterfaceTestCase -class WinregTests(unittest.TestCase): +class WinregTests(RpcInterfaceTestCase):      def setUp(self): -        lp_ctx = get_loadparm() -        self.conn = winreg.winreg("ncalrpc:", lp_ctx) +        lp = self.get_loadparm() +        self.conn = winreg.winreg("ncalrpc:", lp, self.get_credentials())      def get_hklm(self):          return self.conn.OpenHKLM(None,  | 
