mirror of
https://github.com/Novartis/cellxgene-gateway.git
synced 2026-09-15 20:57:59 +08:00
78 lines
3.2 KiB
Python
78 lines
3.2 KiB
Python
import unittest
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
from cellxgene_gateway.cache_entry import CacheEntry
|
|
from cellxgene_gateway.cache_key import CacheKey
|
|
from cellxgene_gateway.items.file.fileitem import FileItem
|
|
from cellxgene_gateway.items.file.fileitem_source import FileItemSource
|
|
from cellxgene_gateway.items.item import ItemType
|
|
from cellxgene_gateway.process_exception import ProcessException
|
|
|
|
|
|
class TestSubprocessBackend(unittest.TestCase):
|
|
@patch("subprocess.Popen")
|
|
def test_launch_GIVEN_no_stdout_THEN_throw_ProcessException(self, popen):
|
|
subprocess = MagicMock()
|
|
subprocess.stdout.readline().decode.return_value = ""
|
|
subprocess.stderr.read().decode.return_value = "An unexpected error"
|
|
popen.return_value = subprocess
|
|
|
|
key = CacheKey(
|
|
FileItem("/czi/", name="pbmc3k.h5ad", type=ItemType.h5ad),
|
|
FileItemSource("/tmp", "local"),
|
|
)
|
|
entry = CacheEntry.for_key(key, 8000)
|
|
from cellxgene_gateway.subprocess_backend import SubprocessBackend
|
|
|
|
backend = SubprocessBackend()
|
|
cellxgene_loc = "/some/cellxgene"
|
|
scripts = ["http://example.com/script.js", "http://example.com/script2.js"]
|
|
|
|
with self.assertRaises(ProcessException) as context:
|
|
backend.launch(cellxgene_loc, scripts, entry)
|
|
popen.assert_called_once_with(
|
|
[
|
|
"yes | /some/cellxgene launch /tmp/czi/pbmc3k.h5ad --port 8000 --host 127.0.0.1 --disable-annotations --disable-gene-sets-save --scripts http://example.com/script.js --scripts http://example.com/script2.js"
|
|
],
|
|
shell=True,
|
|
stderr=-1,
|
|
stdout=-1,
|
|
)
|
|
self.assertEqual("An unexpected error", context.exception.stderr)
|
|
|
|
@patch("subprocess.Popen")
|
|
def test_launch_GIVEN_annotations_enabled_THEN_set_flags(self, popen):
|
|
subprocess = MagicMock()
|
|
subprocess.stdout.readline().decode.return_value = (
|
|
"[cellxgene] Type CTRL-C at any time to exit.\n"
|
|
)
|
|
subprocess.stderr.read().decode.return_value = ""
|
|
popen.return_value = subprocess
|
|
|
|
key = CacheKey(
|
|
FileItem("/czi/", name="pbmc3k.h5ad", type=ItemType.h5ad),
|
|
FileItemSource("/tmp", "local"),
|
|
FileItem(
|
|
"/czi/pbmc3k_annotations/", name="foo.csv", type=ItemType.annotation
|
|
),
|
|
)
|
|
entry = CacheEntry.for_key(key, 8000)
|
|
import cellxgene_gateway.subprocess_backend
|
|
|
|
cellxgene_gateway.subprocess_backend.enable_annotations = True
|
|
try:
|
|
backend = cellxgene_gateway.subprocess_backend.SubprocessBackend()
|
|
cellxgene_loc = "/some/cellxgene"
|
|
|
|
backend.launch(cellxgene_loc, [], entry)
|
|
finally:
|
|
cellxgene_gateway.subprocess_backend.enable_annotations = False
|
|
popen.assert_called_once_with(
|
|
[
|
|
"yes | /some/cellxgene launch /tmp/czi/pbmc3k.h5ad --port 8000 --host 127.0.0.1 --annotations-file /tmp/czi/pbmc3k_annotations/foo.csv --gene-sets-file /tmp/czi/pbmc3k_annotations/foo_gene_sets.csv"
|
|
],
|
|
shell=True,
|
|
stderr=-1,
|
|
stdout=-1,
|
|
)
|