16 lines
504 B
Python
16 lines
504 B
Python
import pytest
|
|
from unittest.mock import patch, MagicMock
|
|
from sut.backend.utils import database
|
|
|
|
# Example test for a database utility function using AAA and mocking
|
|
|
|
def test_get_connection_returns_connection():
|
|
# Arrange
|
|
with patch('sut.backend.utils.database.create_engine') as mock_create_engine:
|
|
mock_engine = MagicMock()
|
|
mock_create_engine.return_value = mock_engine
|
|
# Act
|
|
conn = database.get_connection()
|
|
# Assert
|
|
assert conn == mock_engine
|