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