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
|
defmodule PartitionHelpers do
use ExUnit.Case
def create_partition_docs(db_name, pk1 \\ "foo", pk2 \\ "bar") do
docs =
for i <- 1..100 do
id =
if rem(i, 2) == 0 do
"#{pk1}:#{i}"
else
"#{pk2}:#{i}"
end
group =
if rem(i, 3) == 0 do
"one"
else
"two"
end
%{
:_id => id,
:value => i,
:some => "field",
:group => group
}
end
resp = Couch.post("/#{db_name}/_bulk_docs", body: %{:w => 3, :docs => docs})
assert resp.status_code in [201, 202]
end
def create_partition_ddoc(db_name, opts \\ %{}) do
map_fn = """
function(doc) {
if (doc.some) {
emit(doc.value, doc.some);
}
}
"""
default_ddoc = %{
views: %{
some: %{
map: map_fn
}
}
}
ddoc = Enum.into(opts, default_ddoc)
resp = Couch.put("/#{db_name}/_design/mrtest", body: ddoc)
assert resp.status_code in [201, 202]
assert Map.has_key?(resp.body, "ok") == true
end
def get_ids(resp) do
%{:body => %{"rows" => rows}} = resp
Enum.map(rows, fn row -> row["id"] end)
end
def get_partitions(resp) do
%{:body => %{"rows" => rows}} = resp
Enum.map(rows, fn row ->
[partition, _] = String.split(row["id"], ":")
partition
end)
end
def assert_correct_partition(partitions, correct_partition) do
assert Enum.all?(partitions, fn partition ->
partition == correct_partition
end)
end
end
|
