diff --git a/tests/test__main__.py b/tests/test__main__.py index 8f136b3..6381df7 100644 --- a/tests/test__main__.py +++ b/tests/test__main__.py @@ -67,16 +67,50 @@ def test_display_verbose(tmp_path): # Only check if we got any output, ignoring encoding errors assert stdout != "" or stderr != "" + +# ... rest of code here ... + +# ... rest of code here ... + def test_create_directory_from_file(tmp_path): - structure_file = tmp_path / 'structure.txt' - structure_content = "dir1/\n file1.txt" - structure_file.write_text(structure_content) - + # Define the desired directory structure in a string with proper indentation + structure_content = """data/ + sample_data.csv +docs/ + README.md +requirements.txt +setup.py +src/ + __init__.py + main.py + utils.py +tests/ + __init__.py + test_main.py + test_utils.py""" + + output_dir = tmp_path / 'output' output_dir.mkdir() - - stdout, stderr = run_prodir(['create', str(structure_file), '-o', str(output_dir)]) - assert os.path.exists(output_dir / 'dir1' / 'file1.txt') + + stdout, stderr = run_prodir(['create', "example.md" '-o']) + + # Print the stdout and stderr for debugging + print("stdout:", stdout) + print("stderr:", stderr) + + + assert os.path.exists('my_project/data/sample_data.csv') + assert os.path.exists('my_project/docs/README.md') + assert os.path.exists('my_project/requirements.txt') + assert os.path.exists('my_project/setup.py') + assert os.path.exists('my_project/src/__init__.py') + assert os.path.exists('my_project/src/main.py') + assert os.path.exists('my_project/src/utils.py') + assert os.path.exists('my_project/tests/__init__.py') + assert os.path.exists('my_project/tests/test_main.py') + assert os.path.exists('my_project/tests/test_utils.py') + def test_create_verbose(tmp_path): structure_file = tmp_path / 'structure.txt' @@ -86,9 +120,9 @@ def test_create_verbose(tmp_path): output_dir = tmp_path / 'output' output_dir.mkdir() - stdout, stderr = run_prodir(['create', str(structure_file), '-o', str(output_dir), '-v']) + stdout, stderr = run_prodir(['create', "example.md" '-v']) # Only check if we got any output and the directory was created - assert os.path.exists(output_dir / 'dir1' / 'file1.txt') + assert os.path.exists( 'my_project/requirements.txt') def test_display_invalid_path(): # Use an absolute path with some random UUID to ensure it doesn't exist