Add sys.args tests

den-run-ai · vmuriart · commit 2d6f3714dc90 · 2017-02-08T10:24:31.000-07:00
diff --git a/src/embed_tests/pyimport.cs b/src/embed_tests/pyimport.cs
@@ -52,8 +52,10 @@ public void TearDown()
         [Test]
         public void TestDottedName()
         {
-            PyObject module = PythonEngine.ImportModule("PyImportTest.test.one");
-            Assert.IsNotNull(module, ">>>  import PyImportTest.test.one  # FAILED");
+            PyObject module1 = PythonEngine.ImportModule("PyImportTest.test.one");
+            Assert.IsNotNull(module1, ">>>  import PyImportTest.test.one  # FAILED");
+            PyObject module2 = PythonEngine.ImportModule("PyImportTest.sysargv");
+            Assert.IsNotNull(module2, ">>>  import PyImportTest.sysargv  # FAILED");
         }
     }
 }
diff --git a/src/tests/PyImportTest/sysargv.py b/src/tests/PyImportTest/sysargv.py
@@ -0,0 +1,5 @@
+# -*- coding: utf-8 -*-
+
+import sys
+# if argv is available, as expected, then no exception
+num_args = len(sys.argv)
diff --git a/src/tests/runtests.py b/src/tests/runtests.py
@@ -23,6 +23,9 @@
     clr.AddReference("System.Management")
 
 test_modules = (
+    # has to be first test before other module import clr
+	'test_sysargv',
+	
     # test_module passes on its own, but not here if
     # other test modules that import System.Windows.Forms
     # run first. They must not do module level import/AddReference()
diff --git a/src/tests/test_sysargv.py b/src/tests/test_sysargv.py
@@ -0,0 +1,17 @@
+# -*- coding: utf-8 -*-
+
+import unittest
+import sys
+
+class SysArgvTests(unittest.TestCase):
+    """Test sys.argv state."""
+
+    def test_sys_argv_state(self):
+        """Test sys.argv state doesn't change after clr import."""
+        argv = sys.argv
+        import clr
+        self.assertTrue(argv == sys.argv)
+
+
+def test_suite():
+    return unittest.makeSuite(SysArgvTests)
diff --git a/src/tests/tests.pyproj b/src/tests/tests.pyproj
@@ -29,12 +29,14 @@
   <ItemGroup>
     <Compile Include="leaktest.py" />
     <Compile Include="profile.py" />
+    <Compile Include="PyImportTest\sysargv.py" />
     <Compile Include="PyImportTest\test\one.py" />
     <Compile Include="PyImportTest\test\__init__.py" />
     <Compile Include="PyImportTest\__init__.py" />
     <Compile Include="runtests.py" />
     <Compile Include="stress.py" />
     <Compile Include="stresstest.py" />
+    <Compile Include="test_sysargv.py" />
     <Compile Include="test_array.py" />
     <Compile Include="test_class.py" />
     <Compile Include="test_compat.py" />

Original file line number	Diff line number	Diff line change
`@@ -52,8 +52,10 @@ public void TearDown()`
`52`	`52`	`[Test]`
`53`	`53`	`public void TestDottedName()`
`54`	`54`	`{`
`55`		`- PyObject module = PythonEngine.ImportModule("PyImportTest.test.one");`
`56`		`- Assert.IsNotNull(module, ">>> import PyImportTest.test.one # FAILED");`
	`55`	`+ PyObject module1 = PythonEngine.ImportModule("PyImportTest.test.one");`
	`56`	`+ Assert.IsNotNull(module1, ">>> import PyImportTest.test.one # FAILED");`
	`57`	`+ PyObject module2 = PythonEngine.ImportModule("PyImportTest.sysargv");`
	`58`	`+ Assert.IsNotNull(module2, ">>> import PyImportTest.sysargv # FAILED");`
`57`	`59`	`}`
`58`	`60`	`}`
`59`	`61`	`}`