native_client_sdk/src/tools/tests/getos_test.py

   1 #!/usr/bin/env python
   2 # Copyright (c) 2012 The Chromium Authors. All rights reserved.
   3 # Use of this source code is governed by a BSD-style license that can be
   4 # found in the LICENSE file.
   5
   6 import os
   7 import shutil
   8 import sys
   9 import tempfile
  10 import unittest
  11
  12 SCRIPT_DIR = os.path.dirname(os.path.abspath(__file__))
  13 TOOLS_DIR = os.path.dirname(SCRIPT_DIR)
  14 CHROME_SRC = os.path.dirname(os.path.dirname(os.path.dirname(TOOLS_DIR)))
  15 MOCK_DIR = os.path.join(CHROME_SRC, "third_party", "pymock")
  16
  17 # For the mock library
  18 sys.path.append(MOCK_DIR)
  19 import mock
  20
  21 # For getos, the module under test
  22 sys.path.append(TOOLS_DIR)
  23 import getos
  24 import oshelpers
  25
  26
  27 class TestCaseExtended(unittest.TestCase):
  28   """Monkey patch some 2.7-only TestCase features."""
  29   # TODO(sbc): remove this once we switch to python2.7 everywhere
  30   def assertIn(self, expr1, expr2, msg=None):
  31     if hasattr(super(TestCaseExtended, self), 'assertIn'):
  32       return super(TestCaseExtended, self).assertIn(expr1, expr2, msg)
  33     if expr1 not in expr2:
  34       self.fail(msg or '%r not in %r' % (expr1, expr2))
  35
  36
  37 class TestGetos(TestCaseExtended):
  38   def setUp(self):
  39     self.patch1 = mock.patch.dict('os.environ',
  40                                   {'NACL_SDK_ROOT': os.path.dirname(TOOLS_DIR)})
  41     self.patch1.start()
  42     self.patch2 = mock.patch.object(oshelpers, 'FindExeInPath',
  43                                     return_value='/bin/ls')
  44     self.patch2.start()
  45
  46   def tearDown(self):
  47     self.patch1.stop()
  48     self.patch2.stop()
  49
  50   def testGetSDKPath(self):
  51     """honors environment variable."""
  52     with mock.patch.dict('os.environ', {'NACL_SDK_ROOT': 'dummy'}):
  53       self.assertEqual(getos.GetSDKPath(), 'dummy')
  54
  55   def testGetSDKPathDefault(self):
  56     """defaults to relative path."""
  57     del os.environ['NACL_SDK_ROOT']
  58     self.assertEqual(getos.GetSDKPath(), os.path.dirname(TOOLS_DIR))
  59
  60   def testGetPlatform(self):
  61     """returns a valid platform."""
  62     platform = getos.GetPlatform()
  63     self.assertIn(platform, ('mac', 'linux', 'win'))
  64
  65   def testGetSystemArch(self):
  66     """returns a valid architecture."""
  67     arch = getos.GetSystemArch(getos.GetPlatform())
  68     self.assertIn(arch, ('x86_64', 'x86_32', 'arm'))
  69
  70   def testGetChromePathEnv(self):
  71     """honors CHROME_PATH environment."""
  72     with mock.patch.dict('os.environ', {'CHROME_PATH': '/dummy/file'}):
  73       expect = "Invalid CHROME_PATH.*/dummy/file"
  74       platform = getos.GetPlatform()
  75       if hasattr(self, 'assertRaisesRegexp'):
  76         with self.assertRaisesRegexp(getos.Error, expect):
  77           getos.GetChromePath(platform)
  78       else:
  79         # TODO(sbc): remove this path once we switch to python2.7 everywhere
  80         self.assertRaises(getos.Error, getos.GetChromePath, platform)
  81
  82   def testGetChromePathCheckExists(self):
  83     """checks that existence of explicitly CHROME_PATH is checked."""
  84     mock_location = '/bin/ls'
  85     platform = getos.GetPlatform()
  86     if platform == 'win':
  87       mock_location = 'c:\\nowhere'
  88     with mock.patch.dict('os.environ', {'CHROME_PATH': mock_location}):
  89       with mock.patch('os.path.exists') as mock_exists:
  90         chrome = getos.GetChromePath(platform)
  91         self.assertEqual(chrome, mock_location)
  92         mock_exists.assert_called_with(chrome)
  93
  94   def testGetNaClArch(self):
  95     """returns a valid architecture."""
  96     platform = getos.GetPlatform()
  97     # Since the unix implementation of GetNaClArch will run objdump on the
  98     # chrome binary, and we want to be able to run this test without chrome
  99     # installed we mock the GetChromePath call to return a known system binary,
 100     # which objdump will work with.
 101     with mock.patch('getos.GetChromePath') as mock_chrome_path:
 102       mock_chrome_path.return_value = '/bin/ls'
 103       arch = getos.GetNaClArch(platform)
 104       self.assertIn(arch, ('x86_64', 'x86_32', 'arm'))
 105
 106
 107 class TestGetosWithTempdir(TestCaseExtended):
 108   def setUp(self):
 109     self.tempdir = tempfile.mkdtemp("_sdktest")
 110     self.patch = mock.patch.dict('os.environ',
 111                                   {'NACL_SDK_ROOT': self.tempdir})
 112     self.patch.start()
 113
 114   def tearDown(self):
 115     shutil.rmtree(self.tempdir)
 116     self.patch.stop()
 117
 118   def testGetSDKVersion(self):
 119     """correctly parses README to find SDK version."""
 120     expected_version = (16, 196)
 121     with open(os.path.join(self.tempdir, 'README'), 'w') as out:
 122       out.write('Version: %s\n' % expected_version[0])
 123       out.write('Chrome Revision: %s\n' % expected_version[1])
 124
 125     version = getos.GetSDKVersion()
 126     self.assertEqual(version, expected_version)
 127
 128
 129 if __name__ == '__main__':
 130   unittest.main()