Add cli fixes (#1)

* Create extension

* Update src/migrate/azext_migrate/__init__.py

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

* Fix import issues

* Update src/migrate/setup.py

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

* Small

* Small lint

* Small

* disable lint for this check

* Add json

* Fix licesnse issue

* fix small

* Small

* Get rid of unused variables

* Add service name and code owner

* New version

* Style

* Small

* Update

* Follow standard

* Add suggestions

* Small

* Not preview

* Add flag to become experimental

* Update history

* Fix

* small

* Create get job and remove replication commands

* Add better error handling for jobs command

* Add better messages to remove protected item

* Return job id in remove command

* Move helpers

* Rename get discovered server helper

* Refactor _initialize_replication_infrastructure_helpers

* Refactor new replication

* Refactor jobs

* Refactor delete protected item

* Fix lint issues

* Change release version

* Fix lint issues

* Add fix

* Updae correct version

* Put beta version before

* Fix

* Update

* Fix bifurcation tool issue

* Update amh correctly if not proper

* Use current subscription id if not passed in

* Fix correct location and target resource group creation

* Create list protected items command

* Update versions

* Add __init__.py to all folders

* Create unit tests for get replication

* Update readme with accurate description of commands

* Update readme

* Create tests for jobs and remove commands

* Add more tests to increase coverage to 74%

* Add start migration command

* Address fixes

* Remove doc strings from custom.py

* Add init

* Remove old file locations

* fix failing tests

* Fix secret test

* Fix test

* Fix secret detection

* Fix duplicates

---------

Co-authored-by: Saif Al-Din Ali <[email protected]>
Co-authored-by: Copilot <[email protected]>

Author: 3 people

src/migrate/azext_migrate/tests/latest/test_migrate_commands.py

@@ -1958,7 +1958,7 @@ def mock_format(job):
             return job
 
         mock_cmd = mock.Mock()
-        mock_cmd.cli_ctx.cloud.endpoints.resource_manager = "https://management.azure.com"
+        mock_cmd.cli_ctx.cloud.endpoints.resource_manager = "mock-endpoint"
 
         # Execute - should raise error
         with self.assertRaises((CLIError, KnackCLIError)) as context:
@@ -1990,7 +1990,7 @@ def mock_format(job):
             return {'name': job['name']}
 
         mock_cmd = mock.Mock()
-        mock_cmd.cli_ctx.cloud.endpoints.resource_manager = "https://management.azure.com"
+        mock_cmd.cli_ctx.cloud.endpoints.resource_manager = 'mock-endpoint'
 
         # Execute
         result = list_all_jobs(
@@ -2474,7 +2474,7 @@ def test_parse_protected_item_id_invalid_format(self):
 
         self.assertIn('Invalid target object ID format', str(context.exception))
 
-    def test_parse_protected_item_id_empty(self):
+    def test_parse_protected_item_id_empty_for_remove(self):
         """Test parsing empty protected item ID."""
         from azext_migrate.helpers.replication.remove._parse import parse_protected_item_id
 
@@ -4313,7 +4313,7 @@ def test_parse_job_id_empty(self):
             parse_job_id("")
 
     @mock.patch('azext_migrate.helpers._utils.get_resource_by_id')
-    def test_get_vault_name_from_project_success(self, mock_get_resource):
+    def test_get_vault_name_from_project_success_for_job(self, mock_get_resource):
         """Test successfully getting vault name from project."""
         from azext_migrate.helpers.replication.job._parse import get_vault_name_from_project
 
@@ -5234,7 +5234,7 @@ def test_parse_protected_item_id_invalid(self):
 
         self.assertIn("Invalid protected item ID format", str(context.exception))
 
-    def test_parse_protected_item_id_empty(self):
+    def test_parse_protected_item_id_empty_for_migration(self):
         """Test parsing an empty protected item ID"""
         from azext_migrate.helpers.migration.start._parse import (
             parse_protected_item_id
@@ -5612,7 +5612,7 @@ def test_validate_arc_resource_bridge_not_found_warning(self, mock_send_request)
         # Should complete without exception
 
     @mock.patch('azext_migrate.helpers._utils.get_resource_by_id')
-    def test_get_vault_name_from_project_success(self, mock_get_resource):
+    def test_get_vault_name_from_project_success_for_migration(self, mock_get_resource):
         """Test successfully retrieving vault name from project"""
         from azext_migrate.helpers.replication.job._parse import (
             get_vault_name_from_project