mirror of
https://github.com/LBRYFoundation/lbry-sdk.git
synced 2025-08-23 17:27:25 +00:00
19 lines
645 B
Python
19 lines
645 B
Python
import unittest
|
|
from torba.stream import StreamController
|
|
|
|
class StreamControllerTestCase(unittest.TestCase):
|
|
def test_non_unique_events(self):
|
|
events = []
|
|
controller = StreamController()
|
|
controller.stream.listen(on_data=events.append)
|
|
controller.add("yo")
|
|
controller.add("yo")
|
|
self.assertEqual(events, ["yo", "yo"])
|
|
|
|
def test_unique_events(self):
|
|
events = []
|
|
controller = StreamController(merge_repeated_events=True)
|
|
controller.stream.listen(on_data=events.append)
|
|
controller.add("yo")
|
|
controller.add("yo")
|
|
self.assertEqual(events, ["yo"])
|