mage-ai 0.9.74__py3-none-any.whl → 0.9.79__py3-none-any.whl

mage_ai/tests/api/endpoints/test_pipelines.py

@@ -1,5 +1,3 @@
-from unittest.mock import patch
-
 from mage_ai.data_preparation.models.pipeline import Pipeline
 from mage_ai.tests.api.endpoints.mixins import (
     BaseAPIEndpointTest,
@@ -9,7 +7,6 @@ from mage_ai.tests.api.endpoints.mixins import (
     build_list_endpoint_tests,
     build_update_endpoint_tests,
 )
-from mage_ai.tests.shared.mixins import ProjectPlatformMixin
 
 
 class PipelineAPIEndpointTest(BaseAPIEndpointTest):
@@ -111,134 +108,3 @@ build_delete_endpoint_tests(
         Pipeline.get_all_pipelines(self.repo_path),
     ) == 1,
 )
-
-
-class PipelineWithProjectPlatformAPIEndpointTest(BaseAPIEndpointTest, ProjectPlatformMixin):
-    def setUp(self):
-        with patch('mage_ai.settings.platform.project_platform_activated', lambda: True):
-            with patch('mage_ai.settings.repo.project_platform_activated', lambda: True):
-                super().setUp()
-                self.setup_final()
-
-    def tearDown(self):
-        with patch('mage_ai.settings.platform.project_platform_activated', lambda: True):
-            with patch('mage_ai.settings.repo.project_platform_activated', lambda: True):
-                self.teardown_final()
-                super().tearDown()
-
-
-build_list_endpoint_tests(
-    PipelineWithProjectPlatformAPIEndpointTest,
-    resource='pipelines',
-    list_count=3,
-    result_keys_to_compare=[
-        'blocks',
-        'callbacks',
-        'concurrency_config',
-        'conditionals',
-        'created_at',
-        'data_integration',
-        'description',
-        'executor_config',
-        'executor_count',
-        'executor_type',
-        'name',
-        'notification_config',
-        'retry_config',
-        'run_pipeline_in_one_process',
-        'spark_config',
-        'tags',
-        'type',
-        'updated_at',
-        'uuid',
-        'widgets',
-    ],
-    patch_function_settings=[
-        ('mage_ai.settings.repo.project_platform_activated', lambda: True),
-        ('mage_ai.settings.platform.project_platform_activated', lambda: True),
-    ],
-)
-
-
-build_create_endpoint_tests(
-    PipelineWithProjectPlatformAPIEndpointTest,
-    resource='pipelines',
-    assert_after_create_count=lambda self, mocks: len(
-        Pipeline.get_all_pipelines(self.repo_path),
-    ) == 4,
-    assert_before_create_count=lambda self, mocks: len(
-        Pipeline.get_all_pipelines(self.repo_path),
-    ) == 3,
-    build_payload=lambda self: dict(
-        name=self.faker.unique.name(),
-    ),
-    patch_function_settings=[
-        ('mage_ai.settings.repo.project_platform_activated', lambda: True),
-        ('mage_ai.settings.platform.project_platform_activated', lambda: True),
-    ],
-)
-
-
-build_detail_endpoint_tests(
-    PipelineWithProjectPlatformAPIEndpointTest,
-    resource='pipelines',
-    get_resource_id=lambda self: self.pipeline.uuid,
-    result_keys_to_compare=[
-        'blocks',
-        'callbacks',
-        'concurrency_config',
-        'conditionals',
-        'created_at',
-        'data_integration',
-        'description',
-        'executor_config',
-        'executor_count',
-        'executor_type',
-        'extensions',
-        'name',
-        'notification_config',
-        'retry_config',
-        'run_pipeline_in_one_process',
-        'spark_config',
-        'tags',
-        'type',
-        'updated_at',
-        'uuid',
-        'widgets',
-    ],
-    patch_function_settings=[
-        ('mage_ai.settings.repo.project_platform_activated', lambda: True),
-        ('mage_ai.settings.platform.project_platform_activated', lambda: True),
-    ],
-)
-
-
-build_update_endpoint_tests(
-    PipelineWithProjectPlatformAPIEndpointTest,
-    resource='pipelines',
-    get_resource_id=lambda self: self.pipeline.uuid,
-    build_payload=lambda self: dict(name=self.faker.unique.name()),
-    get_model_before_update=lambda self: self.pipeline,
-    assert_after_update=lambda self, result, model, mocks: model.name != result['name'],
-    patch_function_settings=[
-        ('mage_ai.settings.repo.project_platform_activated', lambda: True),
-        ('mage_ai.settings.platform.project_platform_activated', lambda: True),
-    ],
-)
-
-
-build_delete_endpoint_tests(
-    PipelineWithProjectPlatformAPIEndpointTest,
-    resource='pipelines',
-    get_resource_id=lambda self: self.pipeline.uuid,
-    assert_after_delete_count=lambda self, mocks: len(
-        Pipeline.get_all_pipelines(self.repo_path),
-    ) == 2,
-    assert_before_delete_count=lambda self, mocks: len(
-        Pipeline.get_all_pipelines(self.repo_path),
-    ) == 3,
-    patch_function_settings=[
-        ('mage_ai.settings.repo.project_platform_activated', lambda: True),
-        ('mage_ai.settings.platform.project_platform_activated', lambda: True),
-    ],
-)

