test: Add comprehensive test suite for Issue #1 database functionality
Adds 9 comprehensive tests covering all aspects of Issue #1 implementation: **TestDatabaseInitialization (2 tests):** - DatabaseManager instantiation and configuration - Database schema creation with proper table structure **TestFrontMatterParsing (3 tests):** - FrontMatterParser instantiation - Parsing Issue #1 example content with YAML front matter - Handling markdown content without front matter **TestIntegratedWorkflow (2 tests):** - Complete end-to-end workflow with Issue #1 example file - Multiple file storage and retrieval validation **TestErrorHandling (2 tests):** - Graceful handling of invalid YAML front matter - Exception handling for invalid database paths All tests validate the exact requirements from Issue #1 specification, including the provided example markdown content with front matter. 🤖 Generated with [Claude Code](https://claude.ai/code) Co-Authored-By: Claude <noreply@anthropic.com>
This commit is contained in:
298
tests/test_issue_1_database_initialization.py
Normal file
298
tests/test_issue_1_database_initialization.py
Normal file
@@ -0,0 +1,298 @@
|
||||
"""
|
||||
Test database initialization and markdown storage functionality for Issue #1.
|
||||
|
||||
This test implements the requirements for initializing a SQLite database
|
||||
and storing markdown files with front matter parsing.
|
||||
|
||||
Issue #1: Initialize Database and Store Example Markdown File
|
||||
https://gitea.coulomb.social/coulomb/markitect_project/issues/1
|
||||
"""
|
||||
|
||||
import pytest
|
||||
import sqlite3
|
||||
import tempfile
|
||||
import os
|
||||
from pathlib import Path
|
||||
from datetime import datetime
|
||||
|
||||
# Import will fail initially - this is expected for TDD red phase
|
||||
try:
|
||||
from markitect.database import DatabaseManager
|
||||
from markitect.frontmatter import FrontMatterParser
|
||||
except ImportError:
|
||||
# Expected to fail initially - we'll implement these modules
|
||||
pass
|
||||
|
||||
|
||||
class TestDatabaseInitialization:
|
||||
"""Test database initialization and basic operations."""
|
||||
|
||||
def setup_method(self):
|
||||
"""Set up temporary database for each test."""
|
||||
self.temp_db = tempfile.NamedTemporaryFile(delete=False, suffix='.db')
|
||||
self.temp_db.close()
|
||||
self.db_path = self.temp_db.name
|
||||
|
||||
def teardown_method(self):
|
||||
"""Clean up temporary database after each test."""
|
||||
if os.path.exists(self.db_path):
|
||||
os.unlink(self.db_path)
|
||||
|
||||
def test_database_manager_creation(self):
|
||||
"""Test that DatabaseManager can be instantiated."""
|
||||
# This should fail initially (red phase)
|
||||
db_manager = DatabaseManager(self.db_path)
|
||||
assert db_manager is not None
|
||||
assert db_manager.db_path == self.db_path
|
||||
|
||||
def test_database_initialization_creates_tables(self):
|
||||
"""Test that database initialization creates the markdown_files table."""
|
||||
db_manager = DatabaseManager(self.db_path)
|
||||
db_manager.initialize_database()
|
||||
|
||||
# Verify database file was created
|
||||
assert os.path.exists(self.db_path)
|
||||
|
||||
# Verify table structure
|
||||
conn = sqlite3.connect(self.db_path)
|
||||
cursor = conn.cursor()
|
||||
|
||||
# Check if markdown_files table exists
|
||||
cursor.execute("""
|
||||
SELECT name FROM sqlite_master
|
||||
WHERE type='table' AND name='markdown_files'
|
||||
""")
|
||||
table_exists = cursor.fetchone()
|
||||
assert table_exists is not None
|
||||
|
||||
# Check table schema
|
||||
cursor.execute("PRAGMA table_info(markdown_files)")
|
||||
columns = cursor.fetchall()
|
||||
|
||||
expected_columns = {
|
||||
'id': 'INTEGER',
|
||||
'filename': 'TEXT',
|
||||
'front_matter': 'TEXT',
|
||||
'content': 'TEXT',
|
||||
'created_at': 'TIMESTAMP'
|
||||
}
|
||||
|
||||
actual_columns = {col[1]: col[2] for col in columns}
|
||||
|
||||
for col_name, col_type in expected_columns.items():
|
||||
assert col_name in actual_columns
|
||||
assert col_type.upper() in actual_columns[col_name].upper()
|
||||
|
||||
conn.close()
|
||||
|
||||
|
||||
class TestFrontMatterParsing:
|
||||
"""Test front matter parsing functionality."""
|
||||
|
||||
def test_front_matter_parser_creation(self):
|
||||
"""Test that FrontMatterParser can be instantiated."""
|
||||
parser = FrontMatterParser()
|
||||
assert parser is not None
|
||||
|
||||
def test_parse_example_markdown_with_frontmatter(self):
|
||||
"""Test parsing the specific example markdown from Issue #1."""
|
||||
example_content = '''---
|
||||
title: "My First Document"
|
||||
author: "Test User"
|
||||
created: "2024-01-01"
|
||||
tags: ["example", "test"]
|
||||
---
|
||||
|
||||
# Welcome to MarkiTect
|
||||
|
||||
This is an example markdown document with **bold text** and *italic text*.
|
||||
|
||||
## Features
|
||||
- Front matter parsing
|
||||
- Content storage
|
||||
- Database initialization
|
||||
|
||||
## Code Example
|
||||
```python
|
||||
print("Hello, MarkiTect!")
|
||||
```
|
||||
|
||||
The system should handle this gracefully.'''
|
||||
|
||||
parser = FrontMatterParser()
|
||||
front_matter, content = parser.parse(example_content)
|
||||
|
||||
# Verify front matter parsing
|
||||
assert isinstance(front_matter, dict)
|
||||
assert front_matter['title'] == "My First Document"
|
||||
assert front_matter['author'] == "Test User"
|
||||
assert front_matter['created'] == "2024-01-01"
|
||||
assert front_matter['tags'] == ["example", "test"]
|
||||
|
||||
# Verify content separation
|
||||
assert content.startswith('# Welcome to MarkiTect')
|
||||
assert 'title:' not in content # Front matter should be removed
|
||||
assert '---' not in content # YAML delimiters should be removed
|
||||
assert 'The system should handle this gracefully.' in content
|
||||
|
||||
def test_parse_markdown_without_frontmatter(self):
|
||||
"""Test parsing markdown without front matter."""
|
||||
content_only = '''# Simple Document
|
||||
|
||||
This is just markdown content without front matter.'''
|
||||
|
||||
parser = FrontMatterParser()
|
||||
front_matter, content = parser.parse(content_only)
|
||||
|
||||
assert front_matter == {}
|
||||
assert content == content_only
|
||||
|
||||
|
||||
class TestIntegratedWorkflow:
|
||||
"""Test the complete workflow from initialization to storage."""
|
||||
|
||||
def setup_method(self):
|
||||
"""Set up temporary database for each test."""
|
||||
self.temp_db = tempfile.NamedTemporaryFile(delete=False, suffix='.db')
|
||||
self.temp_db.close()
|
||||
self.db_path = self.temp_db.name
|
||||
|
||||
def teardown_method(self):
|
||||
"""Clean up temporary database after each test."""
|
||||
if os.path.exists(self.db_path):
|
||||
os.unlink(self.db_path)
|
||||
|
||||
def test_complete_workflow_example_file(self):
|
||||
"""Test the complete workflow with the example file from Issue #1."""
|
||||
# Initialize database
|
||||
db_manager = DatabaseManager(self.db_path)
|
||||
db_manager.initialize_database()
|
||||
|
||||
# Example content from Issue #1
|
||||
example_content = '''---
|
||||
title: "My First Document"
|
||||
author: "Test User"
|
||||
created: "2024-01-01"
|
||||
tags: ["example", "test"]
|
||||
---
|
||||
|
||||
# Welcome to MarkiTect
|
||||
|
||||
This is an example markdown document with **bold text** and *italic text*.
|
||||
|
||||
## Features
|
||||
- Front matter parsing
|
||||
- Content storage
|
||||
- Database initialization
|
||||
|
||||
## Code Example
|
||||
```python
|
||||
print("Hello, MarkiTect!")
|
||||
```
|
||||
|
||||
The system should handle this gracefully.'''
|
||||
|
||||
# Store the file
|
||||
result = db_manager.store_markdown_file("example.md", example_content)
|
||||
|
||||
# Verify storage was successful
|
||||
assert result is not None
|
||||
|
||||
# Verify data was stored correctly
|
||||
conn = sqlite3.connect(self.db_path)
|
||||
cursor = conn.cursor()
|
||||
|
||||
cursor.execute("SELECT * FROM markdown_files WHERE filename = ?", ("example.md",))
|
||||
row = cursor.fetchone()
|
||||
|
||||
assert row is not None
|
||||
assert row[1] == "example.md" # filename
|
||||
|
||||
# Verify front matter was stored as JSON
|
||||
import json
|
||||
stored_front_matter = json.loads(row[2])
|
||||
assert stored_front_matter['title'] == "My First Document"
|
||||
assert stored_front_matter['author'] == "Test User"
|
||||
|
||||
# Verify content was stored correctly (without front matter)
|
||||
stored_content = row[3]
|
||||
assert stored_content.startswith('# Welcome to MarkiTect')
|
||||
assert '---' not in stored_content
|
||||
|
||||
# Verify timestamp was set
|
||||
assert row[4] is not None # created_at
|
||||
|
||||
conn.close()
|
||||
|
||||
def test_store_multiple_files(self):
|
||||
"""Test storing multiple markdown files."""
|
||||
db_manager = DatabaseManager(self.db_path)
|
||||
db_manager.initialize_database()
|
||||
|
||||
# Store first file
|
||||
content1 = '''---
|
||||
title: "File One"
|
||||
---
|
||||
# First File'''
|
||||
db_manager.store_markdown_file("file1.md", content1)
|
||||
|
||||
# Store second file
|
||||
content2 = '''---
|
||||
title: "File Two"
|
||||
---
|
||||
# Second File'''
|
||||
db_manager.store_markdown_file("file2.md", content2)
|
||||
|
||||
# Verify both files were stored
|
||||
conn = sqlite3.connect(self.db_path)
|
||||
cursor = conn.cursor()
|
||||
|
||||
cursor.execute("SELECT COUNT(*) FROM markdown_files")
|
||||
count = cursor.fetchone()[0]
|
||||
assert count == 2
|
||||
|
||||
cursor.execute("SELECT filename FROM markdown_files ORDER BY filename")
|
||||
filenames = [row[0] for row in cursor.fetchall()]
|
||||
assert filenames == ["file1.md", "file2.md"]
|
||||
|
||||
conn.close()
|
||||
|
||||
|
||||
class TestErrorHandling:
|
||||
"""Test error handling scenarios."""
|
||||
|
||||
def setup_method(self):
|
||||
"""Set up temporary database for each test."""
|
||||
self.temp_db = tempfile.NamedTemporaryFile(delete=False, suffix='.db')
|
||||
self.temp_db.close()
|
||||
self.db_path = self.temp_db.name
|
||||
|
||||
def teardown_method(self):
|
||||
"""Clean up temporary database after each test."""
|
||||
if os.path.exists(self.db_path):
|
||||
os.unlink(self.db_path)
|
||||
|
||||
def test_invalid_yaml_frontmatter(self):
|
||||
"""Test handling of invalid YAML front matter."""
|
||||
invalid_content = '''---
|
||||
title: "Missing quotes
|
||||
author: Test User
|
||||
invalid: [unclosed list
|
||||
---
|
||||
# Content'''
|
||||
|
||||
parser = FrontMatterParser()
|
||||
# Should handle gracefully without crashing
|
||||
front_matter, content = parser.parse(invalid_content)
|
||||
|
||||
# Should return empty dict for invalid YAML and preserve content
|
||||
assert isinstance(front_matter, dict)
|
||||
assert content.startswith('# Content')
|
||||
|
||||
def test_database_path_with_invalid_directory(self):
|
||||
"""Test database creation with invalid directory path."""
|
||||
invalid_path = "/nonexistent/directory/test.db"
|
||||
|
||||
with pytest.raises(Exception): # Should raise appropriate exception
|
||||
db_manager = DatabaseManager(invalid_path)
|
||||
db_manager.initialize_database()
|
||||
Reference in New Issue
Block a user