def setUp(self): self.url = "localhost:9092" self.topic = "psutil-kafka-topic" self.producer = PsutilsKafkaProducer(self.url, self.topic, reportInterval=5) self.thread = threading.Thread(target=self.producer.start) self.thread.daemon = True
def setUp(self): self.url = "192.168.86.10:9092" self.topic = "psutil-kafka-topic" self.producer = PsutilsKafkaProducer(self.url, self.topic, reportInterval=5) self.consumer = kafka.SimpleConsumer(kafka.KafkaClient(self.url), "group1", self.topic) self.thread = threading.Thread(target=self.producer.start) self.thread.daemon = True
class PsutilsKafkaProducerTest(unittest.TestCase): def setUp(self): self.url = "localhost:9092" self.topic = "psutil-kafka-topic" self.producer = PsutilsKafkaProducer(self.url, self.topic, reportInterval=5) self.thread = threading.Thread(target=self.producer.start) self.thread.daemon = True def testStart(self): self.thread.start() time.sleep(15) self.producer.stop() message = kafka.SimpleConsumer(kafka.KafkaClient(self.url), "group1", self.topic).get_message() assert message is not None