mage_ai/tests/api/endpoints/test_pipelines_with_project_platform.py

@@ -0,0 +1,143 @@
+from unittest.mock import patch
+
+from mage_ai.data_preparation.models.pipeline import Pipeline
+from mage_ai.tests.api.endpoints.mixins import (
+    BaseAPIEndpointTest,
+    build_create_endpoint_tests,
+    build_delete_endpoint_tests,
+    build_detail_endpoint_tests,
+    build_list_endpoint_tests,
+    build_update_endpoint_tests,
+)
+from mage_ai.tests.shared.mixins import ProjectPlatformMixin
+
+
+class PipelineWithProjectPlatformAPIEndpointTest(BaseAPIEndpointTest, ProjectPlatformMixin):
+    def setUp(self):
+        with patch('mage_ai.settings.platform.project_platform_activated', lambda: True):
+            with patch('mage_ai.settings.repo.project_platform_activated', lambda: True):
+                super().setUp()
+                self.setup_final()
+
+    def tearDown(self):
+        with patch('mage_ai.settings.platform.project_platform_activated', lambda: True):
+            with patch('mage_ai.settings.repo.project_platform_activated', lambda: True):
+                self.teardown_final()
+                super().tearDown()
+
+
+build_list_endpoint_tests(
+    PipelineWithProjectPlatformAPIEndpointTest,
+    resource='pipelines',
+    list_count=3,
+    result_keys_to_compare=[
+        'blocks',
+        'callbacks',
+        'concurrency_config',
+        'conditionals',
+        'created_at',
+        'data_integration',
+        'description',
+        'executor_config',
+        'executor_count',
+        'executor_type',
+        'name',
+        'notification_config',
+        'retry_config',
+        'run_pipeline_in_one_process',
+        'spark_config',
+        'tags',
+        'type',
+        'updated_at',
+        'uuid',
+        'widgets',
+    ],
+    patch_function_settings=[
+        ('mage_ai.settings.repo.project_platform_activated', lambda: True),
+        ('mage_ai.settings.platform.project_platform_activated', lambda: True),
+    ],
+)
+
+
+build_create_endpoint_tests(
+    PipelineWithProjectPlatformAPIEndpointTest,
+    resource='pipelines',
+    assert_after_create_count=lambda self, mocks: len(
+        Pipeline.get_all_pipelines(self.repo_path),
+    ) == 4,
+    assert_before_create_count=lambda self, mocks: len(
+        Pipeline.get_all_pipelines(self.repo_path),
+    ) == 3,
+    build_payload=lambda self: dict(
+        name=self.faker.unique.name(),
+    ),
+    patch_function_settings=[
+        ('mage_ai.settings.repo.project_platform_activated', lambda: True),
+        ('mage_ai.settings.platform.project_platform_activated', lambda: True),
+    ],
+)
+
+
+build_detail_endpoint_tests(
+    PipelineWithProjectPlatformAPIEndpointTest,
+    resource='pipelines',
+    get_resource_id=lambda self: self.pipeline.uuid,
+    result_keys_to_compare=[
+        'blocks',
+        'callbacks',
+        'concurrency_config',
+        'conditionals',
+        'created_at',
+        'data_integration',
+        'description',
+        'executor_config',
+        'executor_count',
+        'executor_type',
+        'extensions',
+        'name',
+        'notification_config',
+        'retry_config',
+        'run_pipeline_in_one_process',
+        'spark_config',
+        'tags',
+        'type',
+        'updated_at',
+        'uuid',
+        'widgets',
+    ],
+    patch_function_settings=[
+        ('mage_ai.settings.repo.project_platform_activated', lambda: True),
+        ('mage_ai.settings.platform.project_platform_activated', lambda: True),
+    ],
+)
+
+
+build_update_endpoint_tests(
+    PipelineWithProjectPlatformAPIEndpointTest,
+    resource='pipelines',
+    get_resource_id=lambda self: self.pipeline.uuid,
+    build_payload=lambda self: dict(name=self.faker.unique.name()),
+    get_model_before_update=lambda self: self.pipeline,
+    assert_after_update=lambda self, result, model, mocks: model.name != result['name'],
+    patch_function_settings=[
+        ('mage_ai.settings.repo.project_platform_activated', lambda: True),
+        ('mage_ai.settings.platform.project_platform_activated', lambda: True),
+    ],
+)
+
+
+build_delete_endpoint_tests(
+    PipelineWithProjectPlatformAPIEndpointTest,
+    resource='pipelines',
+    get_resource_id=lambda self: self.pipeline.uuid,
+    assert_after_delete_count=lambda self, mocks: len(
+        Pipeline.get_all_pipelines(self.repo_path),
+    ) == 2,
+    assert_before_delete_count=lambda self, mocks: len(
+        Pipeline.get_all_pipelines(self.repo_path),
+    ) == 3,
+    patch_function_settings=[
+        ('mage_ai.settings.repo.project_platform_activated', lambda: True),
+        ('mage_ai.settings.platform.project_platform_activated', lambda: True),
+    ],
+)

