86 lines
2.4 KiB
Python
86 lines
2.4 KiB
Python
"""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
|