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
|
% 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.
-module(couch_replicator_bulk_get_tests).
-include_lib("couch/include/couch_eunit.hrl").
-include_lib("couch/include/couch_db.hrl").
-include("couch_replicator_test.hrl").
-define(DOC_COUNT, 10).
bulk_get_test_() ->
{
"Use _bulk_get when replicating",
{
foreach,
fun couch_replicator_test_helper:test_setup/0,
fun couch_replicator_test_helper:test_teardown/1,
[
?TDEF_FE(use_bulk_get),
?TDEF_FE(dont_use_bulk_get),
?TDEF_FE(job_enable_overrides_global_disable),
?TDEF_FE(global_disable_works)
]
}
}.
use_bulk_get({_Ctx, {Source, Target}}) ->
populate_db(Source, ?DOC_COUNT),
meck:new(couch_replicator_api_wrap, [passthrough]),
replicate(Source, Target, true),
BulkGets = meck:num_calls(couch_replicator_api_wrap, bulk_get, 3),
JustGets = meck:num_calls(couch_replicator_api_wrap, open_doc_revs, 6),
?assertEqual(0, JustGets),
?assert(BulkGets >= 1),
compare_dbs(Source, Target).
dont_use_bulk_get({_Ctx, {Source, Target}}) ->
populate_db(Source, ?DOC_COUNT),
meck:new(couch_replicator_api_wrap, [passthrough]),
replicate(Source, Target, false),
BulkGets = meck:num_calls(couch_replicator_api_wrap, bulk_get, 3),
JustGets = meck:num_calls(couch_replicator_api_wrap, open_doc_revs, 6),
?assertEqual(0, BulkGets),
?assertEqual(?DOC_COUNT, JustGets),
compare_dbs(Source, Target).
job_enable_overrides_global_disable({_Ctx, {Source, Target}}) ->
populate_db(Source, ?DOC_COUNT),
Persist = false,
config:set("replicator", "use_bulk_get", "false", Persist),
meck:new(couch_replicator_api_wrap, [passthrough]),
replicate(Source, Target, true),
BulkGets = meck:num_calls(couch_replicator_api_wrap, bulk_get, 3),
JustGets = meck:num_calls(couch_replicator_api_wrap, open_doc_revs, 6),
?assertEqual(0, JustGets),
?assert(BulkGets >= 1),
compare_dbs(Source, Target).
global_disable_works({_Ctx, {Source, Target}}) ->
populate_db(Source, ?DOC_COUNT),
Persist = false,
config:set("replicator", "use_bulk_get", "false", Persist),
meck:new(couch_replicator_api_wrap, [passthrough]),
replicate(Source, Target),
BulkGets = meck:num_calls(couch_replicator_api_wrap, bulk_get, 3),
JustGets = meck:num_calls(couch_replicator_api_wrap, open_doc_revs, 6),
?assertEqual(0, BulkGets),
?assertEqual(?DOC_COUNT, JustGets),
compare_dbs(Source, Target).
populate_db(DbName, DocCount) ->
Fun = fun(Id, Acc) -> [#doc{id = integer_to_binary(Id)} | Acc] end,
Docs = lists:foldl(Fun, [], lists:seq(1, DocCount)),
{ok, _} = fabric:update_docs(DbName, Docs, [?ADMIN_CTX]).
compare_dbs(Source, Target) ->
couch_replicator_test_helper:cluster_compare_dbs(Source, Target).
db_url(DbName) ->
couch_replicator_test_helper:cluster_db_url(DbName).
replicate(Source, Target) ->
couch_replicator_test_helper:replicate(
{[
{<<"source">>, db_url(Source)},
{<<"target">>, db_url(Target)},
{<<"worker_processes">>, <<"1">>}
]}
).
replicate(Source, Target, UseBulkGet) ->
couch_replicator_test_helper:replicate(
{[
{<<"source">>, db_url(Source)},
{<<"target">>, db_url(Target)},
{<<"worker_processes">>, <<"1">>},
{<<"use_bulk_get">>, UseBulkGet}
]}
).
|