mage_ai/tests/data_preparation/executors/test_block_executor.py

@@ -98,7 +98,7 @@ class BlockExecutorTest(BaseApiTestCase):
         self.block_executor._execute_conditional = MagicMock(return_value=True)
         self.block_executor._execute = MagicMock(return_value={'result': 'success'})
         # self.block.run_tests = MagicMock()
-        self.block_executor._execute_callback = MagicMock()
+        self.block_executor.execute_callback = MagicMock()
 
         result = self.block_executor.execute(
             analyze_outputs=analyze_outputs,
@@ -167,7 +167,7 @@ class BlockExecutorTest(BaseApiTestCase):
         #     update_tests=False,
         #     dynamic_block_uuid=dynamic_block_uuid,
         # )
-        self.block_executor._execute_callback.assert_called_with(
+        self.block_executor.execute_callback.assert_called_with(
             'on_success',
             block_run_id=None,
             callback_kwargs=dict(retry=dict(attempts=1)),
@@ -245,7 +245,7 @@ class BlockExecutorTest(BaseApiTestCase):
         self.block.callback_blocks = [MagicMock(), MagicMock()]
         self.block.callback_block = MagicMock()
 
-        self.block_executor._execute_callback(
+        self.block_executor.execute_callback(
             callback='on_success',
             global_vars=dict(retry=dict(attempts=1)),
             logging_tags={},

mage_ai/tests/data_preparation/logging/test_logger_manager.py

@@ -1,8 +1,10 @@
 import os
 from datetime import datetime, timedelta
-from unittest.mock import patch
+from unittest.mock import Mock, patch
 
 from mage_ai.data_preparation.logging.logger_manager import LoggerManager
+from mage_ai.data_preparation.models.constants import PipelineType
+from mage_ai.data_preparation.models.pipeline import Pipeline
 from mage_ai.data_preparation.storage.local_storage import LocalStorage
 from mage_ai.tests.base_test import TestCase
 
@@ -19,7 +21,13 @@ class LoggerManagerTest(TestCase):
         # Mock the storage object to avoid actual deletion of files
         logger_manager.storage = MockStorage()
         self.__create_log_dir(logger_manager, mock_pipeline_uuid)
-        logger_manager.delete_old_logs()
+        # Mock the Pipeline.get to return a pipeline with type PipelineType.PYTHON
+        mock_pipeline = Mock(spec=Pipeline)
+        mock_pipeline.type = PipelineType.PYTHON
+        mock_get_pipeline = Mock(spec=Pipeline.get)
+        mock_get_pipeline.return_value = mock_pipeline
+        with patch('mage_ai.data_preparation.models.pipeline.Pipeline.get', mock_get_pipeline):
+            logger_manager.delete_old_logs()
         # Assert that no files were deleted
         self.assertEqual(logger_manager.storage.remove_dir_calls, [])
 
@@ -34,7 +42,13 @@ class LoggerManagerTest(TestCase):
         # Mock the storage object to capture the calls to remove_dir
         logger_manager.storage = MockStorage()
         mock_old_log_folder = self.__create_log_dir(logger_manager, mock_pipeline_uuid)
-        logger_manager.delete_old_logs()
+        # Mock the Pipeline.get to return a pipeline with type PipelineType.PYTHON
+        mock_pipeline = Mock(spec=Pipeline)
+        mock_pipeline.type = PipelineType.PYTHON
+        mock_get_pipeline = Mock(spec=Pipeline.get)
+        mock_get_pipeline.return_value = mock_pipeline
+        with patch('mage_ai.data_preparation.models.pipeline.Pipeline.get', mock_get_pipeline):
+            logger_manager.delete_old_logs()
         # Assert that the folder with old log files was deleted
         self.assertIn(mock_old_log_folder, logger_manager.storage.remove_dir_calls)
 
@@ -74,8 +88,13 @@ class LoggerManagerTest(TestCase):
                 trigger_id=mock_pipeline_config['trigger_id'],
             )
             mock_pipeline_config['log_folder'] = mock_log_folder
-
-        logger_manager.delete_old_logs()
+        # Mock the Pipeline.get to return a pipeline with type PipelineType.PYTHON
+        mock_pipeline = Mock(spec=Pipeline)
+        mock_pipeline.type = PipelineType.PYTHON
+        mock_get_pipeline = Mock(spec=Pipeline.get)
+        mock_get_pipeline.return_value = mock_pipeline
+        with patch('mage_ai.data_preparation.models.pipeline.Pipeline.get', mock_get_pipeline):
+            logger_manager.delete_old_logs()
 
         # Assert that the folders with old log files were deleted for each pipeline
         for mock_pipeline_config in mock_pipeline_configs:

mage_ai/tests/data_preparation/models/block/dynamic/test_counter.py

@@ -1,14 +1,12 @@
 import os
 from unittest.mock import patch
 
-from mage_ai.data_preparation.models.block.dynamic.constants import (
-    CHILD_DATA_VARIABLE_UUID,
-)
 from mage_ai.data_preparation.models.block.dynamic.counter import (
     DynamicBlockItemCounter,
     DynamicChildItemCounter,
     DynamicDuoItemCounter,
 )
+from mage_ai.data_preparation.models.constants import CHILD_DATA_VARIABLE_UUID
 from mage_ai.data_preparation.models.variables.constants import (
     VariableAggregateDataTypeFilename,
 )

mage_ai/tests/data_preparation/models/block/hook/test_hook_block.py

@@ -5,7 +5,7 @@ from faker import Faker
 from mage_ai.data_preparation.models.block.hook.block import HookBlock
 from mage_ai.data_preparation.models.constants import BlockType
 from mage_ai.data_preparation.models.global_hooks.models import HookStatus, HookStrategy
-from mage_ai.shared.constants import ENV_TEST
+from mage_ai.shared.constants import ENV_TEST_MAGE
 from mage_ai.tests.api.operations.test_base import BaseApiTestCase
 from mage_ai.tests.factory import create_pipeline_with_blocks
 from mage_ai.tests.shared.mixins import build_hooks
@@ -46,7 +46,7 @@ class HookBlockTest(BaseApiTestCase):
                 check_status=True,
                 configuration={},
                 context={},
-                env=ENV_TEST,
+                env=ENV_TEST_MAGE,
                 error_on_failure=True,
                 mage=1,
                 pipeline_uuid='',
@@ -85,7 +85,7 @@ class HookBlockTest(BaseApiTestCase):
                 check_status=True,
                 configuration={},
                 context={},
-                env=ENV_TEST,
+                env=ENV_TEST_MAGE,
                 error_on_failure=True,
                 mage=1,
                 pipeline_uuid='',

mage_ai/tests/data_preparation/models/test_block.py

@@ -746,6 +746,13 @@ def test_output(output, *args) -> None:
             ),
         )
 
+    def test_none_block_type(self):
+        block = Block(name='test block', uuid='test uuid', block_type=None)
+        self.assertEqual(
+            block.get_typed_content('test content'),
+            '@None\ntest content',
+        )
+
 
 class BlockProjectPlatformTests(ProjectPlatformMixin):
     def test_file_path(self):

mage_ai/tests/data_preparation/models/test_pipeline.py

@@ -1,4 +1,5 @@
 import asyncio
+import importlib
 import json
 import os
 import shutil
@@ -32,6 +33,57 @@ class PipelineTest(AsyncDBTestCase):
         self.assertTrue(os.path.exists(os.path.join(
             self.repo_path, 'pipelines', 'test_pipeline', 'metadata.yaml')))
 
+    def test_pipeline_in_one_process_true(self):
+        with (patch.dict('os.environ', {'RUN_PIPELINE_IN_ONE_PROCESS': 'True'})):
+            import mage_ai.data_preparation.models.pipeline as pipeline_mod
+            import mage_ai.settings.server as server_mod
+
+            importlib.reload(server_mod)
+            importlib.reload(pipeline_mod)
+
+            pipeline = Pipeline.create(
+                'test pipeline one process true',
+                repo_path=self.repo_path,
+            )
+            self.assertEqual(pipeline.run_pipeline_in_one_process, True)
+
+        importlib.reload(server_mod)
+        importlib.reload(pipeline_mod)
+
+    def test_pipeline_in_one_process_false(self):
+        with (patch.dict('os.environ', {'RUN_PIPELINE_IN_ONE_PROCESS': 'False'})):
+            import mage_ai.data_preparation.models.pipeline as pipeline_mod
+            import mage_ai.settings.server as server_mod
+
+            importlib.reload(server_mod)
+            importlib.reload(pipeline_mod)
+
+            pipeline = Pipeline.create(
+                'test pipeline one process false',
+                repo_path=self.repo_path,
+            )
+            self.assertEqual(pipeline.run_pipeline_in_one_process, False)
+
+        importlib.reload(server_mod)
+        importlib.reload(pipeline_mod)
+
+    def test_pipeline_in_one_process_invalid_string(self):
+        with (patch.dict('os.environ', {'RUN_PIPELINE_IN_ONE_PROCESS': 'InvalidString'})):
+            import mage_ai.data_preparation.models.pipeline as pipeline_mod
+            import mage_ai.settings.server as server_mod
+
+            importlib.reload(server_mod)
+            importlib.reload(pipeline_mod)
+
+            pipeline = Pipeline.create(
+                'test pipeline one process invalid',
+                repo_path=self.repo_path,
+            )
+            self.assertEqual(pipeline.run_pipeline_in_one_process, False)
+
+        importlib.reload(server_mod)
+        importlib.reload(pipeline_mod)
+
     @freeze_time('2023-08-01 08:08:24')
     def test_add_block(self):
         self.__create_pipeline_with_blocks('test pipeline 2')
@@ -637,6 +689,7 @@ class PipelineTest(AsyncDBTestCase):
 
     @freeze_time('2023-08-01 08:08:24')
     def test_save_and_get_data_integration_catalog(self):
+        self.maxDiff = None
         pipeline = self.__create_pipeline_with_integration('test_pipeline_9')
         pipeline.save()
         catalog_config_path = os.path.join(
@@ -845,6 +898,7 @@ class PipelineTest(AsyncDBTestCase):
             pipeline_type=PipelineType.INTEGRATION,
             repo_path=self.repo_path,
         )
+
         source_block = Block.create(
             'source_block',
             'data_loader',
@@ -869,6 +923,7 @@ class PipelineTest(AsyncDBTestCase):
                 ],
             },
         }
