Initial commit

app/tests/api_testing/test_api.py

@@ -0,0 +1,81 @@
+# ------- standard library imports -------
+import json
+import unittest
+
+# ------- local imports -------
+from time import sleep
+
+from app.server.db.extensions import db
+from app.server.db.models import Url
+from app.app import create_app
+from app.tests.api_testing.api_helper import ApiHelper
+
+
+class TestApp(unittest.TestCase):
+    VALID_URL = 'youtube.com'
+    INVALID_URL = 'www.youtube.com/what?a=b&c=d'
+    INVALID_PARAM = 'INVALID'
+
+    def setUp(self):
+        self.helper = ApiHelper()
+        self.app = create_app(config_file='settings.py')
+        sleep(1)
+        self.key = self.helper.get_auth_token(self.app)
+
+    def test_01_shorten_url_success(self):
+        response = self.app.test_client().post(
+            '/api/shorten',
+            headers={'Authorization': f'Bearer {self.key}'},
+            data={'url': self.VALID_URL}
+        )
+
+        res_dict = json.loads(response.get_data(as_text=True))
+
+        short_url = res_dict['short_url']
+        original_url = res_dict['original_url']
+
+        self.assertEqual(response.status_code, 200)
+        self.assertEqual(len(short_url), 5)
+        self.assertEqual(original_url, 'http://youtube.com')
+
+    def test_02_shorten_url_fail(self):
+        response = self.app.test_client().post(
+            '/api/shorten',
+            headers={'Authorization': f'Bearer {self.key}'},
+            data={'url': self.INVALID_URL},
+        )
+
+        res_dict = json.loads(response.get_data(as_text=True))
+
+        self.assertEqual(response.status_code, 404)
+        self.assertEqual(res_dict['success'], False)
+        self.assertEqual(res_dict['message'], 'could not shorten this URL (page_not_found)')
+
+    def test_03_total_clicks(self):
+        # add url to db
+        response = self.app.test_client().post(
+            '/api/shorten',
+            headers={'Authorization': f'Bearer {self.key}'},
+            data={'url': 'youtube.com'},
+        )
+
+        with self.app.app_context():
+            url = Url.query.filter_by(original_url='http://youtube.com').first()
+            short_url = url.short_url
+
+        response = self.app.test_client().get(
+            '/api/total_clicks',
+            data={'url': short_url},
+        )
+
+        res_dict = json.loads(response.get_data(as_text=True))
+        self.assertEqual(res_dict['total'], 0)
+
+    def tearDown(self):
+        db.session.remove()
+        with self.app.app_context():
+            db.drop_all()
+
+
+if __name__ == '__main__':
+    unittest.main()