Trim long user name for kubernetes label (#80)

* Trim long user name for kubernetes label

* Add dots in the end to indicate the username is trimmed

* Add -trimmed at the end of trimmed username

---------

Co-authored-by: pintaoz <[email protected]>

Author: pintaoz-aws

src/hyperpod_cli/commands/job.py

@@ -1142,4 +1142,11 @@ def get_user_name():
         user_name = 'Unknown'
 
     # label value does not allow slash
-    return user_name.replace('/', '-')
+    user_name = user_name.replace('/', '-')
+    # 63 is the max length for a Kubernetes label
+    if len(user_name) > 63:
+        # Add dots in the end to indicate the username is trimmed
+        trimmed_user_name = user_name[:55] + '-trimmed'
+        logger.warning(f"The username is longer than the maximum length (63) of Kubernetes label, trimming to {trimmed_user_name}")
+        return trimmed_user_name
+    return user_name

test/unit_tests/test_job.py

@@ -27,6 +27,7 @@
 from hyperpod_cli.commands.job import (
     cancel_job,
     get_job,
+    get_user_name,
     list_jobs,
     list_pods,
     patch_job,
@@ -2138,3 +2139,16 @@ def test_start_job_with_recipe_missing_required_args(self):
             ],
         )
         self.assertNotEqual(result.exit_code, 0)
+
+    @mock.patch("boto3.client")
+    def test_get_user_name_long_arn(self, mock_boto3_client):
+        mock_client = MagicMock()
+        mock_boto3_client.return_value = mock_client
+
+        mock_identity = MagicMock()
+        mock_client.get_caller_identity.return_value = mock_identity
+
+        long_user_name = "ReadOnly/AmazonSageMaker-a-long-long-long-long-long-long-long-arn-for-testing-get-user-name"
+        long_arn = "arn:aws:iam::0123456789012:assumed-role/" + long_user_name
+        mock_identity.get.return_value = long_arn
+        self.assertEqual(get_user_name(), "AssumedRole-" + long_user_name.replace("/", "-")[:43] + "-trimmed")