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()
