root/galaxy-central/eggs/GeneTrack-2.0.0_beta_1_dev_48da9e998f0caf01c5be731e926f4b0481f658f0-py2.6.egg/tests/authorize_tests.py

import testlib
import os, unittest, random
from django.test import utils
from django.db import connection
from django.conf import settings
from genetrack import conf, util, logger
from genetrack.server.scripts import initializer
from genetrack.server.web import html
from django.contrib.auth.models import User
from django.core.files import File

class AuthorizeTest( unittest.TestCase ):
    """
    Tests the models
    """
    def setUp(self):
        "Setting up the tests database for data insert"
        self.old_name = settings.DATABASE_NAME
        utils.setup_test_environment()
        connection.creation.create_test_db(verbosity=0, autoclobber=True)
        options = util.Params(test_mode=True, delete_everything=False, flush=False, verbosity=0)
        fname = conf.testdata('test-users.csv')
        initializer.load_users(fname, options)
        

    def tearDown(self):
        "Tearing down the database after test"
        connection.creation.destroy_test_db(self.old_name, 0)
        utils.teardown_test_environment()
    
    def test_data_creation(self):
        """
        Create datasets
        """
        
        # it seems that importing it earlier messes up the test database setup
        from genetrack.server.web import authorize

        john = User.objects.get(username='johndoe')
        project = authorize.create_project(user=john, name="Test project")
        stream = File( open(conf.testdata('test-users.csv')) )
        data = authorize.create_data(user=john, pid=project.id, stream=stream, name="Test data")

        # project counts update
        project = authorize.get_project(user=john, pid=project.id)
        self.assertEqual(project.data_count, 1)

        # testing data deletion
        authorize.delete_data(user=john, pid=project.id, dids=[data.id])
        project = authorize.get_project(user=john, pid=project.id)
        self.assertEqual(project.data_count, 0)

    def test_two(self):
        pass

def get_suite():
    "Returns the testsuite"
    tests  = [ 
        AuthorizeTest,
    ]

    return testlib.make_suite( tests )

if __name__ == '__main__':
    suite = get_suite()
    logger.disable('DEBUG')
    unittest.TextTestRunner(verbosity=2).run( suite )