PR-7456 unit tests for ctorm

Author: benbart

.github/workflows/test.yml

@@ -8,17 +8,23 @@ jobs:
     runs-on: ubuntu-latest
     strategy:
       matrix:
-        python-version: ['3.9', '3.10', '3.11', '3.12']
+        python-version: [ '3.9', '3.10', '3.11', '3.12' ]
       fail-fast: false
 
     steps:
-    - uses: actions/checkout@v4
-    - name: Set up Python ${{ matrix.python-version }}
-      uses: actions/setup-python@v5
-      with:
-        python-version: ${{ matrix.python-version }}
-    - name: Install dependencies
-      run: |
-        python -m pip install --upgrade pip
-        python -m pip install poetry "tox<4.0.0" tox-poetry tox-gh-actions
-    - run: make tests
+      - uses: actions/checkout@v4
+      - name: Set up Python ${{ matrix.python-version }}
+        uses: actions/setup-python@v5
+        with:
+          python-version: ${{ matrix.python-version }}
+      - name: Install dependencies
+        run: |
+          python -m pip install --upgrade pip
+          python -m pip install poetry "tox<4.0.0" tox-poetry tox-gh-actions
+      - name: Run tests
+        run: |
+          make tests
+          if [ "${{ matrix.python-version }}" = "3.12" ]; then
+            # ctorm supports py312 only. 
+            make ctorm-test
+          fi

Makefile

@@ -13,3 +13,7 @@ TOX_INIS := $(PACKAGES:%=%/tox.ini)
 .PHONY: tests
 tests: $(TOX_INIS)
 	ss=0; $(foreach package,$(PACKAGES),cd $(package) && tox || ss=1;cd ..;) exit $$ss
+
+.PHONY: ctorm-test
+ctorm-test:
+	cd ctorm && tox -c pyproject.toml

ctorm/pyproject.toml

@@ -9,13 +9,13 @@ authors = [
 requires-python = ">=3.12"
 dependencies = [
     "boto3>=1.43.26",
+    "mypy-boto3-s3>=1.43.14",
 ]
 
 [dependency-groups]
 dev = [
     "black>=26.5.1",
     "boto3-stubs>=1.43.27",
-    "mypy-boto3-s3>=1.43.14",
     "pytest>=8.0.0",
     "ruff>=0.15.13",
 ]
@@ -29,3 +29,17 @@ build-backend = "uv_build"
 
 [tool.uv.build-backend]
 module-root = ""
+
+
+[tool.tox]
+legacy_tox_ini = """
+[tox]
+envlist = py312
+[testenv]
+deps =
+    mypy-boto3-s3
+    pytest
+    pytest-mock
+commands =
+    pytest
+"""

ctorm/tests/__init__.py

@@ -0,0 +1,43 @@
+from unittest.mock import mock_open, patch
+
+from ctorm.config import ConfigError, CtormBucket, CtormConfig
+
+
+def test_ctorm_bucket_initialization():
+    bucket = CtormBucket(bucketname="test_bucket", share=50, ummg_prefix="test_prefix/")
+
+    assert bucket.bucketname == "test_bucket"
+    assert bucket.share == 50
+    assert bucket.ummg_prefix == "test_prefix/"
+    assert bucket.next_cont_token is None
+
+
+def test_ctorm_config_from_file_success():
+    mock_data = """
+    [ctorm]
+    source_buckets = [{"bucketname": "test_bucket", "share": 50}]
+    granules_sqs_queue_url = "https://sqs.queue.url/"
+    """
+    with patch("builtins.open", mock_open(read_data=mock_data)), patch("tomllib.load") as mock_toml:
+        mock_toml.return_value = {
+            "ctorm": {
+                "source_buckets": [{"bucketname": "test_bucket", "share": 50}],
+                "granules_sqs_queue_url": "https://sqs.queue.url/",
+            }
+        }
+        config = CtormConfig.from_file("test.cfg")
+
+        assert config.granules_sqs_queue_url == "https://sqs.queue.url/"
+        assert len(config.source_buckets) == 1
+        assert isinstance(config.source_buckets[0], CtormBucket)
+        assert config.source_buckets[0].bucketname == "test_bucket"
+
+
+def test_ctorm_config_from_file_missing_section():
+    mock_data = "{}"
+    with patch("builtins.open", mock_open(read_data=mock_data)), patch("tomllib.load") as mock_toml:
+        mock_toml.return_value = {}
+        try:
+            CtormConfig.from_file("test.cfg")
+        except ConfigError as e:
+            assert str(e) == "No 'ctorm' section in config file"