+
         return pipeline
 
     def __create_dummy_data_loader_block(self, name, pipeline):

mage_ai/tests/data_preparation/models/test_variable.py

@@ -180,6 +180,7 @@ class VariableTest(DBTestCase):
                 [1, 'test'],
                 [2, 'test2'],
             ],
+            orient="row",
             schema=['col1', 'col2'],
         )
         df2 = pl.DataFrame(
@@ -187,6 +188,7 @@ class VariableTest(DBTestCase):
                 [1, 'test', 3.123, 41414123123124],
                 [2, 'test2', 4.321, 12111111],
             ],
+            orient="row",
             schema=['col1', 'col2', 'col3', 'col4'],
         )
         df2 = df2.cast({'col4': pl.Int64})

mage_ai/tests/data_preparation/test_repo_manager.py

@@ -10,13 +10,10 @@ from mage_ai.data_preparation.repo_manager import (
     get_project_uuid,
     get_repo_config,
     init_project_uuid,
-    init_repo,
     set_project_uuid_from_metadata,
 )
 from mage_ai.settings.repo import MAGE_DATA_DIR_ENV_VAR
-from mage_ai.settings.utils import base_repo_path
 from mage_ai.tests.base_test import DBTestCase
-from mage_ai.tests.shared.mixins import ProjectPlatformMixin
 
 
 def mock_uuid_value():
