forked from apache/cassandra-python-driver
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathtest_resultset.py
More file actions
125 lines (103 loc) · 5.36 KB
/
test_resultset.py
File metadata and controls
125 lines (103 loc) · 5.36 KB
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
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
# Copyright 2013-2017 DataStax, Inc.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
from cassandra.query import named_tuple_factory, dict_factory, tuple_factory
try:
import unittest2 as unittest
except ImportError:
import unittest # noqa
from mock import Mock, PropertyMock
from cassandra.cluster import ResultSet
class ResultSetTests(unittest.TestCase):
def test_iter_non_paged(self):
expected = list(range(10))
rs = ResultSet(Mock(has_more_pages=False), expected)
itr = iter(rs)
self.assertListEqual(list(itr), expected)
def test_iter_paged(self):
expected = list(range(10))
response_future = Mock(has_more_pages=True)
response_future.result.side_effect = (ResultSet(Mock(), expected[-5:]), ) # ResultSet is iterable, so it must be protected in order to be returned whole by the Mock
rs = ResultSet(response_future, expected[:5])
itr = iter(rs)
# this is brittle, depends on internal impl details. Would like to find a better way
type(response_future).has_more_pages = PropertyMock(side_effect=(True, True, False)) # after init to avoid side effects being consumed by init
self.assertListEqual(list(itr), expected)
def test_list_non_paged(self):
# list access on RS for backwards-compatibility
expected = list(range(10))
rs = list(ResultSet(Mock(has_more_pages=False), expected))
for i in range(10):
self.assertEqual(rs[i], expected[i])
self.assertEqual(rs, expected)
def test_list_paged(self):
# list access on RS for backwards-compatibility
expected = list(range(10))
response_future = Mock(has_more_pages=True)
response_future.result.side_effect = (ResultSet(Mock(), expected[-5:]), ) # ResultSet is iterable, so it must be protected in order to be returned whole by the Mock
rs = list(ResultSet(response_future, expected[:5]))
# this is brittle, depends on internal impl details. Would like to find a better way
type(response_future).has_more_pages = PropertyMock(side_effect=(True, True, True, False)) # First two True are consumed on check entering list mode
self.assertEqual(rs[9], expected[9])
self.assertEqual(rs, expected)
def test_has_more_pages(self):
response_future = Mock()
response_future.has_more_pages.side_effect = PropertyMock(side_effect=(True, False))
rs = ResultSet(response_future, [])
type(response_future).has_more_pages = PropertyMock(side_effect=(True, False)) # after init to avoid side effects being consumed by init
self.assertTrue(rs.has_more_pages)
self.assertFalse(rs.has_more_pages)
def test_eq(self):
# no pages
expected = list(range(10))
rs = list(ResultSet(Mock(has_more_pages=False), expected))
# eq before iteration causes list to be materialized
self.assertEqual(rs, expected)
# results can be iterated or indexed once we're materialized
self.assertListEqual(rs, expected)
self.assertEqual(rs[9], expected[9])
self.assertTrue(rs)
# pages
response_future = Mock(has_more_pages=True)
response_future.result.side_effect = (ResultSet(Mock(), expected[-5:]), ) # ResultSet is iterable, so it must be protected in order to be returned whole by the Mock
rs = list(ResultSet(response_future, expected[:5]))
type(response_future).has_more_pages = PropertyMock(side_effect=(True, True, True, False))
# eq before iteration causes list to be materialized
self.assertEqual(rs, expected)
# results can be iterated or indexed once we're materialized
self.assertListEqual(rs, expected)
self.assertEqual(rs[9], expected[9])
self.assertTrue(rs)
def test_bool(self):
self.assertFalse(ResultSet(Mock(has_more_pages=False), []))
self.assertTrue(ResultSet(Mock(has_more_pages=False), [1]))
def test_was_applied(self):
# unknown row factory raises
with self.assertRaises(RuntimeError):
ResultSet(Mock(), []).was_applied
response_future = Mock(row_factory=named_tuple_factory)
# no row
with self.assertRaises(RuntimeError):
ResultSet(response_future, []).was_applied
# too many rows
with self.assertRaises(RuntimeError):
ResultSet(response_future, [tuple(), tuple()]).was_applied
# various internal row factories
for row_factory in (named_tuple_factory, tuple_factory):
for applied in (True, False):
rs = ResultSet(Mock(row_factory=row_factory), [(applied,)])
self.assertEqual(rs.was_applied, applied)
row_factory = dict_factory
for applied in (True, False):
rs = ResultSet(Mock(row_factory=row_factory), [{'[applied]': applied}])
self.assertEqual(rs.was_applied, applied)