scroll_test.rb
2.39 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
require_relative "test_helper"
class ScrollTest < Minitest::Test
def test_scroll
store_names ["Product A", "Product B", "Product C", "Product D", "Product E", "Product F"]
products = Product.search("product", order: {name: :asc}, scroll: '1m', per_page: 2)
assert_equal ["Product A", "Product B"], products.map(&:name)
assert_equal "product", products.entry_name
assert_equal 2, products.size
assert_equal 2, products.length
assert_equal 6, products.total_count
assert_equal 6, products.total_entries
assert products.any?
# scroll for next 2
products = products.scroll
assert_equal ["Product C", "Product D"], products.map(&:name)
# scroll for next 2
products = products.scroll
assert_equal ["Product E", "Product F"], products.map(&:name)
# scroll exhausted
products = products.scroll
assert_equal [], products.map(&:name)
end
def test_scroll_body
store_names ["Product A", "Product B", "Product C", "Product D", "Product E", "Product F"]
products = Product.search("product", body: {query: {match_all: {}}, sort: [{name: "asc"}]}, scroll: '1m', per_page: 2)
assert_equal ["Product A", "Product B"], products.map(&:name)
assert_equal "product", products.entry_name
assert_equal 2, products.size
assert_equal 2, products.length
assert_equal 6, products.total_count
assert_equal 6, products.total_entries
assert products.any?
# scroll for next 2
products = products.scroll
assert_equal ["Product C", "Product D"], products.map(&:name)
# scroll for next 2
products = products.scroll
assert_equal ["Product E", "Product F"], products.map(&:name)
# scroll exhausted
products = products.scroll
assert_equal [], products.map(&:name)
end
def test_scroll_all
store_names ["Product A"]
assert_equal ["Product A"], Product.search("*", scroll: "1m").map(&:name)
end
def test_scroll_no_option
products = Product.search("*")
error = assert_raises Searchkick::Error do
products.scroll
end
assert_match /Pass .+ option/, error.message
end
def test_scroll_block
store_names ["Product A", "Product B", "Product C", "Product D", "Product E", "Product F"]
batches_count = 0
Product.search("*", scroll: "1m", per_page: 2).scroll do |batch|
assert_equal 2, batch.size
batches_count += 1
end
assert_equal 3, batches_count
end
end