FlightDeck/utils/test.py

import os

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

def get_latest_sdk_dir():
    sdks = os.listdir(os.path.join(settings.FRAMEWORK_PATH, 'lib'))
    found = 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
                found = sdk
    return found

class TestCase(_TestCase):
    """
    Base class for tests depending on existance of lib/jetpack-sdk
    """
    def createCore(self, core_dir='jetpack-sdk'):
        " discover the newest dir and link to it "
        # find the newest SDK
        self.sdk_filename = get_latest_sdk_dir()
        self.sdk_path = os.path.join(settings.FRAMEWORK_PATH,
                'lib/', core_dir)
        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)