revert deleted migration test

eadwinCode · eadwinCode · commit c57783c3344e · 2024-06-08T13:28:29.000+01:00
diff --git a/tests/test_migrations/__init__.py b/tests/test_migrations/__init__.py
diff --git a/tests/test_migrations/samples/custom_directory.py b/tests/test_migrations/samples/custom_directory.py
@@ -0,0 +1,46 @@
+#!/bin/env python
+
+import click
+from ellar.app import AppFactory, current_injector
+from ellar.utils.importer import get_main_directory_by_stack
+from ellar_cli.main import create_ellar_cli
+from models import User
+
+from ellar_sql import EllarSQLModule, MigrationOption, model
+
+
+def bootstrap():
+    path = get_main_directory_by_stack(
+        "__main__/__parent__/__parent__/dumbs/custom_directory", stack_level=1
+    )
+    application = AppFactory.create_app(
+        modules=[
+            EllarSQLModule.setup(
+                databases="sqlite:///app.db",
+                migration_options=MigrationOption(
+                    directory="temp_migrations",
+                    context_configure={"compare_types": True},
+                ),
+                root_path=str(path),
+            )
+        ]
+    )
+    return application
+
+
+cli = create_ellar_cli("custom_directory:bootstrap")
+
+
+@cli.command()
+def add_user():
+    session = current_injector.get(model.Session)
+    user = User(name="Custom Directory App Ellar")
+    session.add(user)
+
+    session.commit()
+
+    click.echo(f"<User name={user.name} id={user.id}>")
+
+
+if __name__ == "__main__":
+    cli()
diff --git a/tests/test_migrations/samples/default.py b/tests/test_migrations/samples/default.py
@@ -0,0 +1,41 @@
+#!/bin/env python
+import click
+from ellar.app import AppFactory, current_injector
+from ellar.utils.importer import get_main_directory_by_stack
+from ellar_cli.main import create_ellar_cli
+from models import User
+
+from ellar_sql import EllarSQLModule, model
+
+
+def bootstrap():
+    path = get_main_directory_by_stack(
+        "__main__/__parent__/__parent__/dumbs/default", stack_level=1
+    )
+    application = AppFactory.create_app(
+        modules=[
+            EllarSQLModule.setup(
+                databases="sqlite:///app.db",
+                migration_options={"context_configure": {"compare_types": False}},
+                root_path=str(path),
+            )
+        ]
+    )
+    return application
+
+
+cli = create_ellar_cli("default:bootstrap")
+
+
+@cli.command()
+def add_user():
+    session = current_injector.get(model.Session)
+    user = User(name="default App Ellar")
+    session.add(user)
+
+    session.commit()
+    click.echo(f"<User name={user.name} id={user.id}>")
+
+
+if __name__ == "__main__":
+    cli()
diff --git a/tests/test_migrations/samples/default_async.py b/tests/test_migrations/samples/default_async.py
@@ -0,0 +1,46 @@
+#!/bin/env python
+import ellar_cli.click as click
+from ellar.app import AppFactory, current_injector
+from ellar.utils.importer import get_main_directory_by_stack
+from ellar_cli.main import create_ellar_cli
+from models import User
+from sqlalchemy.ext.asyncio import AsyncSession
+
+from ellar_sql import EllarSQLModule
+
+
+def bootstrap():
+    path = get_main_directory_by_stack(
+        "__main__/__parent__/__parent__/dumbs/default_async", stack_level=1
+    )
+    application = AppFactory.create_app(
+        modules=[
+            EllarSQLModule.setup(
+                databases="sqlite+aiosqlite:///app.db",
+                migration_options={"context_configure": {"compare_types": False}},
+                root_path=str(path),
+            )
+        ]
+    )
+    return application
+
+
+cli = create_ellar_cli("default_async:bootstrap")
+
+
+@cli.command()
+@click.run_as_async
+async def add_user():
+    session = current_injector.get(AsyncSession)
+    user = User(name="default App Ellar")
+    session.add(user)
+
+    await session.commit()
+    await session.refresh(user)
+    await session.close()
+
+    click.echo(f"<User name={user.name} id={user.id}>")
+
+
+if __name__ == "__main__":
+    cli()
diff --git a/tests/test_migrations/samples/models.py b/tests/test_migrations/samples/models.py
@@ -0,0 +1,24 @@
+import os
+
+from ellar_sql import model
+
+
+class Base(model.Model):
+    __base_config__ = {"as_base": True}
+
+
+class User(Base):
+    id = model.Column(model.Integer, primary_key=True)
+
+    if os.environ.get("model_change_name"):
+        name = model.Column(model.String(128))
+    else:
+        name = model.Column(model.String(256))
+
+
+if os.environ.get("multiple_db"):
+
+    class Group(Base):
+        __database__ = "db1"
+        id = model.Column(model.Integer, primary_key=True)
+        name = model.Column(model.String(128))
diff --git a/tests/test_migrations/samples/multiple_database.py b/tests/test_migrations/samples/multiple_database.py
@@ -0,0 +1,49 @@
+#!/bin/env python
+import click
+from ellar.app import AppFactory, current_injector
+from ellar.utils.importer import get_main_directory_by_stack
+from ellar_cli.main import create_ellar_cli
+from models import Group, User
+
+from ellar_sql import EllarSQLModule, model
+
+
+def bootstrap():
+    path = get_main_directory_by_stack(
+        "__main__/__parent__/__parent__/dumbs/multiple", stack_level=1
+    )
+    application = AppFactory.create_app(
+        modules=[
+            EllarSQLModule.setup(
+                databases={
+                    "default": "sqlite:///app.db",
+                    "db1": "sqlite:///app2.db",
+                },
+                migration_options={"context_configure": {"compare_types": False}},
+                root_path=str(path),
+            )
+        ]
+    )
+    return application
+
+
+cli = create_ellar_cli("multiple_database:bootstrap")
+
+
+@cli.command()
+def add_user():
+    session = current_injector.get(model.Session)
+    user = User(name="Multiple Database App Ellar")
+    group = Group(name="group")
+    session.add(user)
+    session.add(group)
+
+    session.commit()
+
+    click.echo(
+        f"<User name={user.name} id={user.id} and Group name={group.name} id={group.id}>"
+    )
+
+
+if __name__ == "__main__":
+    cli()
diff --git a/tests/test_migrations/samples/multiple_database_async.py b/tests/test_migrations/samples/multiple_database_async.py
@@ -0,0 +1,57 @@
+#!/bin/env python
+import ellar_cli.click as click
+from ellar.app import AppFactory, current_injector
+from ellar.utils.importer import get_main_directory_by_stack
+from ellar_cli.main import create_ellar_cli
+from models import Group, User
+from sqlalchemy.ext.asyncio import AsyncSession
+
+from ellar_sql import EllarSQLModule
+
+
+def bootstrap():
+    path = get_main_directory_by_stack(
+        "__main__/__parent__/__parent__/dumbs/multiple_async", stack_level=1
+    )
+    application = AppFactory.create_app(
+        modules=[
+            EllarSQLModule.setup(
+                databases={
+                    "default": "sqlite+aiosqlite:///app.db",
+                    "db1": "sqlite+aiosqlite:///app2.db",
+                },
+                migration_options={"context_configure": {"compare_types": False}},
+                root_path=str(path),
+            )
+        ]
+    )
+    return application
+
+
+cli = create_ellar_cli("multiple_database_async:bootstrap")
+
+
+@cli.command()
+@click.run_as_async
+async def add_user():
+    session = current_injector.get(AsyncSession)
+    user = User(name="Multiple Database App Ellar")
+    group = Group(name="group")
+
+    session.add(user)
+    session.add(group)
+
+    await session.commit()
+
+    await session.refresh(user)
+    await session.refresh(group)
+
+    await session.close()
+
+    click.echo(
+        f"<User name={user.name} id={user.id} and Group name={group.name} id={group.id}>"
+    )
+
+
+if __name__ == "__main__":
+    cli()
diff --git a/tests/test_migrations/test_migrations_commands.py b/tests/test_migrations/test_migrations_commands.py
diff --git a/tests/test_migrations/test_multiple_database.py b/tests/test_migrations/test_multiple_database.py
