diff --git a/.github/workflows/pytest-unit-testing.yml b/.github/workflows/pytest-unit-testing.yml new file mode 100644 index 00000000..0ae4b66e --- /dev/null +++ b/.github/workflows/pytest-unit-testing.yml @@ -0,0 +1,34 @@ +name: Unit testing + +on: + push: + pull_request: + +jobs: + run-tests: + strategy: + fail-fast: false + matrix: + os: [ubuntu-latest, macos-latest, windows-latest] + python-version: + - "3.10" + - "3.11" + - "3.12" + + name: + runs-on: ${{ matrix.os }} + + steps: + - name: Checkout code + uses: actions/checkout@v3 + + - name: Set up Python + uses: actions/setup-python@v4 + with: + python-version: ${{ matrix.python-version }} + + - name: Install dependecies + run: python -m pip install pytest firebase-admin scikit-learn pandas numpy + + - name: Run tests + run: pytest diff --git a/app/tests/__init__.py b/app/tests/__init__.py new file mode 100644 index 00000000..e69de29b diff --git a/app/tests/test_database.py b/app/tests/test_database.py new file mode 100644 index 00000000..7bc93bd5 --- /dev/null +++ b/app/tests/test_database.py @@ -0,0 +1,54 @@ +from unittest.mock import patch, MagicMock +from services.database import create_document, get_documents, get_document, delete_document, update_document + + +@patch('services.database.firestore.client') +def test_create_document(mock_firestore_client): + mock_db = MagicMock() + mock_firestore_client.return_value = mock_db + collection, doc_id, data = 'test_collection', 'test_doc_id', { + 'field': 'value'} + result = create_document(collection, doc_id, data) + mock_db.collection().document().set.assert_called_with(data) + assert result == mock_db.collection().document().set.return_value + + +@patch('services.database.firestore.client') +def test_get_documents(mock_firestore_client): + mock_db = MagicMock() + mock_firestore_client.return_value = mock_db + collection, field, op, value = 'test_collection', 'field', '==', 'value' + result = get_documents(collection, field, op, value) + mock_db.collection().where().stream.assert_called_once() + assert result == mock_db.collection().where().stream.return_value + + +@patch('services.database.firestore.client') +def test_get_document(mock_firestore_client): + mock_db = MagicMock() + mock_firestore_client.return_value = mock_db + collection, doc_id = 'test_collection', 'test_doc_id' + result = get_document(collection, doc_id) + mock_db.collection().document().get.assert_called_once() + assert result == mock_db.collection().document().get.return_value + + +@patch('services.database.firestore.client') +def test_delete_document(mock_firestore_client): + mock_db = MagicMock() + mock_firestore_client.return_value = mock_db + collection, doc_id = 'test_collection', 'test_doc_id' + result = delete_document(collection, doc_id) + mock_db.collection().document().delete.assert_called_once() + assert result == mock_db.collection().document().delete.return_value + + +@patch('services.database.firestore.client') +def test_update_document(mock_firestore_client): + mock_db = MagicMock() + mock_firestore_client.return_value = mock_db + collection, doc_id, data = 'test_collection', 'test_doc_id', { + 'field': 'new_value'} + result = update_document(collection, doc_id, data) + mock_db.collection().document().update.assert_called_with(data) + assert result == mock_db.collection().document().update.return_value