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
|
from tests.test_helper import *
class TestResourceCollection(unittest.TestCase):
class TestResource:
items = ["a", "b", "c", "d", "e"]
@staticmethod
def fetch(query, ids):
return [TestResourceCollection.TestResource.items[int(id)] for id in ids]
def test_iterating_over_contents(self):
collection_data = {
"search_results": {
"page_size": 2,
"ids": ["0", "1", "2", "3", "4"]
}
}
collection = ResourceCollection("some_query", collection_data, TestResourceCollection.TestResource.fetch)
new_items = []
index = 0
for item in collection.items:
self.assertEquals(TestResourceCollection.TestResource.items[index], item)
new_items.append(item)
index += 1
self.assertEquals(5, len(new_items))
|
