diff options
author | Paul J. Davis <paul.joseph.davis@gmail.com> | 2019-12-25 11:37:19 -0600 |
---|---|---|
committer | Paul J. Davis <paul.joseph.davis@gmail.com> | 2019-12-25 13:52:32 -0600 |
commit | cde714908311377271c705d1ccb23652c4c39af5 (patch) | |
tree | cf86e62b114a55217c1be31fa30232bdb9b77ad9 | |
parent | 28fd3cdffb1b24a5a3ba0dc6c6d6b22de981caa0 (diff) | |
download | couchdb-cde714908311377271c705d1ccb23652c4c39af5.tar.gz |
Speedup eunit: couch_mrview_compactor
-rw-r--r-- | src/couch_mrview/src/couch_mrview_compactor.erl | 41 |
1 files changed, 22 insertions, 19 deletions
diff --git a/src/couch_mrview/src/couch_mrview_compactor.erl b/src/couch_mrview/src/couch_mrview_compactor.erl index 17d67f192..d42edc054 100644 --- a/src/couch_mrview/src/couch_mrview_compactor.erl +++ b/src/couch_mrview/src/couch_mrview_compactor.erl @@ -248,11 +248,23 @@ remove_compacted(#mrst{sig = Sig, db_name = DbName} = State) -> -ifdef(TEST). -include_lib("eunit/include/eunit.hrl"). +setup_all() -> + meck:new(couch_index_updater), + meck:new(couch_log). + +teardown_all(_) -> + meck:unload(). + recompact_test_() -> - [ - recompact_success_after_progress(), - recompact_exceeded_retry_count() - ]. + { + setup, + fun setup_all/0, + fun teardown_all/1, + [ + recompact_success_after_progress(), + recompact_exceeded_retry_count() + ] + }. recompact_success_after_progress() -> ?_test(begin @@ -262,12 +274,8 @@ recompact_success_after_progress() -> timer:sleep(100), exit({updated, self(), State#mrst{update_seq = 2}}) end), - try - State = #mrst{fd=self(), update_seq=0}, - ?assertEqual({ok, State#mrst{update_seq = 2}}, recompact(State)) - after - meck:unload(couch_index_updater) - end + State = #mrst{fd=self(), update_seq=0}, + ?assertEqual({ok, State#mrst{update_seq = 2}}, recompact(State)) end). recompact_exceeded_retry_count() -> @@ -277,15 +285,10 @@ recompact_exceeded_retry_count() -> exit(error) end), ok = meck:expect(couch_log, warning, fun(_, _) -> ok end), - try - State = #mrst{fd=self(), db_name=foo, idx_name=bar}, - ExpectedError = {exceeded_recompact_retry_count, - [{db_name, foo}, {idx_name, bar}]}, - ?assertError(ExpectedError, recompact(State)) - after - meck:unload(couch_log), - meck:unload(couch_index_updater) - end + State = #mrst{fd=self(), db_name=foo, idx_name=bar}, + ExpectedError = {exceeded_recompact_retry_count, + [{db_name, foo}, {idx_name, bar}]}, + ?assertError(ExpectedError, recompact(State)) end). -endif. |