feat: add unit tests for SMB brute force module

Author: NtAlexio2

tests/core/lib/test_smb.py

@@ -0,0 +1,45 @@
+from unittest.mock import patch
+from tests.common import TestCase
+from nettacker.core.lib.smb import (
+    SmbLibrary,
+    create_connection
+)
+
+
+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 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):
+        library = SmbLibrary()
+        HOST = "dc-01"
+        PORT = 445
+        USERNAME = "Administrator"
+        PASSWORD = "Password@123"
+
+        mock_smb_connection.return_value = MockSmbConnectionObject(HOST, remoteHost=HOST, sess_port=PORT)
+        self.assertEqual(
+            library.brute_force(
+                host=HOST,
+                port=PORT,
+                username=USERNAME,
+                password=PASSWORD,
+            ),
+            {
+                'host': HOST,
+                'port': PORT,
+                'username': USERNAME,
+                'password': PASSWORD,
+            },
+        )