pygeai 0.1.51b3__py3-none-any.whl → 0.6.0b15__py3-none-any.whl

pygeai/tests/cli/commands/test_feedback.py

@@ -0,0 +1,65 @@
+import unittest
+from unittest.mock import patch, MagicMock
+from pygeai.cli.commands.feedback import show_help, send_feedback
+from pygeai.core.common.exceptions import MissingRequirementException
+from pygeai.cli.commands import Option
+
+
+class TestFeedback(unittest.TestCase):
+    """
+    python -m unittest pygeai.tests.cli.commands.test_feedback.TestFeedback
+    """
+
+    def test_show_help(self):
+        with patch('pygeai.cli.commands.feedback.Console.write_stdout') as mock_stdout:
+            show_help()
+            mock_stdout.assert_called_once()
+
+    def test_send_feedback_missing_required_parameters(self):
+        option_list = [
+            (Option("request_id", ["--request-id"], "", True), "req123")
+        ]
+        with self.assertRaises(MissingRequirementException) as cm:
+            send_feedback(option_list)
+        self.assertEqual(str(cm.exception), "Cannot send feedback without specifying request_id and answer_score")
+
+    def test_send_feedback_success_with_all_parameters(self):
+        option_list = [
+            (Option("request_id", ["--request-id"], "", True), "req123"),
+            (Option("origin", ["--origin"], "", True), "custom-origin"),
+            (Option("answer_score", ["--answer-score"], "", True), "1"),
+            (Option("comments", ["--comments"], "", True), "Great response!")
+        ]
+        with patch('pygeai.cli.commands.feedback.FeedbackClient') as mock_client, \
+             patch('pygeai.cli.commands.feedback.Console.write_stdout') as mock_stdout:
+            mock_client_instance = MagicMock()
+            mock_client.return_value = mock_client_instance
+            mock_client_instance.send_feedback.return_value = {"status": "success", "request_id": "req123"}
+            send_feedback(option_list)
+            mock_client_instance.send_feedback.assert_called_once_with(
+                request_id="req123",
+                origin="custom-origin",
+                answer_score="1",
+                comments="Great response!"
+            )
+            mock_stdout.assert_called_once_with("Feedback detail: \n{'status': 'success', 'request_id': 'req123'}")
+
+    def test_send_feedback_success_with_minimum_parameters(self):
+        option_list = [
+            (Option("request_id", ["--request-id"], "", True), "req123"),
+            (Option("answer_score", ["--answer-score"], "", True), "2")
+        ]
+        with patch('pygeai.cli.commands.feedback.FeedbackClient') as mock_client, \
+             patch('pygeai.cli.commands.feedback.Console.write_stdout') as mock_stdout:
+            mock_client_instance = MagicMock()
+            mock_client.return_value = mock_client_instance
+            mock_client_instance.send_feedback.return_value = {"status": "success", "request_id": "req123"}
+            send_feedback(option_list)
+            mock_client_instance.send_feedback.assert_called_once_with(
+                request_id="req123",
+                origin="user-feedback",  # Default value
+                answer_score="2",
+                comments=None  # Default value when not provided
+            )
+            mock_stdout.assert_called_once_with("Feedback detail: \n{'status': 'success', 'request_id': 'req123'}")
+

pygeai/tests/cli/commands/test_files.py

