def test_get_process(self):
n = JubaNode('localhost', range(10000,10003), None, '/tmp', [])
proc = n.get_process(['sleep', '100'])
proc.start()
time.sleep(1)
proc.stop()
self.assertEqual(b'', proc.stdout)
def test_pool_doublefree(self):
n = JubaNode('localhost', range(10000,10003), None, '/tmp', [])
port1 = n.lease_port()
n.free_port(port1)
self.assertRaises(JubaTestAssertionError, n.free_port, port1)
def test_get_file(self):
n = JubaNode('localhost', range(10000,10003), None, '/tmp', [])
with tempfile.NamedTemporaryFile() as tmp1, tempfile.NamedTemporaryFile() as tmp2:
tmp1.write(b'bar')
tmp1.flush()
n.get_file(tmp1.name, tmp2.name)
self.assertEqual(b'bar', tmp2.read())
def test_pool_skip(self):
n = JubaNode('localhost', range(10000,10003), None, '/tmp', [])
port1 = n.lease_port()
port2 = n.lease_port()
port3 = n.lease_port()
self.assertRaises(JubaSkipTest, n.lease_port)
def test_put_file_delete_file(self):
n = JubaNode('localhost', range(10000,10003), None, '/tmp', [])
path = n.put_file(b'foo')
self.assertTrue(os.path.isfile(path))
with open(path, 'rb') as f:
self.assertEqual(b'foo', f.read())
n.delete_file(path)
self.assertFalse(os.path.isfile(path))
def test_pool_3(self):
n = JubaNode('localhost', range(10000,10003), None, '/tmp', [])
port1 = n.lease_port()
port2 = n.lease_port()
self.assertEqual(10000, port1)
self.assertEqual(10001, port2)
self.assertEqual(3, len(n._ports))
self.assertEqual(1, len(n._free_ports))
self.assertEqual(2, n.ports_used())
n.free_port(port1)
n.free_port(port2)
self.assertEqual(3, len(n._ports))
self.assertEqual(3, len(n._free_ports))
self.assertEqual(0, n.ports_used())
port3 = n.lease_port()
self.assertEqual(10002, port3)
port4 = n.lease_port()
self.assertEqual(10000, port4)
def test_pool_1(self):
n = JubaNode('localhost', range(10,11), None, '/tmp', [])
port = n.lease_port()
self.assertEqual(10, port)
self.assertEqual(1, len(n._ports))
self.assertEqual(0, len(n._free_ports))
n.free_port(port)
self.assertEqual(1, len(n._ports))
self.assertEqual(1, len(n._free_ports))
def test_pool_1(self):
n = JubaNode('localhost', range(10,11), None, '/tmp', [])
port = n.lease_port()
self.assertEqual(10, port)
self.assertEqual(1, len(n._ports))
self.assertEqual(0, len(n._free_ports))
n.free_port(port)
self.assertEqual(1, len(n._ports))
self.assertEqual(1, len(n._free_ports))
def test_put_file(self):
n = JubaNode('localhost', range(10000,10003), None, '/tmp', [])
with tempfile.NamedTemporaryFile() as tmp:
n.put_file(b'foo', tmp.name)
self.assertEqual(b'foo', tmp.read())
def test_pool_nonmember(self):
n = JubaNode('localhost', range(10000,10003), None, '/tmp', [])
self.assertRaises(JubaTestAssertionError, n.free_port, 50000)
def setUp(self):
self.node = JubaNode('127.0.0.1', [12345], None, '/tmp', [])
self.instance = JubaRPCServer(self.node, CLASSIFIER, [])
self.stub_instance = JubaRPCServerStub(self.node)
def test_run_process(self):
n = JubaNode('localhost', range(10000,10003), None, '/tmp', [])
contents_remote = n.run_process(['cat', '/etc/hosts'])
with open('/etc/hosts', 'rb') as f:
contents_local= f.read()
self.assertEqual(contents_local, contents_remote)
def test_get_file_temp(self):
n = JubaNode('localhost', range(10000,10003), None, '/tmp', [])
contents_remote = n.get_file('/etc/hosts')
with open('/etc/hosts', 'rb') as f:
contents_local= f.read()
self.assertEqual(contents_local, contents_remote)
