Unittests base

.gitignore

@@ -3,3 +3,4 @@
 venv
 **/config.py
 globalcfg.py
+.coverage

build.sh

@@ -0,0 +1,13 @@
+#
+# Start Rabbit MQ Server in Docker.
+#
+# Expose ports:
+#   5671 --> 5671, 5672 --> 5672 (used by AMQP 0-9-1 and 1.0 clients without and with TLS)
+#   8080 --> 15672 (Management Plugin)
+#
+# Learn more: https://hub.docker.com/_/rabbitmq/
+
+
+docker run --hostname my-rabbit -d --name codex-bot-rabbitmq -p 5672:5672 -p 5671:5671 -p 8080:15672 rabbitmq:3
+
+docker run --name codex-bot-mongo -d -p 27017:27017 mongo

codexbot/services/telegram/telegram.py

@@ -63,5 +63,7 @@ def set_webhook(self):
             result = requests.get(query)
         except Exception as e:
             logging.debug(e)
+            return False
         else:
             logging.debug(result.content)
+            return result.content

requirements.txt

@@ -17,3 +17,6 @@ setuptools==35.0.2
     - six [required: Any, installed: 1.10.0]
   - six [required: >=1.6.0, installed: 1.10.0]
 wheel==0.29.0
+nose
+coverage
+report

runtests.sh

@@ -0,0 +1,6 @@
+#
+# Run tests and display results with code coverage
+#
+
+
+nosetests -v report --with-coverage --cover-package=codexbot/ tests/*.py

tests/base.py

@@ -0,0 +1,17 @@
+import json
+
+
+class BaseTest(object):
+    """This object represents a Base test and its sets of functions."""
+
+    def __init__(self, *args, **kwargs):
+        super(BaseTest, self).__init__(*args, **kwargs)
+
+    @staticmethod
+    def is_json(string):
+        try:
+            json.loads(string)
+        except ValueError:
+            return False
+
+        return True

tests/db.py

@@ -0,0 +1,23 @@
+import unittest
+
+from codexbot.lib.db import Db
+from tests.base import BaseTest
+
+
+class DatabaseTest(BaseTest, unittest.TestCase):
+
+    def __init__(self, *args, **kwargs):
+        super(BaseTest, self).__init__(*args, **kwargs)
+        self.db_name = 'test_db'
+        self.collection_name = 'test_collection'
+
+    def setUp(self):
+        self.db = Db(self.db_name)
+        self.db.insert(self.collection_name, {'id': 'test_id'})
+
+    def testDatabaseInit(self):
+        result = list(self.db.find(self.collection_name, {'id': 'test_id'}))
+        self.assertNotEquals(len(result), 0)
+
+    def tearDown(self):
+        self.db.remove(self.collection_name)

tests/telegram.py

@@ -0,0 +1,30 @@
+import json
+import unittest
+
+from codexbot.services.telegram import Telegram
+from tests.base import BaseTest
+
+
+class TelegramTestBase(BaseTest, unittest.TestCase):
+
+    def setUp(self):
+        self._telegram = Telegram()
+
+    def testConfiguration(self):
+        # Test presence
+        from codexbot.services.telegram.config import CALLBACK_ROUTE, BOT_NAME, API_TOKEN, API_URL
+        self.assertNotEqual(CALLBACK_ROUTE, "")
+        self.assertNotEqual(BOT_NAME, "")
+        self.assertNotEqual(API_TOKEN, "")
+        self.assertNotEqual(API_URL, "")
+
+        # Test route format
+        self.assertEqual(CALLBACK_ROUTE[0], "/")
+        self.assertNotEqual(CALLBACK_ROUTE[-1], "/")
+
+    def testSetWebhook(self):
+        result = self._telegram.set_webhook()
+        self.assertNotEqual(result, False)
+        self.is_json(result)
+        json_result = json.loads(result)
+        self.assertEqual(json_result.get("ok", None), True)