]> git.ipfire.org Git - thirdparty/samba.git/commitdiff
tests: Factor out prep_creds()
authorVolker Lendecke <vl@samba.org>
Tue, 17 Nov 2020 15:11:11 +0000 (16:11 +0100)
committerJeremy Allison <jra@samba.org>
Thu, 19 Nov 2020 22:56:40 +0000 (22:56 +0000)
3 times the same code can be put together

Signed-off-by: Volker Lendecke <vl@samba.org>
Reviewed-by: Jeremy Allison <jra@samba.org>
python/samba/tests/libsmb.py

index 81d4e482644a60f3ccdfc6dd2d90d36331967f98..0f0d27c9b32e3ee698fa3ce0e5e84276c3539c80 100644 (file)
@@ -51,8 +51,7 @@ class LibsmbTestCase(samba.tests.TestCase):
             except Exception:
                 self.exc = sys.exc_info()
 
-    def test_OpenClose(self):
-
+    def prep_creds(self):
         lp = s3param.get_context()
         lp.load(os.getenv("SMB_CONF_PATH"))
 
@@ -61,6 +60,11 @@ class LibsmbTestCase(samba.tests.TestCase):
         creds.set_username(os.getenv("USERNAME"))
         creds.set_password(os.getenv("PASSWORD"))
 
+        return (lp,creds)
+
+    def test_OpenClose(self):
+        (lp,creds) = self.prep_creds()
+
         c = libsmb.Conn(os.getenv("SERVER_IP"), "tmp",
                         lp, creds, multi_threaded=True,
                         force_smb1=True)
@@ -82,13 +86,7 @@ class LibsmbTestCase(samba.tests.TestCase):
     def test_SMB3EncryptionRequired(self):
         test_dir = 'testing_%d' % random.randint(0, 0xFFFF)
 
-        lp = s3param.get_context()
-        lp.load(os.getenv("SMB_CONF_PATH"))
-
-        creds = credentials.Credentials()
-        creds.guess(lp)
-        creds.set_username(os.getenv("USERNAME"))
-        creds.set_password(os.getenv("PASSWORD"))
+        (lp,creds) = self.prep_creds()
         creds.set_smb_encryption(SMB_ENCRYPTION_REQUIRED)
 
         c = libsmb.Conn(os.getenv("SERVER_IP"), "tmp",
@@ -100,13 +98,7 @@ class LibsmbTestCase(samba.tests.TestCase):
     def test_SMB1EncryptionRequired(self):
         test_dir = 'testing_%d' % random.randint(0, 0xFFFF)
 
-        lp = s3param.get_context()
-        lp.load(os.getenv("SMB_CONF_PATH"))
-
-        creds = credentials.Credentials()
-        creds.guess(lp)
-        creds.set_username(os.getenv("USERNAME"))
-        creds.set_password(os.getenv("PASSWORD"))
+        (lp,creds) = self.prep_creds()
         creds.set_smb_encryption(SMB_ENCRYPTION_REQUIRED)
 
         c = libsmb.Conn(os.getenv("SERVER_IP"), "tmp",