django-components/tests/test_command_create.py
Juro Oravec d3d2d0ab08
feat: allow extensions to add commands (#1017)
* feat: allow extensions to add commands

* refactor: fix tests

* refactor: more test fix

* refactor: more test fixes

* refactor: more linter fixes
2025-03-16 12:03:16 +01:00

153 lines
4.3 KiB
Python

import os
import tempfile
from io import StringIO
from shutil import rmtree
from unittest.mock import patch
import pytest
from django.core.management import call_command
from django.core.management.base import CommandError
from django_components.testing import djc_test
from .testutils import setup_test_config
setup_test_config({"autodiscover": False})
@djc_test
class TestCreateComponentCommand:
def test_default_file_names(self):
temp_dir = tempfile.mkdtemp()
component_name = "defaultcomponent"
call_command("components", "create", component_name, "--path", temp_dir)
expected_files = [
os.path.join(temp_dir, component_name, "script.js"),
os.path.join(temp_dir, component_name, "style.css"),
os.path.join(temp_dir, component_name, "template.html"),
]
for file_path in expected_files:
assert os.path.exists(file_path)
rmtree(temp_dir)
def test_nondefault_creation(self):
temp_dir = tempfile.mkdtemp()
component_name = "testcomponent"
call_command(
"components",
"create",
component_name,
"--path",
temp_dir,
"--js",
"test.js",
"--css",
"test.css",
"--template",
"test.html",
)
expected_files = [
os.path.join(temp_dir, component_name, "test.js"),
os.path.join(temp_dir, component_name, "test.css"),
os.path.join(temp_dir, component_name, "test.html"),
os.path.join(temp_dir, component_name, f"{component_name}.py"),
]
for file_path in expected_files:
assert os.path.exists(file_path), f"File {file_path} was not created"
rmtree(temp_dir)
def test_dry_run(self):
temp_dir = tempfile.mkdtemp()
component_name = "dryruncomponent"
call_command(
"components",
"create",
component_name,
"--path",
temp_dir,
"--dry-run",
)
component_path = os.path.join(temp_dir, component_name)
assert not os.path.exists(component_path)
rmtree(temp_dir)
def test_force_overwrite(self):
temp_dir = tempfile.mkdtemp()
component_name = "existingcomponent"
component_path = os.path.join(temp_dir, component_name)
os.makedirs(component_path)
with open(os.path.join(component_path, f"{component_name}.py"), "w") as f:
f.write("hello world")
call_command(
"components",
"create",
component_name,
"--path",
temp_dir,
"--force",
)
with open(os.path.join(component_path, f"{component_name}.py"), "r") as f:
assert "hello world" not in f.read()
rmtree(temp_dir)
def test_error_existing_component_no_force(self):
temp_dir = tempfile.mkdtemp()
component_name = "existingcomponent_2"
component_path = os.path.join(temp_dir, component_name)
os.makedirs(component_path)
with pytest.raises(CommandError):
call_command("components", "create", component_name, "--path", temp_dir)
rmtree(temp_dir)
def test_verbose_output(self):
temp_dir = tempfile.mkdtemp()
component_name = "verbosecomponent"
out = StringIO()
with patch("sys.stdout", new=out):
call_command(
"components",
"create",
component_name,
"--path",
temp_dir,
"--verbose",
stdout=out,
)
output = out.getvalue()
assert "component at" in output
rmtree(temp_dir)
# TODO_V1 - REMOVE - deprecated
def test_startcomponent(self):
temp_dir = tempfile.mkdtemp()
component_name = "defaultcomponent"
call_command("startcomponent", component_name, "--path", temp_dir)
expected_files = [
os.path.join(temp_dir, component_name, "script.js"),
os.path.join(temp_dir, component_name, "style.css"),
os.path.join(temp_dir, component_name, "template.html"),
]
for file_path in expected_files:
assert os.path.exists(file_path)
rmtree(temp_dir)