@@ -134,63 +131,3 @@ class RepoManagerTest(DBTestCase):
         self.assertEqual(get_project_uuid(), '000000')
 
         os.remove(metadata_path)
-
-
-@patch('mage_ai.settings.platform.project_platform_activated', lambda: True)
-@patch('mage_ai.settings.repo.project_platform_activated', lambda: True)
-class RepoManagerProjectPlatformTest(ProjectPlatformMixin):
-    def test_init(self):
-        repo = RepoConfig(root_project=False)
-        self.assertFalse(repo.root_project)
-        self.assertEqual(repo.repo_path, os.path.join(base_repo_path(), 'mage_platform'))
-        self.assertEqual(repo.variables_dir, os.path.join(base_repo_path(), 'mage_platform'))
-
-        repo = RepoConfig(root_project=True)
-        self.assertTrue(repo.root_project)
-        self.assertEqual(repo.repo_path, base_repo_path())
-        self.assertEqual(repo.variables_dir, base_repo_path())
-
-    def test_from_dict(self):
-        repo = RepoConfig.from_dict(dict(), root_project=False)
-        self.assertFalse(repo.root_project)
-        self.assertEqual(repo.repo_path, os.path.join(base_repo_path(), 'mage_platform'))
-        self.assertEqual(repo.variables_dir, os.path.join(base_repo_path(), 'mage_platform'))
-
-        repo = RepoConfig.from_dict(dict(), root_project=True)
-        self.assertTrue(repo.root_project)
-        self.assertEqual(repo.repo_path, base_repo_path())
-        self.assertEqual(repo.variables_dir, base_repo_path())
-
-    def test_metadata_path(self):
-        repo = RepoConfig(root_project=False)
-        self.assertEqual(
-            repo.metadata_path, os.path.join(base_repo_path(), 'mage_platform/metadata.yaml'),
-        )
-
-        repo = RepoConfig(root_project=True)
-        self.assertEqual(repo.metadata_path, os.path.join(base_repo_path(), 'metadata.yaml'))
-
-    def test_get_repo_config(self):
-        repo = get_repo_config(root_project=False)
-        self.assertFalse(repo.root_project)
-        self.assertEqual(repo.repo_path, os.path.join(base_repo_path(), 'mage_platform'))
-        self.assertEqual(repo.variables_dir, os.path.join(base_repo_path(), 'mage_platform'))
-
-        repo = get_repo_config(root_project=True)
-        self.assertTrue(repo.root_project)
-        self.assertEqual(repo.repo_path, base_repo_path())
-        self.assertEqual(repo.variables_dir, base_repo_path())
-
-    def test_init_repo(self):
-        path = os.path.join(os.path.dirname(base_repo_path()), 'test2')
-        try:
-            shutil.rmtree(path)
-        except Exception:
-            pass
-        with patch('mage_ai.data_preparation.repo_manager.get_repo_config') as mock_get_repo_config:
-            init_repo(path, root_project=True)
-            mock_get_repo_config.assert_called_once_with(path, root_project=True)
-        try:
-            shutil.rmtree(path)
-        except Exception:
-            pass

