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
31 changes: 28 additions & 3 deletions task.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,10 +10,32 @@
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."""
config_path = Path.home() / ".config" / "task-cli" / "config.yaml"
# NOTE: This will crash if config doesn't exist - known bug for bounty testing
"""Load configuration from file.

Creates a default config if the file doesn't exist.
Returns the config content as a string.
"""
config_dir = Path.home() / ".config" / "task-cli"
config_path = config_dir / "config.yaml"

if not config_path.exists():
config_dir.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 All @@ -35,6 +57,9 @@ def main():

args = parser.parse_args()

# Load config (creates default if missing)
load_config()

if args.command == "add":
add_task(args.description)
elif args.command == "list":
Expand Down
36 changes: 36 additions & 0 deletions test_task.py
Original file line number Diff line number Diff line change
@@ -1,10 +1,15 @@
"""Basic tests for task CLI."""

import json
import os
import shutil
import tempfile

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, DEFAULT_CONFIG


def test_validate_description():
Expand All @@ -28,3 +33,34 @@ def test_validate_task_id():

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


def test_load_config_creates_default_when_missing(tmp_path, monkeypatch):
"""Test that load_config creates default config when file is missing."""
monkeypatch.setattr(Path, "home", lambda: tmp_path)

config = load_config()

config_path = tmp_path / ".config" / "task-cli" / "config.yaml"
assert config_path.exists()
assert "storage:" in config
assert "display:" in config


def test_load_config_reads_existing(tmp_path, monkeypatch):
"""Test that load_config reads existing config file."""
monkeypatch.setattr(Path, "home", lambda: tmp_path)

config_dir = tmp_path / ".config" / "task-cli"
config_dir.mkdir(parents=True)
(config_dir / "config.yaml").write_text("custom: true\n")

config = load_config()
assert "custom: true" in config


def test_default_config_is_valid_yaml():
"""Test that DEFAULT_CONFIG is valid YAML-like content."""
assert "storage:" in DEFAULT_CONFIG
assert "display:" in DEFAULT_CONFIG
assert "format: json" in DEFAULT_CONFIG