Browse Source

Fix file path error in cmdctl test code.

git-svn-id: svn://bind10.isc.org/svn/bind10/branches/trac127@2302 e5f2f494-b856-4b98-b285-d166d9295462
Likun Zhang 15 years ago
parent
commit
a75fb3cc92
2 changed files with 11 additions and 4 deletions
  1. 1 0
      src/bin/cmdctl/tests/Makefile.am
  2. 10 4
      src/bin/cmdctl/tests/cmdctl_test.py

+ 1 - 0
src/bin/cmdctl/tests/Makefile.am

@@ -8,5 +8,6 @@ check-local:
 	for pytest in $(PYTESTS) ; do \
 	echo Running test: $$pytest ; \
 	env PYTHONPATH=$(abs_top_srcdir)/src/lib/python:$(abs_top_builddir)/src/lib/python:$(abs_top_builddir)/src/bin/cmdctl \
+	CMDCTL_SPEC_PATH=$(abs_top_srcdir)/src/bin/cmdctl \
 	$(PYCOVERAGE) $(abs_srcdir)/$$pytest ; \
 	done

+ 10 - 4
src/bin/cmdctl/tests/cmdctl_test.py

@@ -19,6 +19,11 @@ import socket
 import tempfile
 from cmdctl import *
 
+if 'CMDCTL_SPEC_PATH' in os.environ:
+    FILE_PATH = os.environ['CMDCTL_SPEC_PATH'] + os.sep
+else:
+    FILE_PATH = '..' + os.sep
+
 # Rewrite the class for unittest.
 class MySecureHTTPRequestHandler(SecureHTTPRequestHandler):
     def __init__(self):
@@ -279,7 +284,8 @@ class MyCommandControl(CommandControl):
         return {}
 
     def _setup_session(self):
-        module_spec = isc.config.module_spec_from_file("../cmdctl.spec.pre.in")
+        spec_file = FILE_PATH + 'cmdctl.spec'
+        module_spec = isc.config.module_spec_from_file(spec_file)
         config = isc.config.config_data.ConfigData(module_spec)
         self._cmdctl_config_data = config.get_full_config()
 
@@ -385,7 +391,7 @@ class TestSecureHTTPServer(unittest.TestCase):
         self.server._create_user_info('/local/not-exist')
         self.assertEqual(0, len(self.server._user_infos))
 
-        self.server._create_user_info('../cmdctl-accounts.csv')
+        self.server._create_user_info(FILE_PATH + 'cmdctl-accounts.csv')
         self.assertEqual(1, len(self.server._user_infos))
         self.assertTrue('root' in self.server._user_infos)
 
@@ -398,8 +404,8 @@ class TestSecureHTTPServer(unittest.TestCase):
                           '../cmdctl-certfile')
 
         self.server._wrap_socket_in_ssl_context(sock, 
-                          '../cmdctl-keyfile.pem',
-                          '../cmdctl-certfile.pem')
+                          FILE_PATH + 'cmdctl-keyfile.pem',
+                          FILE_PATH + 'cmdctl-certfile.pem')
 
 class TestFuncNotInClass(unittest.TestCase):
     def test_check_port(self):