@@ -0,0 +1,161 @@
+import unittest
+from unittest.mock import patch, MagicMock
+from pygeai.cli.commands.files import (
+    show_help,
+    upload_file,
+    get_file,
+    delete_file,
+    get_file_content,
+    get_file_list
+)
+from pygeai.core.common.exceptions import MissingRequirementException
+from pygeai.cli.commands import Option
+
+
+class TestFiles(unittest.TestCase):
+    """
+    python -m unittest pygeai.tests.cli.commands.test_files.TestFiles
+    """
+
+    def test_show_help(self):
+        with patch('pygeai.cli.commands.files.Console.write_stdout') as mock_stdout:
+            show_help()
+            mock_stdout.assert_called_once()
+
+    def test_upload_file_missing_required_parameters(self):
+        option_list = [
+            (Option("organization", ["--organization"], "", True), "org123")
+        ]
+        with self.assertRaises(MissingRequirementException) as cm:
+            upload_file(option_list)
+        self.assertEqual(str(cm.exception), "Organization ID, Project ID and File path are mandatory parameters in order to upload a file.")
+
+    def test_upload_file_success(self):
+        option_list = [
+            (Option("organization", ["--organization"], "", True), "org123"),
+            (Option("project", ["--project"], "", True), "proj123"),
+            (Option("file_path", ["--file-path"], "", True), "/path/to/file"),
+            (Option("file_name", ["--file-name"], "", True), "testfile.txt"),
+            (Option("folder", ["--folder"], "", True), "testfolder")
+        ]
+        with patch('pygeai.cli.commands.files.FileClient') as mock_client, \
+             patch('pygeai.cli.commands.files.Console.write_stdout') as mock_stdout:
+            mock_client_instance = MagicMock()
+            mock_client.return_value = mock_client_instance
+            mock_client_instance.upload_file.return_value = {"id": "file123"}
+            upload_file(option_list)
+            mock_client_instance.upload_file.assert_called_once_with(
+                organization_id="org123",
+                project_id="proj123",
+                file_path="/path/to/file",
+                file_name="testfile.txt",
+                folder="testfolder"
+            )
+            mock_stdout.assert_called_once_with("Uploaded file: \n{'id': 'file123'}")
+
+    def test_get_file_missing_required_parameters(self):
+        option_list = [
+            (Option("organization", ["--organization"], "", True), "org123")
+        ]
+        with self.assertRaises(MissingRequirementException) as cm:
+            get_file(option_list)
+        self.assertEqual(str(cm.exception), "Cannot get file without organization, project and file_id.")
+
+    def test_get_file_success(self):
+        option_list = [
+            (Option("organization", ["--organization"], "", True), "org123"),
+            (Option("project", ["--project"], "", True), "proj123"),
+            (Option("file_id", ["--file-id"], "", True), "file123")
+        ]
+        with patch('pygeai.cli.commands.files.FileClient') as mock_client, \
+             patch('pygeai.cli.commands.files.Console.write_stdout') as mock_stdout:
+            mock_client_instance = MagicMock()
+            mock_client.return_value = mock_client_instance
+            mock_client_instance.get_file.return_value = {"id": "file123", "name": "testfile.txt"}
+            get_file(option_list)
+            mock_client_instance.get_file.assert_called_once_with(
+                organization="org123",
+                project="proj123",
+                file_id="file123"
+            )
+            mock_stdout.assert_called_once_with("File: \n{'id': 'file123', 'name': 'testfile.txt'}")
+
+    def test_delete_file_missing_required_parameters(self):
+        option_list = [
+            (Option("organization", ["--organization"], "", True), "org123")
+        ]
+        with self.assertRaises(MissingRequirementException) as cm:
+            delete_file(option_list)
+        self.assertEqual(str(cm.exception), "Cannot delete file without organization, project and file_id.")
+
+    def test_delete_file_success(self):
+        option_list = [
+            (Option("organization", ["--organization"], "", True), "org123"),
+            (Option("project", ["--project"], "", True), "proj123"),
+            (Option("file_id", ["--file-id"], "", True), "file123")
+        ]
+        with patch('pygeai.cli.commands.files.FileClient') as mock_client, \
+             patch('pygeai.cli.commands.files.Console.write_stdout') as mock_stdout:
+            mock_client_instance = MagicMock()
+            mock_client.return_value = mock_client_instance
+            mock_client_instance.delete_file.return_value = {"status": "deleted"}
+            delete_file(option_list)
+            mock_client_instance.delete_file.assert_called_once_with(
+                file_id="file123",
+                organization="org123",
+                project="proj123"
+            )
+            mock_stdout.assert_called_once_with("Deleted file: \n{'status': 'deleted'}")
+
+    def test_get_file_content_missing_required_parameters(self):
+        option_list = [
+            (Option("organization", ["--organization"], "", True), "org123")
+        ]
+        with self.assertRaises(MissingRequirementException) as cm:
+            get_file_content(option_list)
+        self.assertEqual(str(cm.exception), "Cannot get file content without organization, project and file_id.")
+
+    def test_get_file_content_success(self):
+        option_list = [
+            (Option("organization", ["--organization"], "", True), "org123"),
+            (Option("project", ["--project"], "", True), "proj123"),
+            (Option("file_id", ["--file-id"], "", True), "file123")
+        ]
+        with patch('pygeai.cli.commands.files.FileClient') as mock_client, \
+             patch('pygeai.cli.commands.files.Console.write_stdout') as mock_stdout:
+            mock_client_instance = MagicMock()
+            mock_client.return_value = mock_client_instance
+            mock_client_instance.get_file_content.return_value = "file content data"
+            get_file_content(option_list)
+            mock_client_instance.get_file_content.assert_called_once_with(
+                file_id="file123",
+                organization="org123",
+                project="proj123"
+            )
+            mock_stdout.assert_called_once_with("File content: \nfile content data")
+
+    def test_get_file_list_missing_required_parameters(self):
+        option_list = [
+            (Option("organization", ["--organization"], "", True), "org123")
+        ]
+        with self.assertRaises(MissingRequirementException) as cm:
+            get_file_list(option_list)
+        self.assertEqual(str(cm.exception), "Cannot file list without organization and project id.")
+
+    def test_get_file_list_success(self):
+        option_list = [
+            (Option("organization", ["--organization"], "", True), "org123"),
+            (Option("project", ["--project"], "", True), "proj123")
+        ]
+        with patch('pygeai.cli.commands.files.FileClient') as mock_client, \
+             patch('pygeai.cli.commands.files.Console.write_stdout') as mock_stdout:
+            mock_client_instance = MagicMock()
+            mock_client.return_value = mock_client_instance
+            mock_client_instance.get_file_list.return_value = [{"id": "file1"}, {"id": "file2"}]
+            get_file_list(option_list)
+            mock_client_instance.get_file_list.assert_called_once_with(
+                organization="org123",
+                project="proj123"
+            )
+            mock_stdout.assert_called_once_with("Files list: \n[{'id': 'file1'}, {'id': 'file2'}]")
+

