test: ✅ Add unit tests for SELinux usecase

Author: sodre

tests/unit/sagemaker/local/test_local_image.py

@@ -30,7 +30,7 @@
 from mock import patch, Mock, MagicMock
 
 import sagemaker
-from sagemaker.local.image import _SageMakerContainer, _aws_credentials
+from sagemaker.local.image import _SageMakerContainer, _Volume, _aws_credentials
 
 REGION = "us-west-2"
 BUCKET_NAME = "mybucket"
@@ -567,6 +567,26 @@ def test_train_local_intermediate_output(get_data_source_instance, tmpdir, sagem
                 assert "%s:/opt/ml/output/intermediate" % intermediate_folder_path in volumes
 
 
+@patch("platform.system", Mock(return_value="Linux"))
+@patch("sagemaker.local.image.SELINUX_ENABLED", Mock(return_value=True))
+def test_container_selinux_has_label(tmpdir, sagemaker_session):
+    instance_count = 1
+    image = "my-image"
+    volume = _Volume(str(tmpdir), "/opt/ml/model")
+
+    assert volume.map.endswith(":z")
+
+
+@patch("platform.system", Mock(return_value="Darwin"))
+@patch("sagemaker.local.image.SELINUX_ENABLED", Mock(return_value=True))
+def test_container_has_selinux_no_label(tmpdir, sagemaker_session, monkeypatch):
+    instance_count = 1
+    image = "my-image"
+    volume = _Volume(str(tmpdir), "/opt/ml/model")
+
+    assert not volume.map.endswith(":z")
+
+
 def test_container_has_gpu_support(tmpdir, sagemaker_session):
     instance_count = 1
     image = "my-image"