kopia lustrzana https://github.com/OpenDroneMap/WebODM
83 wiersze
3.2 KiB
Python
83 wiersze
3.2 KiB
Python
|
import os
|
||
|
|
|
||
|
|
from django.contrib.auth.models import User, Group
|
||
|
|
from django.core.exceptions import ValidationError
|
||
|
|
from django.core.files import File
|
||
|
|
from django.test import Client
|
||
|
|
from shutil import rmtree
|
||
|
|
|
||
|
|
from app.models import Setting
|
||
|
|
from app.models import Theme
|
||
|
|
from .classes import BootTestCase
|
||
|
|
|
||
|
|
from app.contexts.settings import load as load_settings
|
||
|
|
from webodm import settings as webodm_settings
|
||
|
|
|
||
|
|
class TestSettings(BootTestCase):
|
||
|
|
|
||
|
|
def setUp(self):
|
||
|
|
pass
|
||
|
|
|
||
|
|
def tearDown(self):
|
||
|
|
pass
|
||
|
|
|
||
|
|
def test_settings(self):
|
||
|
|
# There should always be a Setting object
|
||
|
|
self.assertTrue(Setting.objects.count() == 1, "There's a settings object")
|
||
|
|
|
||
|
|
# There should be "default" Theme object
|
||
|
|
self.assertTrue(Theme.objects.filter(name="Default").count() == 1, "Default theme found")
|
||
|
|
|
||
|
|
# The default settings use the default theme
|
||
|
|
self.assertTrue(Setting.objects.first().theme.id == Theme.objects.get(name="Default").id, "Default theme is associated to settings")
|
||
|
|
|
||
|
|
# We can create a new theme object
|
||
|
|
second_theme = Theme.objects.create(name="Second")
|
||
|
|
|
||
|
|
# We cannot add another setting objects
|
||
|
|
with self.assertRaises(ValidationError):
|
||
|
|
Setting.objects.create(app_name="Test", theme=second_theme)
|
||
|
|
|
||
|
|
# We can retrieve the settings
|
||
|
|
settings = load_settings()['SETTINGS']
|
||
|
|
self.assertTrue(settings is not None, "Can retrieve settings")
|
||
|
|
|
||
|
|
# The default logo has been created in the proper destination
|
||
|
|
default_logo_path = os.path.join(webodm_settings.MEDIA_ROOT, settings.app_logo.name)
|
||
|
|
self.assertTrue(os.path.exists(default_logo_path), "Default logo exists in MEDIA_ROOT/settings")
|
||
|
|
|
||
|
|
# Access smaller logo (should generate a cached copy),
|
||
|
|
# and check that's been created
|
||
|
|
print("Access: " + settings.app_logo_favicon.url)
|
||
|
|
favicon_path = os.path.join(webodm_settings.MEDIA_ROOT, settings.app_logo_favicon.name)
|
||
|
|
self.assertTrue(os.path.exists(favicon_path), "Favicon logo exists")
|
||
|
|
|
||
|
|
# We can update the logo
|
||
|
|
logo = os.path.join('app', 'static', 'app', 'img', 'favicon.png')
|
||
|
|
settings.app_logo.save(os.path.basename(logo), File(open(logo, 'rb')))
|
||
|
|
settings.save()
|
||
|
|
|
||
|
|
# The main logo has been uploaded
|
||
|
|
self.assertTrue("favicon" in settings.app_logo.name, "Logo has been updated")
|
||
|
|
self.assertTrue(os.path.exists(os.path.join(webodm_settings.MEDIA_ROOT, settings.app_logo.name)),
|
||
|
|
"New logo exists in MEDIA_ROOT/settings")
|
||
|
|
|
||
|
|
# The old logo does not exist anymore
|
||
|
|
self.assertFalse(os.path.exists(default_logo_path),
|
||
|
|
"Old logo has been deleted")
|
||
|
|
|
||
|
|
# The old logo caches are gone also
|
||
|
|
self.assertFalse(os.path.exists(favicon_path), "Favicon logo has been removed")
|
||
|
|
|
||
|
|
# Resized images have not been created yet
|
||
|
|
self.assertFalse(os.path.exists(os.path.join(webodm_settings.MEDIA_ROOT, settings.app_logo_36.name)), "Resized logo does not exist")
|
||
|
|
|
||
|
|
# When we access its URL, it gets created (lazy)
|
||
|
|
print("Access: " + settings.app_logo_36.url)
|
||
|
|
|
||
|
|
self.assertTrue(os.path.exists(os.path.join(webodm_settings.MEDIA_ROOT, settings.app_logo_36.name)), "Resized logo does not exist")
|
||
|
|
|
||
|
|
c = Client()
|
||
|
|
|
||
|
|
|