Check nickname (#28)

* add main code

* add __init__.py,__main__.py,module.py,test_module.py

* fix module.py
fix test_module.py

* add test_module.py

* add requirements.txt

* fix tests,module.py
add new lines in __main__.py and __init__.py

* add test_module.py

* delete get_title

* add some tests to test_module.py
fix social_networks.txt

* fix styles
modify social_networks.txt

* fix module.py
fix test_module.py

* Cosmetic fixes

Co-authored-by: manmolecular <regwebghost@yandex.ru>

Author: Iandmee

src/scripts/osint/check_nickname/data/social_networks.txt

@@ -1,12 +1,12 @@
-twitter.com/
 vk.com/
 ok.ru/
 instagram.com/
 github.com/
 reddit.com/user/
-tiktok.com/@
 mastodon.social/@
-blackplanet.com/
 flickr.com/photos/
+gab.com/
+ficwad.com/a/
+www.opendiary.com/m/author/
 
 

src/scripts/osint/check_nickname/module.py

@@ -5,7 +5,7 @@
 
 import aiohttp
 import requests
-
+import string
 from src.core.base.osint import OsintRunner
 from src.core.utils.response import ScriptResponse
 
@@ -55,7 +55,7 @@ async def check_nickname_async(nickname: str, social) -> list:
         while not social.empty():
             url = await social.get()
             try:
-                async with session.get(url) as response:
+                async with session.get(url, timeout=5) as response:
                     if response.status == 200:
                         ans.append(url)
             except:
@@ -76,6 +76,11 @@ def __init__(self, logger: str = __name__):
     async def __run(*args, **kwargs):
         try:
             username = kwargs.get("username")
+            bad_symbols = list(set(username).intersection(string.punctuation))
+            if bad_symbols:
+                return ScriptResponse.error(
+                    message=f"Invalid characters: {', '.join(bad_symbols)}"
+                )
             social = asyncio.Queue()
             for site in Networks().net:
                 await social.put(

src/scripts/osint/check_nickname/test_module.py

@@ -0,0 +1,50 @@
+#!/usr/bin/env python3
+
+import string
+from random import choice, randrange
+from unittest import TestCase
+
+from .module import Runner
+
+
+class NicknameCheckTest(TestCase):
+    def setUp(self):
+        """
+        Setup something before each test function
+        :return: None
+        """
+        self.runner = Runner()
+
+    def test_create_runner(self) -> None:
+        """
+        Test creation of the class instance
+        :return: None
+        """
+        self.assertIsNotNone(self.runner)
+        self.assertIsInstance(self.runner, Runner)
+
+    def test_nicknames(self) -> None:
+        """
+        Test different usernames with random acceptable
+        characters
+        :return: None
+        """
+        response = self.runner.run(username="")
+        self.assertEqual(response.get("status"), "success")
+        for _ in range(5):
+            nickname = "".join(choice(string.ascii_letters) for _ in range(randrange(5, 10)))
+            response = self.runner.run(username=nickname)
+            self.assertEqual(response.get("status"), "success")
+
+    def test_special_nicknames(self) -> None:
+        """
+        Test not acceptable characters in username
+        and not acceptable formats of the stdin
+        :return:
+        """
+        response = self.runner.run(username=None)
+        self.assertEqual(response.get("status"), "error")
+        for _ in range(5):
+            nickname = "".join(choice(string.punctuation) for _ in range(randrange(3, 5)))
+            response = self.runner.run(username=nickname)
+            self.assertEqual(response.get("status"), "error")