示例#1
0
 def test_run_command_stdin(self):
     connector = BaseCommandDBConnector()
     stdin = SpooledTemporaryFile()
     stdin.write(b'foo')
     stdin.seek(0)
     # Run
     stdout, stderr = connector.run_command('cat', stdin=stdin)
     self.assertEqual(stdout.read(), b'foo')
     self.assertFalse(stderr.read())
 def test_run_command_with_env(self):
     connector = BaseCommandDBConnector()
     # Empty env
     stdout, stderr = connector.run_command('env')
     self.assertTrue(stdout.read())
     # env from self.env
     connector.env = {'foo': 'bar'}
     stdout, stderr = connector.run_command('env')
     self.assertIn(b'foo=bar\n', stdout.read())
     # method overide gloabal env
     stdout, stderr = connector.run_command('env', env={'foo': 'ham'})
     self.assertIn(b'foo=ham\n', stdout.read())
     # get a var from parent env
     os.environ['bar'] = 'foo'
     stdout, stderr = connector.run_command('env')
     self.assertIn(b'bar=foo\n', stdout.read())
     # Conf overides parendt env
     connector.env = {'bar': 'bar'}
     stdout, stderr = connector.run_command('env')
     self.assertIn(b'bar=bar\n', stdout.read())
     # method overides all
     stdout, stderr = connector.run_command('env', env={'bar': 'ham'})
     self.assertIn(b'bar=ham\n', stdout.read())
示例#3
0
 def test_run_command_with_env(self):
     connector = BaseCommandDBConnector()
     # Empty env
     stdout, stderr = connector.run_command('env')
     self.assertFalse(stdout.read())
     # env from self.env
     connector.env = {'foo': 'bar'}
     stdout, stderr = connector.run_command('env')
     self.assertEqual(stdout.read(), b'foo=bar\n')
     # method overide gloabal env
     stdout, stderr = connector.run_command('env', env={'foo': 'ham'})
     self.assertEqual(stdout.read(), b'foo=ham\n')
 def test_run_command_with_parent_env(self):
     connector = BaseCommandDBConnector(use_parent_env=False)
     # Empty env
     stdout, stderr = connector.run_command('env')
     self.assertFalse(stdout.read())
     # env from self.env
     connector.env = {'foo': 'bar'}
     stdout, stderr = connector.run_command('env')
     self.assertEqual(stdout.read(), b'foo=bar\n')
     # method overide gloabal env
     stdout, stderr = connector.run_command('env', env={'foo': 'ham'})
     self.assertEqual(stdout.read(), b'foo=ham\n')
     # no var from parent env
     os.environ['bar'] = 'foo'
     stdout, stderr = connector.run_command('env')
     self.assertNotIn(b'bar=foo\n', stdout.read())
 def test_run_command_with_env(self):
     connector = BaseCommandDBConnector()
     # Empty env
     stdout, stderr = connector.run_command('env')
     self.assertTrue(stdout.read())
     # env from self.env
     connector.env = {'foo': 'bar'}
     stdout, stderr = connector.run_command('env')
     self.assertIn(b'foo=bar\n', stdout.read())
     # method overide gloabal env
     stdout, stderr = connector.run_command('env', env={'foo': 'ham'})
     self.assertIn(b'foo=ham\n', stdout.read())
     # get a var from parent env
     os.environ['bar'] = 'foo'
     stdout, stderr = connector.run_command('env')
     self.assertIn(b'bar=foo\n', stdout.read())
     # Conf overides parendt env
     connector.env = {'bar': 'bar'}
     stdout, stderr = connector.run_command('env')
     self.assertIn(b'bar=bar\n', stdout.read())
     # method overides all
     stdout, stderr = connector.run_command('env', env={'bar': 'ham'})
     self.assertIn(b'bar=ham\n', stdout.read())
 def test_run_command_with_parent_env(self):
     connector = BaseCommandDBConnector(use_parent_env=False)
     # Empty env
     stdout, stderr = connector.run_command('env')
     self.assertFalse(stdout.read())
     # env from self.env
     connector.env = {'foo': 'bar'}
     stdout, stderr = connector.run_command('env')
     self.assertEqual(stdout.read(), b'foo=bar\n')
     # method overide gloabal env
     stdout, stderr = connector.run_command('env', env={'foo': 'ham'})
     self.assertEqual(stdout.read(), b'foo=ham\n')
     # no var from parent env
     os.environ['bar'] = 'foo'
     stdout, stderr = connector.run_command('env')
     self.assertNotIn(b'bar=foo\n', stdout.read())
示例#7
0
 def test_run_command_error(self):
     connector = BaseCommandDBConnector()
     with self.assertRaises(exceptions.CommandConnectorError):
         connector.run_command('echa 123')
示例#8
0
 def test_run_command(self):
     connector = BaseCommandDBConnector()
     stdout, stderr = connector.run_command('echo 123')
     self.assertEqual(stdout.read(), b'123\n')
     self.assertEqual(stderr.read(), b'')