diff options
Diffstat (limited to 'src/third_party/wiredtiger/test/suite/test_env01.py')
-rw-r--r-- | src/third_party/wiredtiger/test/suite/test_env01.py | 168 |
1 files changed, 168 insertions, 0 deletions
diff --git a/src/third_party/wiredtiger/test/suite/test_env01.py b/src/third_party/wiredtiger/test/suite/test_env01.py new file mode 100644 index 00000000000..491ef9e8eac --- /dev/null +++ b/src/third_party/wiredtiger/test/suite/test_env01.py @@ -0,0 +1,168 @@ +#!/usr/bin/env python +# +# Public Domain 2014-2016 MongoDB, Inc. +# Public Domain 2008-2014 WiredTiger, Inc. +# +# This is free and unencumbered software released into the public domain. +# +# Anyone is free to copy, modify, publish, use, compile, sell, or +# distribute this software, either in source code form or as a compiled +# binary, for any purpose, commercial or non-commercial, and by any +# means. +# +# In jurisdictions that recognize copyright laws, the author or authors +# of this software dedicate any and all copyright interest in the +# software to the public domain. We make this dedication for the benefit +# of the public at large and to the detriment of our heirs and +# successors. We intend this dedication to be an overt act of +# relinquishment in perpetuity of all present and future rights to this +# software under copyright law. +# +# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, +# EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF +# MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. +# IN NO EVENT SHALL THE AUTHORS BE LIABLE FOR ANY CLAIM, DAMAGES OR +# OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, +# ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR +# OTHER DEALINGS IN THE SOFTWARE. + +import os +import wiredtiger, wttest + +# test_priv01.py +# Test privileged operations. +# This is a variant of test_config02.py. This test should be run as both +# normal and privileged (e.g. root) user, and should pass in both cases. +class test_priv01(wttest.WiredTigerTestCase): + """ + This tests privileged operations. + The only part of WiredTiger that currently cares about privilege + levels is the 'use_environment_priv' configuration option for + wiredtiger_open, which by design, behaves differently depending on + whether the current process is running as a privileged user or + not. This test should be run as both normal and privileged + (e.g. root) user to fully test both cases. + """ + table_name1 = 'test_priv01' + nentries = 100 + + def setUp(self): + if os.name == 'nt': + self.skipTest('Unix specific test skipped on Windows') + super(test_priv01, self).setUp() + + # Each test needs to set up its connection in its own way, + # so override these methods to do nothing + def setUpConnectionOpen(self, dir): + return None + + def setUpSessionOpen(self, conn): + return None + + def populate_and_check(self): + """ + Create entries, and read back in a cursor: key=string, value=string + """ + create_args = 'key_format=S,value_format=S' + self.session.create("table:" + self.table_name1, create_args) + cursor = self.session.open_cursor('table:' + self.table_name1, None, None) + for i in range(0, self.nentries): + cursor[str(1000000 + i)] = 'value' + str(i) + i = 0 + cursor.reset() + for key, value in cursor: + self.assertEqual(key, str(1000000 + i)) + self.assertEqual(value, ('value' + str(i))) + i += 1 + self.assertEqual(i, self.nentries) + cursor.close() + + def checkfiles(self, dirname): + self.assertTrue(os.path.exists(dirname + os.sep + self.table_name1 + ".wt")) + + def checknofiles(self, dirname): + nfiles = len([nm for nm in os.listdir(dirname) if os.path.isfile(nm)]) + self.assertEqual(nfiles, 0) + + def common_test(self, homearg, homeenv, configextra): + """ + Call wiredtiger_open and run a simple test. + homearg is the first arg to wiredtiger_open, it may be null. + WIREDTIGER_HOME is set to homeenv, if it is not null. + configextra are any extra configuration strings needed on the open. + """ + configarg = 'create' + if configextra != None: + configarg += ',' + configextra + if homeenv == None: + os.unsetenv('WIREDTIGER_HOME') + else: + os.putenv('WIREDTIGER_HOME', homeenv) + try: + self.conn = self.wiredtiger_open(homearg, configarg) + self.session = self.conn.open_session(None) + self.populate_and_check() + finally: + os.unsetenv('WIREDTIGER_HOME') + + def test_home_and_env_conf_priv(self): + # If homedir is set, the environment is ignored + hdir = 'homedir' + edir = 'envdir' + os.mkdir(hdir) + os.mkdir(edir) + self.common_test(hdir, edir, 'use_environment_priv=true') + self.checkfiles(hdir) + self.checknofiles(edir) + + def test_home_and_missing_env_priv(self): + # If homedir is set, it is used no matter what + hdir = 'homedir' + os.mkdir(hdir) + self.common_test(hdir, None, 'use_environment_priv=true') + self.checkfiles(hdir) + + def test_env_conf_nopriv(self): + edir = 'envdir' + os.mkdir(edir) + if os.getuid() != os.geteuid(): + print 'Running ' + str(self) + ' as privileged user' + self.assertRaisesWithMessage(wiredtiger.WiredTigerError, + lambda: self.common_test(None, edir, None), + '/WIREDTIGER_HOME environment variable set but\ + process lacks privileges to use that environment variable/') + + def test_env_conf_priv(self): + edir = 'envdir' + os.mkdir(edir) + privarg = 'use_environment_priv=true' + if os.getuid() != os.geteuid(): + print 'Running ' + str(self) + ' as privileged user' + self.common_test(None, edir, privarg) + self.checkfiles(edir) + self.checknofiles(".") + + def test_env_conf_without_env_var_priv(self): + # no env var set, so should use current directory + self.common_test(None, None, 'use_environment_priv=true') + self.checkfiles(".") + + def test_env_conf(self): + # If no homedir is set, use environment + edir = 'envdir' + os.mkdir(edir) + self.common_test(None, edir, 'use_environment=true') + # this check fails because of test harness files: + #self.checknofiles('.') + self.checkfiles(edir) + + def test_env_conf_off(self): + # If no homedir is set, use environment + edir = 'envdir' + os.mkdir(edir) + self.common_test(None, edir, 'use_environment=false') + self.checknofiles(edir) + self.checkfiles('.') + +if __name__ == '__main__': + wttest.run() |