feat: unit tests

justinmerrell · justinmerrell · commit 6d50d3c09347 · 2022-12-12T09:13:45.000-05:00
diff --git a/.github/workflows/CI_tests.yml b/.github/workflows/CI_tests.yml
@@ -0,0 +1,37 @@
+name: CI | Unit Tests
+
+on:
+  push:
+    branches:
+      - '**'
+      - 'master'
+      - '!master-ci'
+      - '!release'
+
+  pull_request:
+    branches:
+      - master
+
+  workflow_dispatch:
+
+jobs:
+  test:
+    runs-on: ubuntu-latest
+
+    steps:
+      - uses: actions/checkout@v2
+
+      - name: Set up Python 3.11
+        uses: actions/setup-python@v2
+        with:
+          python-version: 3.11.0
+
+      - name: Install Dependencies
+        run: |
+          python -m pip install --upgrade pip
+          python setup.py install
+          pip install pytest
+
+      - name: Run Tests
+        run: |
+          pytest
diff --git a/tests/__init__.py b/tests/__init__.py
@@ -0,0 +1 @@
+''' Allows the tests to be run from the top level directory.'''
diff --git a/tests/serverless_tests/__init__.py b/tests/serverless_tests/__init__.py
@@ -0,0 +1 @@
+''' Allows the import of serverless_tests as a module. '''
diff --git a/tests/serverless_tests/test_module_download.py b/tests/serverless_tests/test_module_download.py
@@ -0,0 +1,41 @@
+''' Tests for runpod | serverless| modules | download.py '''
+
+import unittest
+from unittest import mock
+from unittest.mock import patch, MagicMock, mock_open
+
+from runpod.serverless.modules.download import download_input_objects
+
+
+def mock_requests_get(*args, **kwargs):
+    '''
+    Mocks requests.get
+    '''
+    class MockResponse:
+        ''' Mocks requests.get response '''
+
+        def __init__(self, content, status_code):
+            '''
+            Mocks requests.get response
+            '''
+            self.content = content
+            self.status_code = status_code
+
+    if args[0] == 'https://example.com/picture.jpg':
+        return MockResponse(b'nothing', 200)
+
+    return MockResponse(None, 404)
+
+
+class TestDownloadInputObjects(unittest.TestCase):
+    ''' Tests for download_input_objects '''
+
+    @patch('requests.get', side_effect=mock_requests_get)
+    @patch('builtins.open', new_callable=mock_open)
+    def test_download_input_objects(self):
+        '''
+        Tests download_input_objects
+        '''
+        objects = download_input_objects(
+            ['https://example.com/picture.jpg', ]
+        )

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1 @@`
	`1`	`+''' Allows the tests to be run from the top level directory.'''`
Original file line number	Diff line number	Diff line change
`@@ -0,0 +1 @@`
	`1`	`+''' Allows the import of serverless_tests as a module. '''`