def test_http_cs(self):
proc = Process(encoding='utf-8')
proc.spawn([sys.executable, 'httpcs.py'], stdout=PIPE)
result = proc.stdout.read()
self.assertEqual(result, 'result = pong\n')
proc.wait(timeout=1)
self.assertEqual(proc.returncode, 0)
proc.close()
def test_netcat(self):
proc = Process()
proc.spawn([sys.executable, '-u', 'netcat.py', 'xkcd.com', '80'], stdin=PIPE, stdout=PIPE)
proc.stdin.write(b'GET / HTTP/1.1\r\nHost: xkcd.com\r\nConnection: close\r\n\r\n')
result = proc.stdout.read()
self.assertTrue(b'<html' in result)
proc.wait(timeout=1)
self.assertEqual(proc.returncode, 0)
proc.close()
def test_fortune(self):
proc = Process(encoding='ascii')
proc.spawn([sys.executable, '-u', 'fortune.py'], stdout=PIPE)
line = proc.stdout.readline()
self.assertTrue(line.startswith('Listen on '))
addr = gruvi.paddr(line[10:])
client = HttpClient()
client.connect(addr)
client.request('GET', '/')
response = client.getresponse()
fortune = response.read().decode('ascii')
client.close()
self.assertTrue('Albert Einstein' in fortune)
proc.send_signal(signal.SIGINT)
proc.wait(timeout=1)
self.assertEqual(proc.returncode, 0)
proc.close()
def test_echoserver2(self):
proc = Process(encoding='ascii')
proc.spawn([sys.executable, '-u', 'echoserver2.py'], stdout=PIPE)
line = proc.stdout.readline()
self.assertTrue(line.startswith('Listen on '))
addr = gruvi.paddr(line[10:])
client = StreamClient()
client.connect(addr)
client.write(b'foo')
self.assertEqual(client.read(3), b'foo')
client.write(b'foo bar baz\n')
self.assertEqual(client.readline(), b'foo bar baz\n')
client.close()
proc.send_signal(signal.SIGINT)
proc.wait(timeout=1)
self.assertEqual(proc.returncode, 0)
proc.close()