Ejemplo n.º 1
0
 def test_next_limit(self):
     q = Queue()
     upload_size = 50
     consumer = Consumer(q, '', upload_size)
     for i in range(10000):
         q.put(i)
     next = consumer.next()
     self.assertEqual(next, list(range(upload_size)))
Ejemplo n.º 2
0
 def test_dropping_oversize_msg(self):
     q = Queue()
     consumer = Consumer(q, '')
     oversize_msg = {'m': 'x' * MAX_MSG_SIZE}
     q.put(oversize_msg)
     next = consumer.next()
     self.assertEqual(next, [])
     self.assertTrue(q.empty())
Ejemplo n.º 3
0
 def test_next_limit(self):
     q = Queue()
     flush_at = 50
     consumer = Consumer(q, '', flush_at)
     for i in range(10000):
         q.put(i)
     next = consumer.next()
     self.assertEqual(next, list(range(flush_at)))
Ejemplo n.º 4
0
 def test_next_limit(self):
     q = Queue()
     upload_size = 50
     consumer = Consumer(q, '', upload_size)
     for i in range(10000):
         q.put(i)
     next = consumer.next()
     self.assertEqual(next, list(range(upload_size)))
Ejemplo n.º 5
0
 def test_dropping_oversize_msg(self):
     q = Queue()
     consumer = Consumer(q, '')
     oversize_msg = {'m': 'x' * MAX_MSG_SIZE}
     q.put(oversize_msg)
     next = consumer.next()
     self.assertEqual(next, [])
     self.assertTrue(q.empty())
Ejemplo n.º 6
0
 def test_next(self):
     q = Queue()
     consumer = Consumer(q, '')
     q.put(1)
     next = consumer.next()
     self.assertEqual(next, [1])
Ejemplo n.º 7
0
 def test_next(self):
     q = Queue()
     consumer = Consumer(q, '')
     q.put(1)
     next = consumer.next()
     self.assertEqual(next, [1])