from scripttease.lib.commands.django import *


def test_django():
    c = django("check")
    assert "./manage.py check" in c.get_statement()

    c = django("collectstatic")
    assert "./manage.py collectstatic" in c.get_statement()

    c = django("migrate")
    assert "./manage.py migrate" in c.get_statement()

    c = django("custom", "arg1", "arg2", venv="python", settings="tenants.example.settings", quiet=True)
    s = c.get_statement()
    assert "./manage.py custom" in s
    assert "arg1" in s
    assert "arg2" in s
    assert "--settings=" in s
    assert "source python/bin/activate" in s
    assert "--quiet" in s


def test_django_check():
    c = django_check(venv="python")
    s = c.get_statement()
    assert "./manage.py check" in s
    assert "source python/bin/activate" in s


def test_django_collect_static():
    c = django_static(venv="python")
    s = c.get_statement()
    assert "./manage.py collectstatic" in s
    assert "source python/bin/activate" in s


def test_django_createsuperuser():
    c = django_createsuperuser("root", email="root@example.com")
    s = c.get_statement()
    assert "./manage.py createsuperuser" in s
    assert '--username="root"' in s
    assert '--email="root@example.com"' in s
    assert '--noinput' in s


def test_django_dumpdata():
    c = django_dump("projects")
    s = c.get_statement()
    assert "./manage.py dumpdata" in s
    assert "projects >" in s
    assert '--format="json"' in s
    assert "--indent=4" in s
    assert "../fixtures/projects/initial.json" in s

    c = django_dump("projects.Category")
    s = c.get_statement()
    assert "./manage.py dumpdata" in s
    assert "projects.Category >" in s
    assert '--format="json"' in s
    assert "--indent=4" in s
    assert "../fixtures/projects/category.json" in s


def test_django_loaddata():
    c = django_load("projects")
    s = c.get_statement()
    assert "./manage.py loaddata" in s
    assert "../fixtures/projects/initial.json" in s

    c = django_load("projects.Category")
    s = c.get_statement()
    assert "./manage.py loaddata" in s
    assert "../fixtures/projects/category.json" in s


def test_django_migrate():
    c = django_migrate(cd="/path/to/project/", venv="python")
    s = c.get_statement(cd=True)
    assert "./manage.py migrate" in s
    assert "source python/bin/activate" in s
    assert "cd /path/to/project/" in s