openstack/deb-python-kafka
Code Issues Proposed changes

Integration test writes to a file instead of stdout

Browse Source
This commit is contained in:
David Arthur
2012-10-30 10:40:41 -04:00
parent d8cf862947
commit 8ca656d380
1 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
test/integration.py
View File

@@ -6,7 +6,6 @@ import shlex
import shutil import shutil
import socket import socket
import subprocess import subprocess
import sys
import tempfile import tempfile
from threading import Thread, Event from threading import Thread, Event
import time import time
@@ -44,6 +43,7 @@ class KafkaFixture(Thread):
# Create the log directory # Create the log directory
logDir = os.path.join(self.tmpDir, 'logs') logDir = os.path.join(self.tmpDir, 'logs')
os.mkdir(logDir) os.mkdir(logDir)
stdout = open(os.path.join(logDir, 'stdout'), 'w')
# Create the config file # Create the config file
logConfig = "test/resources/log4j.properties" logConfig = "test/resources/log4j.properties"
@@ -63,7 +63,7 @@ class KafkaFixture(Thread):
(rlist, wlist, xlist) = select.select([proc.stdout], [], [], 1) (rlist, wlist, xlist) = select.select([proc.stdout], [], [], 1)
if proc.stdout in rlist: if proc.stdout in rlist:
read = proc.stdout.readline() read = proc.stdout.readline()
sys.stdout.write(read) stdout.write(read)
self.capture += read self.capture += read
if self.shouldDie.is_set(): if self.shouldDie.is_set():
@@ -71,7 +71,7 @@ class KafkaFixture(Thread):
killed = True killed = True
if proc.poll() is not None: if proc.poll() is not None:
shutil.rmtree(self.tmpDir) #shutil.rmtree(self.tmpDir)
if killed: if killed:
break break
else: else:
@@ -206,13 +206,14 @@ class IntegrationTest(unittest.TestCase):
t1 = int(time.time()*1000) # now t1 = int(time.time()*1000) # now
t2 = t1 + 60000 # one minute from now t2 = t1 + 60000 # one minute from now
req = OffsetRequest("test-offset-request", 0, t1, 1024) req = OffsetRequest("test-offset-request", 0, t1, 1024)
print self.kafka.get_offsets(req) self.kafka.get_offsets(req)
req = OffsetRequest("test-offset-request", 0, t2, 1024) req = OffsetRequest("test-offset-request", 0, t2, 1024)
print self.kafka.get_offsets(req) self.kafka.get_offsets(req)
def test_10k_messages(self): def test_10k_messages(self):
msg_tmpl = "this is a test message with a few bytes in it. this is message number %d" msg_tmpl = "this is a test message with a few bytes in it. this is message number %d"
# TODO 10k actually fails, why?
msg = KafkaClient.create_gzip_message(*[msg_tmpl % i for i in range(1000)]) msg = KafkaClient.create_gzip_message(*[msg_tmpl % i for i in range(1000)])
req = ProduceRequest("test-10k", 0, [msg]) req = ProduceRequest("test-10k", 0, [msg])
self.kafka.send_message_set(req) self.kafka.send_message_set(req)