def test_nested_env_context_manager(self):
"""
tests that we apply environment variables correctly
"""
self.host.connect()
value1 = random_string(8)
with env(self.host, 'TESTVAR', value1):
self.assertEqual(self.host.run('printenv TESTVAR').message.strip(), value1)
value2 = random_string(8)
with env(self.host, 'TESTVAR', value2):
self.assertEqual(self.host.run('printenv TESTVAR').message.strip(), value2)
self.assertEqual(self.host.run('printenv TESTVAR').message.strip(), value1)
def test_rm_rmdir_open_stat(self):
"""
can remove a directory
can't remove a directory that's not empty
"""
self.assertTrue(self.host.connect().status)
self.assertTrue(self.host.connected)
dirname = "%s%s" % (self.host.run_and_check('mktemp -p /tmp'), 'test_dir')
filepath = "%s/%s" % (dirname, 'file')
self.host.mkdir(dirname)
chars = 10000
data = random_string(chars)
fhandle = self.host.open(filepath, mode='rw')
fhandle.write(data)
fhandle.close()
fhandle = self.host.open(filepath, mode='r')
self.assertEquals(fhandle.read(), data)
stat = self.host.stat(filepath)
self.assertEqual(stat.get('size'), chars)
self.assertRaises(ConnectionError, self.host.rmdir, dirname)
self.host.rm(filepath)
self.host.rmdir(dirname)
def test_rm_rmdir_open_stat(self):
"""
can remove a directory
can't remove a directory that's not empty
"""
self.assertTrue(self.host.connect().status)
self.assertTrue(self.host.connected)
dirname = "%s%s" % (self.host.run_and_check('mktemp -p /tmp'),
'test_dir')
filepath = "%s/%s" % (dirname, 'file')
self.host.mkdir(dirname)
chars = 10000
data = random_string(chars)
fhandle = self.host.open(filepath, mode='rw')
fhandle.write(data)
fhandle.close()
fhandle = self.host.open(filepath, mode='r')
self.assertEquals(fhandle.read(), data)
stat = self.host.stat(filepath)
self.assertEqual(stat.get('size'), chars)
self.assertRaises(ConnectionError, self.host.rmdir, dirname)
self.host.rm(filepath)
self.host.rmdir(dirname)
def test_nested_env_context_manager(self):
"""
tests that we apply environment variables correctly
"""
self.host.connect()
value1 = random_string(8)
with env(self.host, 'TESTVAR', value1):
self.assertEqual(
self.host.run('printenv TESTVAR').message.strip(), value1)
value2 = random_string(8)
with env(self.host, 'TESTVAR', value2):
self.assertEqual(
self.host.run('printenv TESTVAR').message.strip(), value2)
self.assertEqual(
self.host.run('printenv TESTVAR').message.strip(), value1)
