def test_pipe(self): """Tests basic TQM predicaments""" tqm = TQM() reader = tqm.get_reader_for('dupa') writer = tqm.get_writer_for('dupa') writer.put_nowait('wtf') self.assertEqual(reader.get(), 'wtf')
def test_tqm_blockiterablereader(self): tqm = TQM() reader = tqm.get_reader_for('dupa', 1) writer = tqm.get_writer_for('dupa') writer.put('A') writer.put('B') i = 0 for msg in reader: self.assertEquals(msg, {0: 'A', 1: 'B'}[i]) i += 1 writer.put('C') writer.put('D') i = 0 for msg in reader: self.assertEquals(msg, {0: 'C', 1: 'D'}[i]) i += 1