Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 3 additions & 0 deletions .gitignore
Original file line number Diff line number Diff line change
@@ -0,0 +1,3 @@
__pycache__/
*.pyc
.pytest_cache/
19 changes: 17 additions & 2 deletions task.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,10 +10,25 @@
from commands.done import mark_done


DEFAULT_CONFIG = """# Task storage settings
storage:
format: json
max_tasks: 1000

# Display settings
display:
color: true
unicode: true
"""


def load_config():
"""Load configuration from file."""
"""Load configuration from file. Creates default config if missing."""
config_path = Path.home() / ".config" / "task-cli" / "config.yaml"
# NOTE: This will crash if config doesn't exist - known bug for bounty testing
if not config_path.exists():
config_path.parent.mkdir(parents=True, exist_ok=True)
config_path.write_text(DEFAULT_CONFIG)
print(f"Created default config at {config_path}")
with open(config_path) as f:
return f.read()

Expand Down
29 changes: 29 additions & 0 deletions test_task.py
Original file line number Diff line number Diff line change
@@ -1,10 +1,13 @@
"""Basic tests for task CLI."""

import json
import os
import shutil
import pytest
from pathlib import Path
from commands.add import add_task, validate_description
from commands.done import validate_task_id
from task import load_config


def test_validate_description():
Expand All @@ -28,3 +31,29 @@ def test_validate_task_id():

with pytest.raises(ValueError):
validate_task_id(tasks, 99)


def test_load_config_missing_creates_default():
"""Test that load_config creates default config when file is missing."""
config_dir = Path.home() / ".config" / "task-cli"
config_path = config_dir / "config.yaml"
backup = None

try:
# Back up existing config if present
if config_path.exists():
backup = config_path.read_text()
config_path.unlink()

# load_config should create default config, not crash
result = load_config()
assert config_path.exists()
assert "storage:" in result
assert "format: json" in result
assert "display:" in result
finally:
# Restore original config
if backup is not None:
config_path.write_text(backup)
elif config_path.exists():
config_path.unlink()