Initial upload: snippet-manager with CI/CD workflow
This commit is contained in:
85
tests/test_export.py
Normal file
85
tests/test_export.py
Normal file
@@ -0,0 +1,85 @@
|
||||
"""Tests for import/export functionality."""
|
||||
|
||||
import json
|
||||
import os
|
||||
import tempfile
|
||||
|
||||
import pytest
|
||||
|
||||
from snip.db.database import Database
|
||||
from snip.export.handlers import export_snippets, import_snippets
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def db():
|
||||
with tempfile.NamedTemporaryFile(suffix=".db", delete=False) as f:
|
||||
db_path = f.name
|
||||
database = Database(db_path)
|
||||
database.init_db()
|
||||
yield database
|
||||
os.unlink(db_path)
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def export_file():
|
||||
with tempfile.NamedTemporaryFile(suffix=".json", delete=False) as f:
|
||||
file_path = f.name
|
||||
yield file_path
|
||||
if os.path.exists(file_path):
|
||||
os.unlink(file_path)
|
||||
|
||||
|
||||
def test_export_all(db, export_file):
|
||||
"""Test exporting all snippets."""
|
||||
db.add_snippet(title="Test 1", code="code1", tags=["test"])
|
||||
db.add_snippet(title="Test 2", code="code2", tags=["test"])
|
||||
|
||||
snippets = db.export_all()
|
||||
export_snippets(snippets, export_file)
|
||||
|
||||
with open(export_file, "r") as f:
|
||||
data = json.load(f)
|
||||
|
||||
assert data["version"] == "1.0"
|
||||
assert "exported_at" in data
|
||||
assert len(data["snippets"]) == 2
|
||||
|
||||
|
||||
def test_import_skip_strategy(db, export_file):
|
||||
"""Test import with skip strategy."""
|
||||
db.add_snippet(title="Existing", code="existing_code")
|
||||
|
||||
snippets = [{"title": "Existing", "code": "new_code"}, {"title": "New", "code": "new_code"}]
|
||||
export_snippets(snippets, export_file)
|
||||
|
||||
imported, skipped = import_snippets(db, export_file, strategy="skip")
|
||||
assert imported == 1
|
||||
assert skipped == 1
|
||||
|
||||
|
||||
def test_import_replace_strategy(db, export_file):
|
||||
"""Test import with replace strategy."""
|
||||
snippet_id = db.add_snippet(title="Existing", code="old_code")
|
||||
|
||||
snippets = [{"title": "Existing", "code": "new_code"}]
|
||||
export_snippets(snippets, export_file)
|
||||
|
||||
imported, skipped = import_snippets(db, export_file, strategy="replace")
|
||||
assert imported == 1
|
||||
|
||||
updated = db.get_snippet(snippet_id)
|
||||
assert updated["code"] == "new_code"
|
||||
|
||||
|
||||
def test_import_duplicate_strategy(db, export_file):
|
||||
"""Test import with duplicate strategy."""
|
||||
db.add_snippet(title="Existing", code="existing")
|
||||
|
||||
snippets = [{"title": "Existing", "code": "existing"}]
|
||||
export_snippets(snippets, export_file)
|
||||
|
||||
imported, skipped = import_snippets(db, export_file, strategy="duplicate")
|
||||
assert imported == 1
|
||||
|
||||
all_snippets = db.list_snippets(limit=100)
|
||||
assert len(all_snippets) == 2
|
||||
Reference in New Issue
Block a user