1 files changed, 112 insertions, 0 deletions
diff --git a/morphlib/buildenvironment_tests.py b/morphlib/buildenvironment_tests.py
new file mode 100644
index 00000000..7ae7c2d5
--- /dev/null
+++ b/morphlib/buildenvironment_tests.py
@@ -0,0 +1,112 @@
+# Copyright (C) 2012-2013  Codethink Limited
+#
+# This program 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; version 2 of the License.
+#
+# This program 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 this program; if not, write to the Free Software Foundation, Inc.,
+# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
+
+
+import copy
+import unittest
+
+import morphlib
+from morphlib import buildenvironment
+
+
+class BuildEnvironmentTests(unittest.TestCase):
+
+    def setUp(self):
+        self.settings = {
+            'prefix': '/usr',
+            'no-ccache': True,
+            'no-distcc': True
+        }
+        self.fake_env = {
+            'PATH': '/fake_bin',
+        }
+
+    def new_build_env(self, settings=None, target=None, **kws):
+        settings = settings or self.settings
+        target = target or self.target
+        return buildenvironment.BuildEnvironment(settings, target, **kws)
+
+    def new_build_env(self, settings=None, arch='x86_64'):
+        settings = settings or self.settings
+        return buildenvironment.BuildEnvironment(settings, arch)
+
+    def test_copies_whitelist_vars(self):
+        env = self.fake_env
+        safe = {
+            'DISTCC_HOSTS': 'example.com:example.co.uk',
+            'LD_PRELOAD': '/buildenv/lib/libbuildenv.so',
+            'LD_LIBRARY_PATH': '/buildenv/lib:/buildenv/lib64',
+            'FAKEROOTKEY': 'b011de73',
+            'FAKED_MODE': 'non-fakeroot',
+            'FAKEROOT_FD_BASE': '-1',
+        }
+        env.update(safe)
+        old_osenv = buildenvironment.BuildEnvironment._osenv
+        buildenvironment.BuildEnvironment._osenv = env
+
+        buildenv = self.new_build_env()
+        self.assertEqual(sorted(safe.items()),
+                         sorted([(k, buildenv.env[k]) for k in safe.keys()]))
+
+        buildenvironment.BuildEnvironment._osenv = old_osenv
+
+    def test_user_spellings_equal(self):
+        buildenv = self.new_build_env()
+        self.assertTrue(buildenv.env['USER'] == buildenv.env['USERNAME'] ==
+                        buildenv.env['LOGNAME'])
+
+    def test_environment_overrides(self):
+        buildenv = self.new_build_env()
+        self.assertEqual(buildenv.env['TERM'], buildenv._override_term)
+        self.assertEqual(buildenv.env['SHELL'], buildenv._override_shell)
+        self.assertEqual(buildenv.env['USER'], buildenv._override_username)
+        self.assertEqual(buildenv.env['USERNAME'], buildenv._override_username)
+        self.assertEqual(buildenv.env['LOGNAME'], buildenv._override_username)
+        self.assertEqual(buildenv.env['LC_ALL'], buildenv._override_locale)
+        self.assertEqual(buildenv.env['HOME'], buildenv._override_home)
+
+    def test_arch_x86_64(self):
+        b = self.new_build_env(arch='x86_64')
+        self.assertEqual(b.env['MORPH_ARCH'], 'x86_64')
+        self.assertEqual(b.env['TARGET'], 'x86_64-baserock-linux-gnu')
+        self.assertEqual(b.env['TARGET_STAGE1'], 'x86_64-bootstrap-linux-gnu')
+
+    def test_arch_x86_32(self):
+        b = self.new_build_env(arch='x86_32')
+        self.assertEqual(b.env['MORPH_ARCH'], 'x86_32')
+        self.assertEqual(b.env['TARGET'], 'i686-baserock-linux-gnu')
+        self.assertEqual(b.env['TARGET_STAGE1'], 'i686-bootstrap-linux-gnu')
+
+    def test_arch_armv7l(self):
+        b = self.new_build_env(arch='armv7l')
+        self.assertEqual(b.env['MORPH_ARCH'], 'armv7l')
+        self.assertEqual(b.env['TARGET'], 'armv7l-baserock-linux-gnueabi')
+        self.assertEqual(b.env['TARGET_STAGE1'],
+                         'armv7l-bootstrap-linux-gnueabi')
+
+    def test_arch_armv7b(self):
+        b = self.new_build_env(arch='armv7b')
+        self.assertEqual(b.env['MORPH_ARCH'], 'armv7b')
+        self.assertEqual(b.env['TARGET'], 'armv7b-baserock-linux-gnueabi')
+        self.assertEqual(b.env['TARGET_STAGE1'],
+                         'armv7b-bootstrap-linux-gnueabi')
+
+    def test_ccache_vars_set(self):
+        new_settings = copy.copy(self.settings)
+        new_settings['no-ccache'] = False
+        new_settings['no-distcc'] = False
+        buildenv = self.new_build_env(settings=new_settings)
+        self.assertTrue(buildenv._ccache_path in buildenv.extra_path)
+        self.assertEqual(buildenv.env['CCACHE_PREFIX'], 'distcc')