mage_ai/tests/data_preparation/test_repo_manager_project_platform.py

@@ -0,0 +1,67 @@
+import os
+import shutil
+from unittest.mock import patch
+
+from mage_ai.data_preparation.repo_manager import RepoConfig, get_repo_config, init_repo
+from mage_ai.settings.utils import base_repo_path
+from mage_ai.tests.shared.mixins import ProjectPlatformMixin
+
+
+@patch('mage_ai.settings.platform.project_platform_activated', lambda: True)
+@patch('mage_ai.settings.repo.project_platform_activated', lambda: True)
+class RepoManagerProjectPlatformTest(ProjectPlatformMixin):
+    def test_init(self):
+        repo = RepoConfig(root_project=False)
+        self.assertFalse(repo.root_project)
+        self.assertEqual(repo.repo_path, os.path.join(base_repo_path(), 'mage_platform'))
+        self.assertEqual(repo.variables_dir, os.path.join(base_repo_path(), 'mage_platform'))
+
+        repo = RepoConfig(root_project=True)
+        self.assertTrue(repo.root_project)
+        self.assertEqual(repo.repo_path, base_repo_path())
+        self.assertEqual(repo.variables_dir, base_repo_path())
+
+    def test_from_dict(self):
+        repo = RepoConfig.from_dict(dict(), root_project=False)
+        self.assertFalse(repo.root_project)
+        self.assertEqual(repo.repo_path, os.path.join(base_repo_path(), 'mage_platform'))
+        self.assertEqual(repo.variables_dir, os.path.join(base_repo_path(), 'mage_platform'))
+
+        repo = RepoConfig.from_dict(dict(), root_project=True)
+        self.assertTrue(repo.root_project)
+        self.assertEqual(repo.repo_path, base_repo_path())
+        self.assertEqual(repo.variables_dir, base_repo_path())
+
+    def test_metadata_path(self):
+        repo = RepoConfig(root_project=False)
+        self.assertEqual(
+            repo.metadata_path, os.path.join(base_repo_path(), 'mage_platform/metadata.yaml'),
+        )
+
+        repo = RepoConfig(root_project=True)
+        self.assertEqual(repo.metadata_path, os.path.join(base_repo_path(), 'metadata.yaml'))
+
+    def test_get_repo_config(self):
+        repo = get_repo_config(root_project=False)
+        self.assertFalse(repo.root_project)
+        self.assertEqual(repo.repo_path, os.path.join(base_repo_path(), 'mage_platform'))
+        self.assertEqual(repo.variables_dir, os.path.join(base_repo_path(), 'mage_platform'))
+
+        repo = get_repo_config(root_project=True)
+        self.assertTrue(repo.root_project)
+        self.assertEqual(repo.repo_path, base_repo_path())
+        self.assertEqual(repo.variables_dir, base_repo_path())
+
+    def test_init_repo(self):
+        path = os.path.join(os.path.dirname(base_repo_path()), 'test2')
+        try:
+            shutil.rmtree(path)
+        except Exception:
+            pass
+        with patch('mage_ai.data_preparation.repo_manager.get_repo_config') as mock_get_repo_config:
+            init_repo(path, root_project=True)
+            mock_get_repo_config.assert_called_once_with(path, root_project=True)
+        try:
+            shutil.rmtree(path)
+        except Exception:
+            pass