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
126
127
128
129
130
131
132
133
134
|
import unittest.mock
import pytest
from szurubooru import search
from szurubooru.func import cache
def test_retrieving_from_cache():
config = unittest.mock.MagicMock()
with unittest.mock.patch("szurubooru.func.cache.has"), unittest.mock.patch(
"szurubooru.func.cache.get"
):
cache.has.side_effect = lambda *args: True
executor = search.Executor(config)
executor.execute("test:whatever", 1, 10)
assert cache.get.called
def test_putting_equivalent_queries_into_cache():
config = search.configs.PostSearchConfig()
with unittest.mock.patch("szurubooru.func.cache.has"), unittest.mock.patch(
"szurubooru.func.cache.put"
):
hashes = []
def appender(key, _value):
hashes.append(key)
cache.has.side_effect = lambda *args: False
cache.put.side_effect = appender
executor = search.Executor(config)
executor.execute("safety:safe test", 1, 10)
executor.execute("safety:safe test", 1, 10)
executor.execute("safety:safe test ", 1, 10)
executor.execute(" safety:safe test", 1, 10)
executor.execute(" SAFETY:safe test", 1, 10)
executor.execute("test safety:safe", 1, 10)
assert len(hashes) == 6
assert len(set(hashes)) == 1
def test_putting_non_equivalent_queries_into_cache():
config = search.configs.PostSearchConfig()
with unittest.mock.patch("szurubooru.func.cache.has"), unittest.mock.patch(
"szurubooru.func.cache.put"
):
hashes = []
def appender(key, _value):
hashes.append(key)
cache.has.side_effect = lambda *args: False
cache.put.side_effect = appender
executor = search.Executor(config)
args = [
("", 1, 10),
("creation-time:2016", 1, 10),
("creation-time:2015", 1, 10),
("creation-time:2016-01", 1, 10),
("creation-time:2016-02", 1, 10),
("creation-time:2016-01-01", 1, 10),
("creation-time:2016-01-02", 1, 10),
("tag-count:1,3", 1, 10),
("tag-count:1,2", 1, 10),
("tag-count:1", 1, 10),
("tag-count:1..3", 1, 10),
("tag-count:1..4", 1, 10),
("tag-count:2..3", 1, 10),
("tag-count:1..", 1, 10),
("tag-count:2..", 1, 10),
("tag-count:..3", 1, 10),
("tag-count:..4", 1, 10),
("-tag-count:1..3", 1, 10),
("-tag-count:1..4", 1, 10),
("-tag-count:2..3", 1, 10),
("-tag-count:1..", 1, 10),
("-tag-count:2..", 1, 10),
("-tag-count:..3", 1, 10),
("-tag-count:..4", 1, 10),
("safety:safe", 1, 10),
("safety:safe", 1, 20),
("safety:safe", 2, 10),
("safety:sketchy", 1, 10),
("safety:safe test", 1, 10),
("-safety:safe", 1, 10),
("-safety:safe", 1, 20),
("-safety:safe", 2, 10),
("-safety:sketchy", 1, 10),
("-safety:safe test", 1, 10),
("safety:safe -test", 1, 10),
("-test", 1, 10),
]
for arg in args:
executor.execute(*arg)
assert len(hashes) == len(args)
assert len(set(hashes)) == len(args)
@pytest.mark.parametrize(
"input",
[
"special:fav",
"special:liked",
"special:disliked",
"-special:fav",
"-special:liked",
"-special:disliked",
],
)
def test_putting_auth_dependent_queries_into_cache(user_factory, input):
config = search.configs.PostSearchConfig()
with unittest.mock.patch("szurubooru.func.cache.has"), unittest.mock.patch(
"szurubooru.func.cache.put"
):
hashes = []
def appender(key, _value):
hashes.append(key)
cache.has.side_effect = lambda *args: False
cache.put.side_effect = appender
executor = search.Executor(config)
executor.config.user = user_factory()
executor.execute(input, 1, 1)
assert len(set(hashes)) == 1
executor.config.user = user_factory()
executor.execute(input, 1, 1)
assert len(set(hashes)) == 2
executor.execute(input, 1, 1)
assert len(set(hashes)) == 2
|
