def test_put_file(self): with tempfile.NamedTemporaryFile( ) as tmp1, tempfile.NamedTemporaryFile() as tmp2: tmp1.write('foo') tmp1.flush() SyncRemoteProcess.put_file('localhost', tmp1.name, tmp2.name) self.assertEqual('foo', tmp2.read())
def test_put_file(self): with tempfile.NamedTemporaryFile() as tmp1, tempfile.NamedTemporaryFile() as tmp2: tmp1.write('foo') tmp1.flush() SyncRemoteProcess.put_file('localhost', tmp1.name, tmp2.name) self.assertEqual('foo', tmp2.read())
def test_get_file(self): with tempfile.NamedTemporaryFile() as tmp: SyncRemoteProcess.get_file('localhost', '/etc/hosts', tmp.name) with open('/etc/hosts', 'r') as expected_file: self.assertEqual(expected_file.read(), tmp.read())
def test_run(self): result = SyncRemoteProcess.run('localhost', ['/bin/echo', '-n', 'foo']) self.assertEqual('foo', result)
def test_run_timeout(self): result = SyncRemoteProcess.run('localhost', ['/bin/echo', '-n', 'baz'], {}, 5) self.assertEquals('baz', result)
def test_run_envvar(self): result = SyncRemoteProcess.run('localhost', ['/bin/echo', '-n', '${PARAM}'], {'PARAM': 'bar'}) self.assertEquals('bar', result)
def test_get_file(self): with tempfile.NamedTemporaryFile() as tmp: SyncRemoteProcess.get_file('localhost', '/etc/hosts', tmp.name) with open('/etc/hosts', 'rb') as expected_file: self.assertEqual(expected_file.read(), tmp.read())
def test_run_timeout(self): result = SyncRemoteProcess.run('localhost', ['/bin/echo', '-n', 'baz'], {}, 5) self.assertEquals(b'baz', result)
def test_run_envvar(self): result = SyncRemoteProcess.run('localhost', ['/bin/echo', '-n', '${PARAM}'], {'PARAM': 'bar'}) self.assertEquals(b'bar', result)
def test_run(self): result = SyncRemoteProcess.run('localhost', ['/bin/echo', '-n', 'foo']) self.assertEqual(b'foo', result)