pygeai/tests/cli/commands/test_gam.py

@@ -0,0 +1,201 @@
+import unittest
+from unittest.mock import patch, Mock
+from pygeai.cli.commands.gam import (
+    show_help,
+    generate_signin_url,
+    get_access_token,
+    get_user_info,
+    refresh_access_token,
+    get_authentication_types,
+    Option
+)
+from pygeai.core.common.exceptions import MissingRequirementException
+
+
+class TestGamCommands(unittest.TestCase):
+    """
+    python -m unittest pygeai.tests.cli.commands.test_gam.TestGamCommands
+    """
+    def setUp(self):
+        # Helper to create Option objects for testing
+        self.mock_option = lambda name, value: (Option(name, [f"--{name}"], f"Description for {name}", True), value)
+
+    @patch('pygeai.cli.commands.gam.Console.write_stdout')
+    @patch('pygeai.cli.commands.gam.build_help_text')
+    def test_show_help(self, mock_build_help, mock_write_stdout):
+        mock_help_text = "Mocked help text"
+        mock_build_help.return_value = mock_help_text
+
+        show_help()
+
+        mock_build_help.assert_called_once()
+        mock_write_stdout.assert_called_once_with(mock_help_text)
+
+    @patch('pygeai.cli.commands.gam.Console.write_stdout')
+    @patch('pygeai.cli.commands.gam.GAMClient')
+    def test_generate_signin_url_success(self, mock_client, mock_write_stdout):
+        mock_instance = Mock()
+        mock_client.return_value = mock_instance
+        mock_instance.generate_signing_url.return_value = "https://example.com/signin"
+        option_list = [
+            self.mock_option("client_id", "client123"),
+            self.mock_option("redirect_uri", "https://callback.com"),
+            self.mock_option("scope", "gam_user_data"),
+            self.mock_option("state", "random_state"),
+            self.mock_option("response_type", "code")
+        ]
+
+        generate_signin_url(option_list)
+
+        mock_instance.generate_signing_url.assert_called_once_with(
+            client_id="client123",
+            redirect_uri="https://callback.com",
+            scope="gam_user_data",
+            state="random_state",
+            response_type="code"
+        )
+        mock_write_stdout.assert_called_once_with("GAM Signin URL: \nhttps://example.com/signin")
+
+    def test_generate_signin_url_missing_required_fields(self):
+        option_list = [
+            self.mock_option("client_id", "client123"),
+            self.mock_option("redirect_uri", "https://callback.com")
+        ]
+
+        with self.assertRaises(MissingRequirementException) as context:
+            generate_signin_url(option_list)
+
+        self.assertEqual(str(context.exception), "client_id, redirect_uri, and state are required for generating signin URL")
+
+    @patch('pygeai.cli.commands.gam.Console.write_stdout')
+    @patch('pygeai.cli.commands.gam.GAMClient')
+    def test_get_access_token_success_with_client_credentials(self, mock_client, mock_write_stdout):
+        mock_instance = Mock()
+        mock_client.return_value = mock_instance
+        mock_instance.get_access_token.return_value = {"access_token": "token123"}
+        option_list = [
+            self.mock_option("client_id", "client123"),
+            self.mock_option("client_secret", "secret456"),
+            self.mock_option("grant_type", "password"),
+            self.mock_option("authentication_type_name", "local"),
+            self.mock_option("scope", "gam_user_data")
+        ]
+
+        get_access_token(option_list)
+
+        mock_instance.get_access_token.assert_called_once_with(
+            client_id="client123",
+            client_secret="secret456",
+            grant_type="password",
+            authentication_type_name="local",
+            scope="gam_user_data",
+            username=None,
+            password=None,
+            initial_properties=None,
+            repository=None,
+            request_token_type="OAuth"
+        )
+        mock_write_stdout.assert_called_once_with("GAM Access Token: \n{'access_token': 'token123'}")
+
+    @patch('pygeai.cli.commands.gam.Console.write_stdout')
+    @patch('pygeai.cli.commands.gam.GAMClient')
+    def test_get_access_token_success_with_username_password(self, mock_client, mock_write_stdout):
+        mock_instance = Mock()
+        mock_client.return_value = mock_instance
+        mock_instance.get_access_token.return_value = {"access_token": "token123"}
+        option_list = [
+            self.mock_option("username", "user1"),
+            self.mock_option("password", "pass123"),
+            self.mock_option("grant_type", "password"),
+            self.mock_option("authentication_type_name", "local"),
+            self.mock_option("scope", "gam_user_data")
+        ]
+
+        get_access_token(option_list)
+
+        mock_instance.get_access_token.assert_called_once_with(
+            client_id=None,
+            client_secret=None,
+            grant_type="password",
+            authentication_type_name="local",
+            scope="gam_user_data",
+            username="user1",
+            password="pass123",
+            initial_properties=None,
+            repository=None,
+            request_token_type="OAuth"
+        )
+        mock_write_stdout.assert_called_once_with("GAM Access Token: \n{'access_token': 'token123'}")
+
+    def test_get_access_token_missing_credentials(self):
+        option_list = []
+
+        with self.assertRaises(MissingRequirementException) as context:
+            get_access_token(option_list)
+
+        self.assertEqual(str(context.exception), "Cannot get access token without specifying valid credentials")
+
+    @patch('pygeai.cli.commands.gam.Console.write_stdout')
+    @patch('pygeai.cli.commands.gam.GAMClient')
+    def test_get_user_info_success(self, mock_client, mock_write_stdout):
+        mock_instance = Mock()
+        mock_client.return_value = mock_instance
+        mock_instance.get_user_info.return_value = {"user": "user1", "email": "user1@example.com"}
+        option_list = [self.mock_option("access_token", "token123")]
+
+        get_user_info(option_list)
+
+        mock_instance.get_user_info.assert_called_once_with(access_token="token123")
+        mock_write_stdout.assert_called_once_with("GAM User info: \n{'user': 'user1', 'email': 'user1@example.com'}")
+
+    def test_get_user_info_missing_access_token(self):
+        option_list = []
+
+        with self.assertRaises(MissingRequirementException) as context:
+            get_user_info(option_list)
+
+        self.assertEqual(str(context.exception), "Cannot get user info without the access token")
+
+    @patch('pygeai.cli.commands.gam.Console.write_stdout')
+    @patch('pygeai.cli.commands.gam.GAMClient')
+    def test_refresh_access_token_success(self, mock_client, mock_write_stdout):
+        mock_instance = Mock()
+        mock_client.return_value = mock_instance
+        mock_instance.refresh_access_token.return_value = {"access_token": "new_token123"}
+        option_list = [
+            self.mock_option("client_id", "client123"),
+            self.mock_option("client_secret", "secret456"),
+            self.mock_option("grant_type", "refresh_token"),
+            self.mock_option("refresh_token", "refresh_token789")
+        ]
+
+        refresh_access_token(option_list)
+
+        mock_instance.refresh_access_token.assert_called_once_with(
+            client_id="client123",
+            client_secret="secret456",
+            grant_type="refresh_token",
+            refresh_token="refresh_token789"
+        )
+        mock_write_stdout.assert_called_once_with("GAM Access Token: \n{'access_token': 'new_token123'}")
+
+    def test_refresh_access_token_missing_credentials(self):
+        option_list = []
+
+        with self.assertRaises(MissingRequirementException) as context:
+            refresh_access_token(option_list)
+
+        self.assertEqual(str(context.exception), "Cannot refresh access token without specifying valid credentials")
+
+    @patch('pygeai.cli.commands.gam.Console.write_stdout')
+    @patch('pygeai.cli.commands.gam.GAMClient')
+    def test_get_authentication_types_success(self, mock_client, mock_write_stdout):
+        mock_instance = Mock()
+        mock_client.return_value = mock_instance
+        mock_instance.get_authentication_types.return_value = {"types": ["local", "oauth"]}
+
+        get_authentication_types()
+
+        mock_instance.get_authentication_types.assert_called_once()
+        mock_write_stdout.assert_called_once_with("GAM Authentication Types: \n{'types': ['local', 'oauth']}")
+

