FlightDeck/utils/test.py

import os

from test_utils import TestCase as _TestCase
from django.conf import settings


class TestCase(_TestCase):
    """
    Base class for tests depending on existance of lib/jetpack-sdk
    """
    def createCore(self):
        " discover the newest dir and link to it "
        # find the newest SDK
        sdks = os.listdir(os.path.join(settings.FRAMEWORK_PATH, 'lib'))
        self.sdk_filename = None
        sdk_time = -1
        for sdk in sdks:
            if sdk != '__init__.py':
                sdk_inf = os.stat(os.path.join(settings.FRAMEWORK_PATH, 'lib',
                    sdk))
                if sdk_time < 0 or sdk_time > sdk_inf.st_ctime:
                    sdk_time = sdk_inf.st_ctime
                    self.sdk_filename = sdk
        self.sdk_path = os.path.join(settings.FRAMEWORK_PATH,
                'lib/jetpack-sdk')
        sdk_orig = os.path.join(settings.FRAMEWORK_PATH, 'lib',
                self.sdk_filename)
        self.core_link_created = False
        if not os.path.exists(self.sdk_path):
            os.symlink(sdk_orig, self.sdk_path)
            self.core_link_created = True

    def deleteCore(self):
        " remove symlink "
        if not hasattr(self, 'core_link_created'):
            return
        if self.core_link_created:
            os.remove(self.sdk_path)