def setUp(self): self._server = SocketWriter() self.stream = MorseStream('localhost', 60000, maxlength=3)
def setUp(self): self._server = SocketWriter() self.stream = MorseStream('localhost', 60000, maxlength = 3)
class TestPyMorseStream(unittest.TestCase): def setUp(self): self._server = SocketWriter() self.stream = MorseStream('localhost', 60000, maxlength=3) def test_get(self): self.assertEqual(self.stream.get(), [0]) self.assertEqual(self.stream.get(), [1]) def test_last(self): self.assertIsNone(self.stream.last()) self.stream.get() self.assertEqual(self.stream.last(), [0]) self.stream.get() self.assertEqual(self.stream.last(), [1]) def test_last_n(self): self.assertIsNone(self.stream.last()) self.assertIsNone(self.stream.last(2)) self.assertIsNone(self.stream.last(10)) self.stream.get() self.assertEqual(self.stream.last(1), [[0]]) self.assertEqual(self.stream.last(2), [[0]]) self.stream.get() self.assertEqual(self.stream.last(1), [[1]]) self.assertEqual(self.stream.last(2), [[1], [0]]) self.stream.get() self.assertEqual(self.stream.last(3), [[2], [1], [0]]) self.stream.get() self.assertEqual(self.stream.last(3), [[3], [2], [1]]) self.assertEqual(self.stream.last(4), [[3], [2], [1]]) def test_subscribe(self): self.ok = False self.i = 0 self.stream.subscribe(self.on_data) self.stream.get() time.sleep(0.2) self.assertTrue(self.ok) self.ok = False self.stream.get() time.sleep(0.2) self.assertTrue(self.ok) def test_unsubscribe(self): self.ok = False self.i = 0 self.stream.subscribe(self.on_data) self.stream.get() time.sleep(0.2) self.assertTrue(self.ok) self.ok = False self.stream.unsubscribe(self.on_data) self.stream.get() time.sleep(0.2) self.assertFalse(self.ok) def on_data(self, record): self.ok = True self.assertEqual(record, [self.i]) self.i += 1 def tearDown(self): self.stream.close() self._server.close()
class TestPyMorseStream(unittest.TestCase): def setUp(self): self._server = SocketWriter() self.stream = MorseStream('localhost', 60000, maxlength = 3) def test_get(self): self.assertEqual(self.stream.get(), [0]) self.assertEqual(self.stream.get(), [1]) def test_last(self): self.assertIsNone(self.stream.last()) self.stream.get() self.assertEqual(self.stream.last(), [0]) self.stream.get() self.assertEqual(self.stream.last(), [1]) def test_last_n(self): self.assertIsNone(self.stream.last()) self.assertIsNone(self.stream.last(2)) self.assertIsNone(self.stream.last(10)) self.stream.get() self.assertEqual(self.stream.last(1), [[0]]) self.assertEqual(self.stream.last(2), [[0]]) self.stream.get() self.assertEqual(self.stream.last(1), [[1]]) self.assertEqual(self.stream.last(2), [[1],[0]]) self.stream.get() self.assertEqual(self.stream.last(3), [[2],[1],[0]]) self.stream.get() self.assertEqual(self.stream.last(3), [[3], [2],[1]]) self.assertEqual(self.stream.last(4), [[3], [2],[1]]) def test_subscribe(self): self.ok = False self.i = 0 self.stream.subscribe(self.on_data) self.stream.get() time.sleep(0.2) self.assertTrue(self.ok) self.ok = False self.stream.get() time.sleep(0.2) self.assertTrue(self.ok) def test_unsubscribe(self): self.ok = False self.i = 0 self.stream.subscribe(self.on_data) self.stream.get() time.sleep(0.2) self.assertTrue(self.ok) self.ok = False self.stream.unsubscribe(self.on_data) self.stream.get() time.sleep(0.2) self.assertFalse(self.ok) def on_data(self, record): self.ok = True self.assertEqual(record, [self.i]) self.i += 1 def tearDown(self): self.stream.close() self._server.close()