def test_keep_streaming(self):
dataset = Dataset(
self.session,
"Foo Name",
title=self._TITLE,
create=True,
independents=self._INDEPENDENTS,
dependents=self._DEPENDENTS)
data = self._get_records_simple([(1, 2, 3)], dataset.data.dtype)
dataset.addData(data)
listener = 'listener'
# Start streaming the listener
dataset.keepStreaming(listener, 0)
# Check the listener is notified of the data already available
self.hub.onDataAvailable.assert_called_with(None, [listener])
self.hub.reset_mock()
# Keep streaming for more data added
dataset.keepStreaming(listener, 1)
# Add more data.
dataset.addData(data)
# Trigger the listener again.
self.hub.onDataAvailable.assert_called_with(None, set([listener]))
def test_keep_streaming(self):
dataset = Dataset(self.session,
"Foo Name",
title=self._TITLE,
create=True,
independents=self._INDEPENDENTS,
dependents=self._DEPENDENTS)
data = self._get_records_simple([(1, 2, 3)], dataset.data.dtype)
dataset.addData(data)
listener = 'listener'
# Start streaming the listener
dataset.keepStreaming(listener, 0)
# Check the listener is notified of the data already available
self.hub.onDataAvailable.assert_called_with(None, [listener])
self.hub.reset_mock()
# Keep streaming for more data added
dataset.keepStreaming(listener, 1)
# Add more data.
dataset.addData(data)
# Trigger the listener again.
self.hub.onDataAvailable.assert_called_with(None, set([listener]))
