mirror of
https://github.com/sissbruecker/linkding.git
synced 2025-08-10 12:17:56 +02:00
90 lines
3.0 KiB
Python
90 lines
3.0 KiB
Python
import os
|
|
import shutil
|
|
import tempfile
|
|
|
|
from django.conf import settings
|
|
from django.test import TestCase, override_settings
|
|
|
|
from bookmarks.services import bookmarks
|
|
from bookmarks.tests.helpers import BookmarkFactoryMixin
|
|
|
|
|
|
class BookmarkAssetsTestCase(TestCase, BookmarkFactoryMixin):
|
|
def setUp(self):
|
|
self.temp_dir = tempfile.mkdtemp()
|
|
self.override = override_settings(LD_ASSET_FOLDER=self.temp_dir)
|
|
self.override.enable()
|
|
|
|
def tearDown(self):
|
|
self.override.disable()
|
|
shutil.rmtree(self.temp_dir)
|
|
|
|
def setup_asset_file(self, filename):
|
|
filepath = os.path.join(settings.LD_ASSET_FOLDER, filename)
|
|
with open(filepath, "w") as f:
|
|
f.write("test")
|
|
|
|
def setup_asset_with_file(self, bookmark):
|
|
filename = f"temp_{bookmark.id}.html.gzip"
|
|
self.setup_asset_file(filename)
|
|
asset = self.setup_asset(bookmark=bookmark, file=filename)
|
|
return asset
|
|
|
|
def test_delete_bookmark_deletes_asset_file(self):
|
|
bookmark = self.setup_bookmark()
|
|
asset = self.setup_asset_with_file(bookmark)
|
|
self.assertTrue(
|
|
os.path.exists(os.path.join(settings.LD_ASSET_FOLDER, asset.file))
|
|
)
|
|
|
|
bookmark.delete()
|
|
self.assertFalse(
|
|
os.path.exists(os.path.join(settings.LD_ASSET_FOLDER, asset.file))
|
|
)
|
|
|
|
def test_bulk_delete_bookmarks_deletes_asset_files(self):
|
|
bookmark1 = self.setup_bookmark()
|
|
asset1 = self.setup_asset_with_file(bookmark1)
|
|
bookmark2 = self.setup_bookmark()
|
|
asset2 = self.setup_asset_with_file(bookmark2)
|
|
bookmark3 = self.setup_bookmark()
|
|
asset3 = self.setup_asset_with_file(bookmark3)
|
|
self.assertTrue(
|
|
os.path.exists(os.path.join(settings.LD_ASSET_FOLDER, asset1.file))
|
|
)
|
|
self.assertTrue(
|
|
os.path.exists(os.path.join(settings.LD_ASSET_FOLDER, asset2.file))
|
|
)
|
|
self.assertTrue(
|
|
os.path.exists(os.path.join(settings.LD_ASSET_FOLDER, asset3.file))
|
|
)
|
|
|
|
bookmarks.delete_bookmarks(
|
|
[bookmark1.id, bookmark2.id, bookmark3.id], self.get_or_create_test_user()
|
|
)
|
|
|
|
self.assertFalse(
|
|
os.path.exists(os.path.join(settings.LD_ASSET_FOLDER, asset1.file))
|
|
)
|
|
self.assertFalse(
|
|
os.path.exists(os.path.join(settings.LD_ASSET_FOLDER, asset2.file))
|
|
)
|
|
self.assertFalse(
|
|
os.path.exists(os.path.join(settings.LD_ASSET_FOLDER, asset3.file))
|
|
)
|
|
|
|
def test_save_updates_file_size(self):
|
|
# File does not exist initially
|
|
bookmark = self.setup_bookmark()
|
|
asset = self.setup_asset(bookmark=bookmark, file="temp.html.gz")
|
|
self.assertIsNone(asset.file_size)
|
|
|
|
# Add file, save again
|
|
self.setup_asset_file(asset.file)
|
|
asset.save()
|
|
self.assertEqual(asset.file_size, 4)
|
|
|
|
# Create asset with initial file
|
|
asset = self.setup_asset(bookmark=bookmark, file="temp.html.gz")
|
|
self.assertEqual(asset.file_size, 4)
|