root/cleverbox/trunk/cleverbox/tests/__init__.py

# This file is part of the "Cleverbox" program.
#
# Cleverbox is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# Cleverbox is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with Cleverbox.  If not, see <http://www.gnu.org/licenses/>.
#
# Copyright 2008, Tristan Rivoallan

import unittest
from cleverbox.tests import client_test, project_test, environment_test, upgrade_test

def validate_layout(layout):

    import grp, os, pwd, stat

    for (entry, spec) in layout.items():
        st = os.stat(entry)
        mode = st[stat.ST_MODE]

        # Validate file type
        func = getattr(stat, spec[0])
        assert func.__call__(mode), '"%s" type is correct (%s)' % (entry, spec[0])

        # Validate file mode
        assert oct(stat.S_IMODE(mode)) == spec[1], '"%s" mode is correct (expected %s, got %s)' % (entry, spec[1], oct(stat.S_IMODE(mode)))

        # Validate ownership
        uid = st[stat.ST_UID]
        gid = st[stat.ST_GID]
        expected_uid = pwd.getpwnam(spec[2])[2]
        expected_gid = grp.getgrnam(spec[3])[2]
        assert uid == expected_uid, '"%s" owner is correct (expected %s, got %s)' % (entry, spec[2], pwd.getpwuid(uid)[0])
        assert gid == expected_gid, '"%s" group is correct (expected %s, got %s)' % (entry, spec[3], grp.getgrgid(uid)[0])

def create_environment(version):

    import os, tempfile
    from cleverbox.model.environment import Environment

    test_root = os.path.join(tempfile.gettempdir(), 'cleverbox-tempenv')
    env_path = os.path.join(test_root, 'etc', 'cleverbox')
    os.makedirs(env_path)
    env = Environment(env_path)
    parameters = {'general' :
                     {'clients_root'     : os.path.join(test_root, 'var', 'cleverbox'),
                      'assets_dir'       : os.path.join('/home/trivoallan/workspace/cleverbox-trunk/assets'),
                      'apache_group'     : 'www-data',
                      'ssh_user'         : 'root',
                      'ssh_group'        : 'root',
                      'default_profile'  : 'default',
                      'authbackend_pass' : '',
                      'domain'           : 'test.domain.tld',
                      'apache_user'      : 'www-data',
                      'apache_group'     : 'www-data'},
                  'trac' :
                     {'lib_dir'          : '/usr/share/python-support/trac',
                      'assets_dir'       : '/usr/share/trac'}}
    env.create(version, parameters)

    return env


def suite():
    suite = unittest.TestSuite()
    suite.addTest(client_test.suite())
    suite.addTest(project_test.suite())
    suite.addTest(environment_test.suite())
    suite.addTest(upgrade_test.suite())

    return suite

if __name__ == '__main__':
    unittest.main(defaultTest='suite')