def test_producer_init_simplest(self): producer = Producer(Mock()) self.assertEqual( producer.__repr__(), "<Producer <class 'afkak.partitioner.RoundRobinPartitioner'>:" "Unbatched:1:1000>") producer.stop()
def test_producer_init_batch(self): producer = Producer(Mock(), batch_send=True) looper = producer.sendLooper self.assertEqual(type(looper), LoopingCall) self.assertTrue(looper.running) producer.stop() self.assertFalse(looper.running) self.assertEqual( producer.__repr__(), "<Producer <class 'afkak.partitioner.RoundRobinPartitioner'>:" "10cnt/32768bytes/30secs:1:1000>")
def test_producer_bad_codec_type(self): with self.assertRaises(TypeError): p = Producer(Mock(), codec='bogus') p.__repr__() # pragma: no cover # STFU pyflakes
def test_producer_bad_codec_value(self): with self.assertRaises(UnsupportedCodecError): p = Producer(Mock(), codec=99) p.__repr__() # pragma: no cover # STFU pyflakes
def test_producer_non_integral_batch_every_b(self): client = Mock() with self.assertRaises(TypeError): producer = Producer(client, batch_send=True, batch_every_b="10") producer.__repr__() # pragma: no cover # STFU pyflakes