[NaCl SDK] Convert python scripts from optparse to argparse.

native_client_sdk/src/tools/tests/getos_test.py

 #!/usr/bin/env python
 # Copyright (c) 2012 The Chromium Authors. All rights reserved.
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 
 import os
 import shutil
 import sys
 import tempfile
 import unittest
 
 SCRIPT_DIR = os.path.dirname(os.path.abspath(__file__))
 TOOLS_DIR = os.path.dirname(SCRIPT_DIR)
 CHROME_SRC = os.path.dirname(os.path.dirname(os.path.dirname(TOOLS_DIR)))
-MOCK_DIR = os.path.join(CHROME_SRC, "third_party", "pymock")
+MOCK_DIR = os.path.join(CHROME_SRC, 'third_party', 'pymock')
 
 # For the mock library
 sys.path.append(MOCK_DIR)
 import mock
 
 # For getos, the module under test
 sys.path.append(TOOLS_DIR)
 import getos
 import oshelpers
 
@@ skipping 70 matching lines @@
     platform = getos.GetPlatform()
     # Since the unix implementation of GetNaClArch will run objdump on the
     # chrome binary, and we want to be able to run this test without chrome
     # installed we mock the GetChromePath call to return a known system binary,
     # which objdump will work with.
     with mock.patch('getos.GetChromePath') as mock_chrome_path:
       mock_chrome_path.return_value = '/bin/ls'
       arch = getos.GetNaClArch(platform)
       self.assertIn(arch, ('x86_64', 'x86_32', 'arm'))
 
+  def testMainInvalidArgs(self):
+    with self.assertRaises(SystemExit):
+      with mock.patch('sys.stderr'):
+        getos.main('--foo')
+
+  @mock.patch('sys.stdout', mock.Mock())
+  @mock.patch('getos.GetPlatform')
+  def testMainNoArgs(self, mock_get_platform):
+    mock_get_platform.return_value = 'platform'
+    getos.main([])
+
+  @mock.patch('sys.stdout', mock.Mock())
+  @mock.patch('getos.GetSystemArch')
+  def testMainArgsParsing(self, mock_system_arch):
+    mock_system_arch.return_value = 'dummy'
+    getos.main(['--arch'])
+    mock_system_arch.assert_called()
+
 
 class TestGetosWithTempdir(TestCaseExtended):
   def setUp(self):
     self.tempdir = tempfile.mkdtemp("_sdktest")
     self.patch = mock.patch.dict('os.environ',
                                   {'NACL_SDK_ROOT': self.tempdir})
     self.patch.start()
 
   def tearDown(self):
     shutil.rmtree(self.tempdir)
@@ skipping 42 matching lines @@
     required_version = (17, 50)
     self.assertRaisesRegexp(
         getos.Error,
         r'SDK version too old \(current: 16.100, required: 17.50\)',
         getos.CheckVersion,
         required_version)
 
 
 if __name__ == '__main__':
   unittest.main()