Skip to content

Commit 0495b89

Browse files
author
brentru
committed
refactoring and re-testing client unit tests for api-v2 release
1 parent 8714a0a commit 0495b89

File tree

1 file changed

+32
-36
lines changed

1 file changed

+32
-36
lines changed

tests/test_client.py

Lines changed: 32 additions & 36 deletions
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@ class TestClient(base.IOTestCase):
2727

2828
def get_client(self):
2929
# Construct an Adafruit IO REST client and return it.
30-
return Client(self.get_test_username(), self.get_test_key(), proxies=PROXIES, base_url=BASE_URL)
30+
return Client(self.get_test_username(), self.get_test_key(), proxies=PROXIES, base_url=BASE_URL, api_version = "v2")
3131

3232
def ensure_feed_deleted(self, client, feed):
3333
# Delete the specified feed if it exists.
@@ -50,6 +50,7 @@ def empty_feed(self, client, feed):
5050
data = client.data(feed)
5151
for d in data:
5252
client.delete(feed, d.id)
53+
5354

5455
def test_set_key_and_username(self):
5556
username = "unique_username"
@@ -60,27 +61,28 @@ def test_set_key_and_username(self):
6061

6162
def test_send_and_receive(self):
6263
io = self.get_client()
63-
self.ensure_feed_deleted(io, 'TestFeed')
64-
response = io.send_data('TestFeed', 'foo')
64+
self.ensure_feed_deleted(io, 'testfeed')
65+
test_feed = io.create_feed(Feed(name="testfeed"))
66+
response = io.send_data('testfeed', 'foo')
6567
self.assertEqual(response.value, 'foo')
66-
data = io.receive('TestFeed')
68+
data = io.receive('testfeed')
6769
self.assertEqual(data.value, 'foo')
6870

6971
def test_send_batch_data(self):
7072
io = self.get_client()
71-
self.ensure_feed_deleted(io, 'TestFeed')
72-
feed = Feed(name="TestFeed")
73-
test_feed = aio.create_feed(feed)
73+
self.ensure_feed_deleted(io, 'testfeed')
74+
test_feed = io.create_feed(Feed(name="testfeed"))
7475
data_list = [Data(value='batch'), Data(value='batch')]
7576
io.send_batch_data(test_feed.key, 'foo')
76-
data = io.receive('TestFeed')
77+
data = io.receive('testfeed')
7778
self.assertEqual(data.value, 'batch')
7879

7980
def test_receive_next(self):
8081
io = self.get_client()
81-
self.ensure_feed_deleted(io, 'TestFeed')
82-
io.send_data('TestFeed', 1)
83-
data = io.receive_next('TestFeed')
82+
self.ensure_feed_deleted(io, 'testfeed')
83+
test_feed = io.create_feed(Feed(name="testfeed"))
84+
io.send_data('testfeed', 1)
85+
data = io.receive_next('testfeed')
8486
self.assertEqual(int(data.value), 1)
8587

8688
# BUG: Previous jumps too far back: https://github.com/adafruit/io/issues/55
@@ -99,53 +101,47 @@ def test_receive_previous(self):
99101

100102
def test_data_on_feed_returns_all_data(self):
101103
io = self.get_client()
102-
self.ensure_feed_deleted(io, 'TestFeed')
103-
io.send_data('TestFeed', 1)
104-
io.send_data('TestFeed', 2)
105-
result = io.data('TestFeed')
104+
self.ensure_feed_deleted(io, 'testfeed')
105+
test_feed = io.create_feed(Feed(name="testfeed"))
106+
io.send_data('testfeed', 1)
107+
io.send_data('testfeed', 2)
108+
result = io.data('testfeed')
106109
self.assertEqual(len(result), 2)
107110
self.assertEqual(int(result[0].value), 2)
108111
self.assertEqual(int(result[1].value), 1)
109112

110113
def test_data_on_feed_and_data_id_returns_data(self):
111114
io = self.get_client()
112-
self.ensure_feed_deleted(io, 'TestFeed')
113-
data = io.send_data('TestFeed', 1)
114-
result = io.data('TestFeed', data.id)
115+
self.ensure_feed_deleted(io, 'testfeed')
116+
test_feed = io.create_feed(Feed(name="testfeed"))
117+
data = io.send_data('testfeed', 1)
118+
result = io.data('testfeed', data.id)
115119
self.assertEqual(data.id, result.id)
116120
self.assertEqual(int(data.value), int(result.value))
117121

118122
def test_create_data(self):
119123
aio = self.get_client()
120-
self.ensure_feed_deleted(aio, 'TestFeed')
121-
feed = Feed(name="TestFeed")
122-
test_feed = aio.create_feed(feed)
123-
aio.send_data('TestFeed', 1) # Make sure TestFeed exists.
124+
self.ensure_feed_deleted(aio, 'testfeed')
125+
test_feed = aio.create_feed(Feed(name="testfeed"))
126+
aio.send_data('testfeed', 1) # Make sure TestFeed exists.
124127
data = Data(value=42)
125-
result = aio.create_data('TestFeed', data)
128+
result = aio.create_data('testfeed', data)
126129
self.assertEqual(int(result.value), 42)
127130

128131
def test_append_by_feed_name(self):
129132
io = self.get_client()
130-
self.ensure_feed_deleted(io, 'TestFeed')
131-
feed = io.create_feed(Feed(name='TestFeed'))
132-
result = io.append('TestFeed', 42)
133+
self.ensure_feed_deleted(io, 'testfeed')
134+
feed = io.create_feed(Feed(name='testfeed'))
135+
result = io.append('testfeed', 42)
133136
self.assertEqual(int(result.value), 42)
134137

135138
def test_append_by_feed_key(self):
136139
io = self.get_client()
137-
self.ensure_feed_deleted(io, 'Test Feed Fancy Name')
138-
feed = io.create_feed(Feed(name='Test Feed Fancy Name'))
140+
self.ensure_feed_deleted(io, 'test feed fancy name')
141+
feed = io.create_feed(Feed(name='test feed fancy name'))
139142
result = io.append(feed.key, 42)
140143
self.assertEqual(int(result.value), 42)
141144

142-
def test_append_by_feed_id(self):
143-
io = self.get_client()
144-
self.ensure_feed_deleted(io, 'TestFeed')
145-
feed = io.create_feed(Feed(name='TestFeed'))
146-
result = io.append(feed.id, 42)
147-
self.assertEqual(int(result.value), 42)
148-
149145
def test_create_feed(self):
150146
io = self.get_client()
151147
self.ensure_feed_deleted(io, 'TestFeed')
@@ -219,4 +215,4 @@ def test_delete_group(self):
219215

220216
# TODO: Get by group name, key, and ID
221217
# TODO: Get data by name, key, ID
222-
# TODO: Tests around Adafruit IO keys (make multiple, test they work, etc.)
218+
# TODO: Tests around Adafruit IO keys (make multiple, test they work, etc.)

0 commit comments

Comments
 (0)