ran make pre-commit

Author: NtAlexio2

nettacker/core/lib/smb.py

@@ -24,17 +24,17 @@ def brute_force(self, *args, **kwargs):
             domain = kwargs["domain"]
             response.update({"domain": domain})
 
-        password = ''
+        password = ""
         if "password" in kwargs:
             password = kwargs["password"]
             response.update({"password": password})
 
-        lm = ''
+        lm = ""
         if "lm" in kwargs:
             lm = kwargs["lm"]
             response.update({"lm": lm})
 
-        nt = ''
+        nt = ""
         if "nt" in kwargs:
             nt = kwargs["nt"]
             response.update({"nt": nt})

tests/core/lib/test_smb.py

@@ -1,24 +1,21 @@
 from unittest.mock import patch
-from tests.common import TestCase
-from nettacker.core.lib.smb import (
-    SmbLibrary,
-    create_connection
-)
 
+from nettacker.core.lib.smb import SmbLibrary
+from tests.common import TestCase
 
 SMB_SESSION_PORT = 445
 
+
 class MockSmbConnectionObject:
-    def __init__(self, remoteName='', remoteHost='', sess_port=SMB_SESSION_PORT):
-        self._sess_port     = sess_port
-        self._remoteHost    = remoteHost
-        self._remoteName    = remoteName
+    def __init__(self, remoteName="", remoteHost="", sess_port=SMB_SESSION_PORT):
+        self._sess_port = sess_port
+        self._remoteHost = remoteHost
+        self._remoteName = remoteName
 
-    def login(self, user, password, domain = '', lmhash = '', nthash = ''):
+    def login(self, user, password, domain="", lmhash="", nthash=""):
         return None
 
 
-
 class TestSmbMethod(TestCase):
     @patch("nettacker.core.lib.smb.create_connection")
     def test_brute_force_password(self, mock_smb_connection):
@@ -28,7 +25,9 @@ def test_brute_force_password(self, mock_smb_connection):
         USERNAME = "Administrator"
         PASSWORD = "Password@123"
 
-        mock_smb_connection.return_value = MockSmbConnectionObject(HOST, remoteHost=HOST, sess_port=PORT)
+        mock_smb_connection.return_value = MockSmbConnectionObject(
+            HOST, remoteHost=HOST, sess_port=PORT
+        )
         self.assertEqual(
             library.brute_force(
                 host=HOST,
@@ -37,9 +36,9 @@ def test_brute_force_password(self, mock_smb_connection):
                 password=PASSWORD,
             ),
             {
-                'host': HOST,
-                'port': PORT,
-                'username': USERNAME,
-                'password': PASSWORD,
+                "host": HOST,
+                "port": PORT,
+                "username": USERNAME,
+                "password": PASSWORD,
             },
         )