|
|
"""Test simple msa.""" |
|
|
|
|
|
from pathlib import Path |
|
|
from unittest import mock |
|
|
|
|
|
import pytest |
|
|
from folding_studio.api_call.msa import simple_msa |
|
|
from folding_studio.config import API_URL, REQUEST_TIMEOUT |
|
|
from folding_studio.utils.data_model import MSARequestParams |
|
|
|
|
|
|
|
|
@pytest.fixture(autouse=True) |
|
|
def mock_post(): |
|
|
post_mock = mock.Mock() |
|
|
mock_response = mock.MagicMock() |
|
|
mock_response.ok = True |
|
|
post_mock.return_value = mock_response |
|
|
with mock.patch("requests.post", post_mock): |
|
|
yield post_mock |
|
|
|
|
|
|
|
|
@pytest.fixture(autouse=True) |
|
|
def mock_get_auth_headers(): |
|
|
with mock.patch( |
|
|
"folding_studio.api_call.msa.get_auth_headers", |
|
|
return_value={"Authorization": "Bearer identity_token"}, |
|
|
) as m: |
|
|
yield m |
|
|
|
|
|
|
|
|
def test_simple_msa_pass( |
|
|
tmp_path: Path, mock_post: pytest.FixtureRequest, headers: dict[str, str] |
|
|
): |
|
|
"""Test simple msa pass.""" |
|
|
|
|
|
file = tmp_path / "fasta_file.fasta" |
|
|
file.touch() |
|
|
|
|
|
params = MSARequestParams( |
|
|
ignore_cache=False, |
|
|
msa_mode="search", |
|
|
) |
|
|
|
|
|
simple_msa( |
|
|
file, |
|
|
params, |
|
|
project_code="FOLDING_DEV", |
|
|
) |
|
|
|
|
|
mock_post.assert_called_once_with( |
|
|
API_URL + "searchMSA", |
|
|
data=params.model_dump(mode="json"), |
|
|
headers=headers, |
|
|
files=mock.ANY, |
|
|
timeout=REQUEST_TIMEOUT, |
|
|
params={"project_code": "FOLDING_DEV"}, |
|
|
) |
|
|
|
