main coverage

athena_federation/__init__.py

@@ -5,6 +5,7 @@
 # Import the SDK
 from .athena_data_source import AthenaDataSource  # noqa: F401
 from .lambda_handler import AthenaLambdaHandler  # noqa: F401
+from .main import main  # noqa: F401
 from .models import *  # noqa: F401,F403
 from .sdk import AthenaFederationSDK  # noqa: F401
 from .utils import AthenaSDKUtils  # noqa: F401

tests/test_main.py

@@ -0,0 +1,9 @@
+from athena_federation import main
+
+
+def test_main(capsys):
+    main()
+    captured = capsys.readouterr()
+    assert "Hello from athena_federation!" in captured.out
+    assert "Arguments: " in captured.out
+    assert "Environment:  " in captured.out