root/cleverbox/trunk/cleverbox/tests/environment_test.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 grp, logging, os, pwd, shutil, stat, tempfile, unittest
from cleverbox.model.environment import Environment
import cleverbox.tests

# Don't bother with output from scripts
logging.basicConfig(stream=open('/dev/null', 'w+'))

class EnvironmentTestCase(unittest.TestCase):
    
    # Fixtures
    def setUp(self):
        self.env = cleverbox.tests.create_environment('0.5test')

    def tearDown(self):
        pass
        shutil.rmtree(self.env.path)
        shutil.rmtree(self.env.config.get('general', 'clients_root'))
    
    def test_create(self):
        """Environment.create() generates a correct directory layout."""

        expected = {self.env.get_path('VERSION')                               : ('S_ISREG', '0640', 'root', 'www-data'), 
                    self.env.get_path('cleverbox.ini')                         : ('S_ISREG', '0640', 'root', 'www-data'), 
                    self.env.get_path('clients-available')                     : ('S_ISDIR', '0750', 'root', 'www-data'), 
                    self.env.get_path('clients-enabled')                       : ('S_ISDIR', '0750', 'root', 'www-data'), 
                    self.env.get_path('projects-available')                    : ('S_ISDIR', '0750', 'root', 'www-data'), 
                    self.env.get_path('projects-enabled')                      : ('S_ISDIR', '0750', 'root', 'www-data'), 
                    self.env.get_path('profiles')                              : ('S_ISDIR', '0750', 'root', 'www-data'), 
                    self.env.get_path('profiles/default')                      : ('S_ISDIR', '0750', 'root', 'www-data'), 
                    self.env.get_path('profiles/default/trac-defaults.ini')    : ('S_ISREG', '0640', 'root', 'www-data'), 
                    self.env.get_path('profiles/default/project.apache.conf')  : ('S_ISREG', '0640', 'root', 'www-data'), 
                    self.env.get_path('profiles/default/permissions.ini')      : ('S_ISREG', '0640', 'root', 'www-data'),
                    self.env.config.get('general', 'clients_root')             : ('S_ISDIR', '0750', 'root', 'www-data')}

        cleverbox.tests.validate_layout(expected)            

    def test_create_creates_enough_elements(self):
        """Environment.create() creates the right number of elements on filesystem."""
        assert len(os.listdir(self.env.path)) == 7
        assert len(os.listdir(os.path.join(self.env.path, 'profiles'))) == 1
        assert len(os.listdir(os.path.join(self.env.path, 'profiles', 'default'))) == 3

    def test_needs_upgrade_triggers_when_needs(self):
        """Environment.needs_upgrade() returns true if environment needs an upgrade."""
        assert self.env.needs_upgrade('0.6')

    def test_needs_upgrade_does_not_trigger_when_unneeded(self):
        """Environment.needs_upgrade() returns false if environment does not need an upgrade."""
        assert self.env.needs_upgrade('0.4') == False

    def test_get_version(self):
        """Environment.get_version() returns the right version number."""
        assert self.env.get_version() == '0.5test'

    def test_get_path(self):
        """Environment.get_path() returns the right path."""
        assert self.env.get_path('TEST') == os.path.join(self.env.path, 'TEST')

            
def suite():
    return unittest.makeSuite(EnvironmentTestCase, 'test')

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