Skip to content

Commit 1512275

Browse files
committed
batch data test
1 parent c21867e commit 1512275

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

tests/test_client.py

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -66,6 +66,16 @@ def test_send_and_receive(self):
6666
data = io.receive('TestFeed')
6767
self.assertEqual(data.value, 'foo')
6868

69+
def test_send_batch_data(self):
70+
io = self.get_client()
71+
self.ensure_feed_deleted(io, 'TestFeed')
72+
feed = Feed(name="TestFeed")
73+
test_feed = aio.create_feed(feed)
74+
data_list = [Data(value='batch'), Data(value='batch')]
75+
io.send_batch_data(test_feed.key, 'foo')
76+
data = io.receive('TestFeed')
77+
self.assertEqual(data.value, 'batch')
78+
6979
def test_receive_next(self):
7080
io = self.get_client()
7181
self.ensure_feed_deleted(io, 'TestFeed')
@@ -108,6 +118,8 @@ def test_data_on_feed_and_data_id_returns_data(self):
108118
def test_create_data(self):
109119
io = self.get_client()
110120
self.ensure_feed_deleted(io, 'TestFeed')
121+
feed = Feed(name="TestFeed")
122+
test_feed = aio.create_feed(feed)
111123
io.send_data('TestFeed', 1) # Make sure TestFeed exists.
112124
data = Data(value=42)
113125
result = io.create_data('TestFeed', data)

0 commit comments

Comments
 (0)