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_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)