From 3cfcc9ab8d138c4d59d4f25b169ba097c48eb065 Mon Sep 17 00:00:00 2001
From: Brian Scholer <1260690+briantist@users.noreply.github.com>
Date: Sun, 8 Oct 2023 16:36:27 -0400
Subject: [PATCH] add tests for IncomingCollectionStream

---
 tests/unit/conftest.py                        |  7 ++--
 .../test_incoming_collection_stream.py        | 40 +++++++++++++++++++
 2 files changed, 44 insertions(+), 3 deletions(-)
 create mode 100644 tests/unit/utilities/test_incoming_collection_stream.py

diff --git a/tests/unit/conftest.py b/tests/unit/conftest.py
index 0ba1086..352462f 100644
--- a/tests/unit/conftest.py
+++ b/tests/unit/conftest.py
@@ -5,6 +5,7 @@
 import json
 import sys
 
+from pathlib import Path
 from unittest import mock
 from shutil import copytree
 from artifactory import _ArtifactoryAccessor, _FakePathTemplate, ArtifactoryPath
@@ -32,14 +33,14 @@ def client(app):
 
 
 @pytest.fixture
-def virtual_fs_repo(fixture_finder, tmp_path):
+def virtual_fs_repo(fixture_finder, tmp_path: Path):
     repo = tmp_path / 'repo'
     copytree(fixture_finder('artifactory', 'virtual'), repo)
     return repo
 
 
 @pytest.fixture
-def mock_artifactory_accessor(fixture_loader, virtual_fs_repo):
+def mock_artifactory_accessor(fixture_loader, virtual_fs_repo: Path):
     class MockArtifactoryAccessor(_ArtifactoryAccessor):
         def __init__(self) -> None:
             super().__init__()
@@ -68,7 +69,7 @@ def get_stat_json(self, pathobj, key=None):
 
 
 @pytest.fixture
-def mock_artifactory_path(mock_artifactory_accessor, virtual_fs_repo):
+def mock_artifactory_path(mock_artifactory_accessor, virtual_fs_repo: Path):
     _artifactory_accessor = mock_artifactory_accessor()
 
     class MockArtifactoryPath(ArtifactoryPath):
diff --git a/tests/unit/utilities/test_incoming_collection_stream.py b/tests/unit/utilities/test_incoming_collection_stream.py
new file mode 100644
index 0000000..f6076f2
--- /dev/null
+++ b/tests/unit/utilities/test_incoming_collection_stream.py
@@ -0,0 +1,40 @@
+# -*- coding: utf-8 -*-
+# (c) 2023 Brian Scholer (@briantist)
+
+import pytest
+import tarfile
+
+from pathlib import Path
+from base64io import Base64IO
+from galactory.utilities import IncomingCollectionStream
+
+
+@pytest.fixture
+def collection_tarball(virtual_fs_repo: Path, tmp_path: Path):
+    collection = next(virtual_fs_repo.glob("**/*.tar.gz"))
+    gz_path = tmp_path / collection.name
+    with tarfile.open(gz_path, mode='w:gz') as tar:
+        tar.add(collection)
+    return gz_path
+
+
+@pytest.fixture
+def base64_tarball(collection_tarball: Path, tmp_path: Path):
+    b64_path = tmp_path / f"{collection_tarball.name}.b64"
+    with open(collection_tarball, mode='rb') as raw, open(b64_path, mode='wb') as w, Base64IO(w) as f:
+        f.write(raw.read())
+    return b64_path
+
+
+class TestIncomingCollectionStream:
+    @pytest.mark.parametrize('format', [None, 'auto', 'undefined', 'raw'])
+    def test_raw(self, collection_tarball: Path, format: str):
+        with open(collection_tarball, mode='rb') as f:
+            assert IncomingCollectionStream.detected_stream(f) is f
+            assert IncomingCollectionStream(f, format=format)
+
+    @pytest.mark.parametrize('format', [None, 'auto', 'undefined', 'base64'])
+    def test_base64(self, base64_tarball: Path, format: str):
+        with open(base64_tarball, mode='rb') as f:
+            assert isinstance(IncomingCollectionStream.detected_stream(f), Base64IO)
+            assert isinstance(IncomingCollectionStream(f, format=format), Base64IO)