import unittest
import unit

class TestUnitPythonEnvironment(unit.TestUnitApplicationPython):

    def setUpClass():
        unit.TestUnit().check_modules('python')

    def test_python_environment_name_null(self):
        self.load('environment')

        self.assertIn('error', self.conf({
            "va\0r": "val1"
        }, 'applications/environment/environment'), 'name null')

    def test_python_environment_name_equals(self):
        self.load('environment')

        self.assertIn('error', self.conf({
            "var=": "val1"
        }, 'applications/environment/environment'), 'name equals')

    def test_python_environment_value_null(self):
        self.load('environment')

        self.assertIn('error', self.conf({
            "var": "\0val"
        }, 'applications/environment/environment'), 'value null')

    def test_python_environment_update(self):
        self.load('environment')

        self.conf({
            "var": "val1"
        }, 'applications/environment/environment')

        self.assertEqual(self.get(headers={
            'Host': 'localhost',
            'X-Variables': 'var',
            'Connection': 'close'
        })['body'], 'val1,', 'set')

        self.conf({
            "var": "val2"
        }, 'applications/environment/environment')

        self.assertEqual(self.get(headers={
            'Host': 'localhost',
            'X-Variables': 'var',
            'Connection': 'close'
        })['body'], 'val2,', 'update')

    def test_python_environment_replace(self):
        self.load('environment')

        self.conf({
            "var1": "val1"
        }, 'applications/environment/environment')

        self.assertEqual(self.get(headers={
            'Host': 'localhost',
            'X-Variables': 'var1',
            'Connection': 'close'
        })['body'], 'val1,', 'set')

        self.conf({
            "var2": "val2"
        }, 'applications/environment/environment')

        self.assertEqual(self.get(headers={
            'Host': 'localhost',
            'X-Variables': 'var1,var2',
            'Connection': 'close'
        })['body'], 'val2,', 'replace')

    def test_python_environment_clear(self):
        self.load('environment')

        self.conf({
            "var1": "val1",
            "var2": "val2"
        }, 'applications/environment/environment')

        self.assertEqual(self.get(headers={
            'Host': 'localhost',
            'X-Variables': 'var1,var2',
            'Connection': 'close'
        })['body'], 'val1,val2,', 'set')

        self.conf({}, 'applications/environment/environment')

        self.assertEqual(self.get(headers={
            'Host': 'localhost',
            'X-Variables': 'var1,var2',
            'Connection': 'close'
        })['body'], '', 'clear')

    def test_python_environment_replace_default(self):
        self.load('environment')

        pwd_default = self.get(headers={
            'Host': 'localhost',
            'X-Variables': 'PWD',
            'Connection': 'close'
        })['body']

        self.assertGreater(len(pwd_default), 1, 'get default')

        self.conf({
            "PWD": "new/pwd"
        }, 'applications/environment/environment')

        self.assertEqual(self.get(headers={
            'Host': 'localhost',
            'X-Variables': 'PWD',
            'Connection': 'close'
        })['body'], 'new/pwd,', 'replace default')

        self.conf({}, 'applications/environment/environment')

        self.assertEqual(self.get(headers={
            'Host': 'localhost',
            'X-Variables': 'PWD',
            'Connection': 'close'
        })['body'], pwd_default, 'restore default')

if __name__ == '__main__':
    TestUnitPythonEnvironment.main()