1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
|
from tests.test_helper import *
class TestResourceCollection(unittest.TestCase):
collection_data = {
"search_results": {
"page_size": 2,
"ids": ["0", "1", "2", "3", "4"]
}
}
class TestResource:
items = ["a", "b", "c", "d", "e"]
@staticmethod
def fetch(_, ids):
return [TestResourceCollection.TestResource.items[int(resource_id)] for resource_id in ids]
def test_iterating_over_contents(self):
collection = ResourceCollection("some_query", self.collection_data, TestResourceCollection.TestResource.fetch)
new_items = []
index = 0
for item in collection.items:
self.assertEqual(TestResourceCollection.TestResource.items[index], item)
new_items.append(item)
index += 1
self.assertEqual(5, len(new_items))
def test_iterate_using_iterator_protocol(self):
collection = ResourceCollection("some_query", self.collection_data, TestResourceCollection.TestResource.fetch)
for test_elem, coll_elem in zip(self.TestResource.items, collection):
self.assertEqual(test_elem, coll_elem)
def test_ids_returns_array_of_ids(self):
collection = ResourceCollection("some_query", self.collection_data, TestResourceCollection.TestResource.fetch)
self.assertEqual(collection.ids, self.collection_data['search_results']['ids'])
def test_ids_returns_array_of_empty_ids(self):
empty_collection_data = {
"search_results": {
"page_size": 2,
"ids": []
}
}
collection = ResourceCollection("some_query", empty_collection_data, TestResourceCollection.TestResource.fetch)
self.assertEqual(collection.ids, [])
def test_no_search_results(self):
bad_collection_data = {}
with self.assertRaisesRegex(UnexpectedError, "Unprocessable entity due to an invalid request"):
ResourceCollection("some_query", bad_collection_data, TestResourceCollection.TestResource.fetch)
|