pygeai/tests/cli/commands/test_llm.py

@@ -0,0 +1,114 @@
+import unittest
+from unittest.mock import patch, MagicMock
+
+from pygeai.cli.commands.llm import show_help, get_provider_list, get_provider_data, get_provider_models, get_model_data, llm_commands
+from pygeai.cli.commands import Option
+from pygeai.core.common.exceptions import MissingRequirementException
+
+
+class TestLlmCommands(unittest.TestCase):
+    """
+    python -m unittest pygeai.tests.cli.commands.test_llm.TestLlmCommands
+    """
+    def setUp(self):
+        self.mock_console = MagicMock()
+        self.mock_client = MagicMock()
+        self.mock_client.get_provider_list.return_value = {"providers": ["provider1", "provider2"]}
+        self.mock_client.get_provider_data.return_value = {"name": "provider1", "details": "test_data"}
+        self.mock_client.get_provider_models.return_value = {"models": ["model1", "model2"]}
+        self.mock_client.get_model_data.return_value = {"name": "model1", "details": "test_data"}
+        self.provider_option_list = [
+            (Option("provider_name", ["--provider-name"], "", True), "provider1")
+        ]
+        self.model_option_list_with_name = [
+            (Option("provider_name", ["--provider-name"], "", True), "provider1"),
+            (Option("model_name", ["--model-name"], "", True), "model1")
+        ]
+        self.model_option_list_with_id = [
+            (Option("provider_name", ["--provider-name"], "", True), "provider1"),
+            (Option("model_id", ["--model-id"], "", True), "model1")
+        ]
+
+    @patch('pygeai.cli.commands.llm.Console.write_stdout')
+    @patch('pygeai.cli.commands.llm.LlmClient')
+    def test_get_provider_list(self, mock_client_class, mock_write_stdout):
+        mock_client_class.return_value = self.mock_client
+
+        get_provider_list()
+
+        mock_client_class.assert_called_once()
+        self.mock_client.get_provider_list.assert_called_once()
+        mock_write_stdout.assert_called_once_with("Provider list: \n{'providers': ['provider1', 'provider2']}")
+
+    @patch('pygeai.cli.commands.llm.Console.write_stdout')
+    @patch('pygeai.cli.commands.llm.LlmClient')
+    def test_get_provider_data_success(self, mock_client_class, mock_write_stdout):
+        mock_client_class.return_value = self.mock_client
+
+        get_provider_data(self.provider_option_list)
+
+        mock_client_class.assert_called_once()
+        self.mock_client.get_provider_data.assert_called_once_with(provider_name="provider1")
+        mock_write_stdout.assert_called_once_with("Provider detail: \n{'name': 'provider1', 'details': 'test_data'}")
+
+    def test_get_provider_data_missing_provider_name(self):
+        option_list_empty = []
+
+        with self.assertRaises(MissingRequirementException):
+            get_provider_data(option_list_empty)
+
+    @patch('pygeai.cli.commands.llm.Console.write_stdout')
+    @patch('pygeai.cli.commands.llm.LlmClient')
+    def test_get_provider_models_success(self, mock_client_class, mock_write_stdout):
+        mock_client_class.return_value = self.mock_client
+
+        get_provider_models(self.provider_option_list)
+
+        mock_client_class.assert_called_once()
+        self.mock_client.get_provider_models.assert_called_once_with(provider_name="provider1")
+        mock_write_stdout.assert_called_once_with("Provider models: \n{'models': ['model1', 'model2']}")
+
+    def test_get_provider_models_missing_provider_name(self):
+        option_list_empty = []
+
+        with self.assertRaises(MissingRequirementException):
+            get_provider_models(option_list_empty)
+
+    @patch('pygeai.cli.commands.llm.Console.write_stdout')
+    @patch('pygeai.cli.commands.llm.LlmClient')
+    def test_get_model_data_success_with_model_name(self, mock_client_class, mock_write_stdout):
+        mock_client_class.return_value = self.mock_client
+
+        get_model_data(self.model_option_list_with_name)
+
+        mock_client_class.assert_called_once()
+        self.mock_client.get_model_data.assert_called_once_with(provider_name="provider1", model_name="model1")
+        mock_write_stdout.assert_called_once_with("Model details: \n{'name': 'model1', 'details': 'test_data'}")
+
+    @patch('pygeai.cli.commands.llm.Console.write_stdout')
+    @patch('pygeai.cli.commands.llm.LlmClient')
+    def test_get_model_data_success_with_model_id(self, mock_client_class, mock_write_stdout):
+        mock_client_class.return_value = self.mock_client
+
+        get_model_data(self.model_option_list_with_id)
+
+        mock_client_class.assert_called_once()
+        self.mock_client.get_model_data.assert_called_once_with(provider_name="provider1", model_name="model1")
+        mock_write_stdout.assert_called_once_with("Model details: \n{'name': 'model1', 'details': 'test_data'}")
+
+    def test_get_model_data_missing_provider_name(self):
+        option_list_no_provider = [
+            (Option("model_name", ["--model-name"], "", True), "model1")
+        ]
+
+        with self.assertRaises(MissingRequirementException):
+            get_model_data(option_list_no_provider)
+
+    def test_get_model_data_missing_model_info(self):
+        option_list_no_model = [
+            (Option("provider_name", ["--provider-name"], "", True), "provider1")
+        ]
+
+        with self.assertRaises(MissingRequirementException):
+            get_model_data(option_list_no_model)
+