knacky/mimic
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(c2): integrate Mythic command and control (sprint 8) #11

Merged
knacky merged 16 commits from sprint/8-c2 into main 2026-06-11 10:29:19 +00:00
Conversation 0 Commits 16 Files Changed 60 +4 -9
1 changed files with 4 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit b83316f715 - Show all commits

13
backend/tests/test_migration_0006_c2.py
View File

@@ -92,27 +92,22 @@ def _run_upgrade(engine, migration_mod):
with engine.begin() as conn:
ctx = MigrationContext.configure(conn)
ops = Operations(ctx)
# Patch op module for the migration
import alembic.op as alembic_op
original_proxy = alembic_op._proxy # type: ignore[attr-defined]
alembic_op._proxy = ops # type: ignore[attr-defined]
ops._install_proxy() # type: ignore[attr-defined]
try:
migration_mod.upgrade()
finally:
alembic_op._proxy = original_proxy # type: ignore[attr-defined]
ops._remove_proxy() # type: ignore[attr-defined]
def _run_downgrade(engine, migration_mod):
with engine.begin() as conn:
ctx = MigrationContext.configure(conn)
ops = Operations(ctx)
import alembic.op as alembic_op
original_proxy = alembic_op._proxy # type: ignore[attr-defined]
alembic_op._proxy = ops # type: ignore[attr-defined]
ops._install_proxy() # type: ignore[attr-defined]
try:
migration_mod.downgrade()
finally:
alembic_op._proxy = original_proxy # type: ignore[attr-defined]
ops._remove_proxy() # type: ignore[attr-defined]
class TestMigration0006Upgrade: