OpenDroneMap-WebODM/app/tests/classes.py

import os

from django import db
from django.contrib.auth.models import User
from django.test import TestCase
from django.test import TransactionTestCase
from shutil import rmtree

from app.boot import boot
from app.models import Project
from webodm import settings


def setupUsers():
    User.objects.create_superuser(username='testsuperuser',
                                  email='superuser@test.com',
                                  password='test1234')
    User.objects.create_user(username='testuser',
                             email='user@test.com',
                             password='test1234')
    User.objects.create_user(username='testuser2',
                             email='user2@test.com',
                             password='test1234')


def setupProjects():
    Project.objects.create(
        owner=User.objects.get(username="testsuperuser"),
        name="Super User Test Project",
        description="This is a test project"
    )
    Project.objects.create(
        owner=User.objects.get(username="testuser"),
        name="User Test Project",
        description="This is a test project"
    )
    Project.objects.create(
        owner=User.objects.get(username="testuser2"),
        name="User 2 Test Project",
        description="This is a test project"
    )


def cleanup():
    if settings.TESTING and \
            os.path.exists(settings.MEDIA_ROOT) and \
                    "_test" in settings.MEDIA_ROOT[-6:]:
        rmtree(settings.MEDIA_ROOT)
        print("Cleaned " + settings.MEDIA_ROOT)


class BootTestCase(TestCase):
    '''
    This class provides optional default mock data as well as 
    proper boot initialization code. All tests for the app
    module should derive from this class instead of TestCase.

    We don't use fixtures because we have signal initialization login
    for some models, which doesn't play well with them.
    '''
    @classmethod
    def setUpTestData(cls):
        super(BootTestCase, cls).setUpTestData()
        cleanup()
        boot()
        setupUsers()
        setupProjects()

    @classmethod
    def tearDownClass(cls):
        super(BootTestCase, cls).tearDownClass()


class BootTransactionTestCase(TransactionTestCase):
    '''
    Same as above, but inherits from TransactionTestCase
    '''
    def setUp(self):
        super().setUp()
        cleanup()
        boot()
        setupUsers()
        setupProjects()

    @classmethod
    def tearDownClass(cls):
        super(BootTransactionTestCase, cls).tearDownClass()