Adding tests for couch_task_status.erl

Thanks Bob Dionne



git-svn-id: https://svn.apache.org/repos/asf/couchdb/trunk@788246 13f79535-47bb-0310-9956-ffa450edef68

1 files changed, 197 insertions, 0 deletions

diff --git a/test/etap/090-task-status.t b/test/etap/090-task-status.t
new file mode 100755
index 00000000..d2df4f91
--- /dev/null
+++ b/test/etap/090-task-status.t
@@ -0,0 +1,197 @@
+#!/usr/bin/env escript
+%% -*- erlang -*-
+
+main(_) ->
+    code:add_pathz("src/couchdb"),
+    etap:plan(16),
+    case (catch test()) of
+        ok ->
+            etap:end_tests();
+        Other ->
+            etap:diag(io_lib:format("Test died abnormally: ~p", [Other])),
+            etap:bail(Other)
+    end,
+    ok.
+
+check_status(Pid,ListPropLists) ->
+    From = list_to_binary(pid_to_list(Pid)),
+    Element = lists:foldl(
+        fun(PropList,Acc) ->
+	        case proplists:get_value(pid,PropList) of
+			    From ->
+                    [PropList | Acc];
+				_ ->
+                    []
+			end
+		end,
+        [], ListPropLists
+    ),
+    proplists:get_value(status,hd(Element)).
+
+loop() ->
+    receive
+	{add, From} ->
+	    Resp = couch_task_status:add_task("type", "task", "init"),
+	    From ! {ok, self(), Resp},
+	    loop();
+	{update, Status, From} ->
+	    Resp = couch_task_status:update(Status),
+	    From ! {ok, self(), Resp},
+	    loop();
+	{update_frequency, Msecs, From} ->
+	    Resp = couch_task_status:set_update_frequency(Msecs),
+	    From ! {ok, self(), Resp},
+	    loop();
+	{done, From} ->
+	    From ! {ok, self(), ok}
+    end.
+
+call(Pid, Command) ->
+    Pid ! {Command, self()},
+    wait(Pid).
+
+call(Pid, Command, Arg) ->
+    Pid ! {Command, Arg, self()},
+    wait(Pid).
+
+wait(Pid) ->
+    receive
+        {ok, Pid, Msg} -> Msg
+    after 1000 ->
+        throw(timeout_error)
+    end.    
+
+test() ->   
+    {ok, TaskStatusPid} = couch_task_status:start_link(),
+    
+    TaskUpdater = fun() -> loop() end,
+    % create three updaters
+    Pid1 = spawn(TaskUpdater),
+    Pid2 = spawn(TaskUpdater),
+    Pid3 = spawn(TaskUpdater),
+
+    ok = call(Pid1, add),
+    etap:is(
+        length(couch_task_status:all()),
+        1,
+        "Started a task"
+    ),
+
+    etap:is(
+        call(Pid1, add),
+        {add_task_error, already_registered},
+        "Unable to register multiple tasks for a single Pid."
+    ),
+
+    etap:is(
+        check_status(Pid1, couch_task_status:all()),
+        <<"init">>,
+		"Task status was set to 'init'."
+    ),
+
+    call(Pid1,update,"running"),
+    etap:is(
+        check_status(Pid1,couch_task_status:all()),
+        <<"running">>,
+		"Status updated to 'running'."
+	),
+
+
+    call(Pid2,add),
+    etap:is(
+        length(couch_task_status:all()),
+        2,
+        "Started a second task."
+    ),
+    
+    etap:is(
+        check_status(Pid2, couch_task_status:all()),
+        <<"init">>,
+		"Second tasks's status was set to 'init'."
+	),
+	
+    call(Pid2, update, "running"),
+    etap:is(
+        check_status(Pid2, couch_task_status:all()),
+        <<"running">>,
+		"Second task's status updated to 'running'."
+	),
+
+
+    call(Pid3, add),
+    etap:is(
+        length(couch_task_status:all()),
+        3,
+        "Registered a third task."
+    ),
+    
+    etap:is(
+        check_status(Pid3, couch_task_status:all()),
+        <<"init">>,
+		"Third tasks's status was set to 'init'."
+	),
+	
+    call(Pid3, update, "running"),
+    etap:is(
+        check_status(Pid3, couch_task_status:all()),
+        <<"running">>,
+		"Third task's status updated to 'running'."
+	),
+
+
+    call(Pid3, update_frequency, 500),
+    call(Pid3, update, "still running"),
+    etap:is(
+        check_status(Pid3, couch_task_status:all()),
+        <<"still running">>,
+		"Third task's status updated to 'still running'."
+	),
+
+    call(Pid3, update, "skip this update"),
+    etap:is(
+        check_status(Pid3, couch_task_status:all()),
+        <<"still running">>,
+		"Status update dropped because of frequency limit."
+	),
+
+    call(Pid3, update_frequency, 0),
+    call(Pid3, update, "don't skip"),
+    etap:is(
+        check_status(Pid3, couch_task_status:all()),
+        <<"don't skip">>,
+		"Status updated after reseting frequency limit."
+	),
+
+
+    call(Pid1, done),
+    etap:is(
+        length(couch_task_status:all()),
+        2,
+        "First task finished."
+    ),
+    
+    call(Pid2, done),
+    etap:is(
+        length(couch_task_status:all()),
+        1,
+        "Second task finished."
+    ),
+    
+    call(Pid3, done),
+    etap:is(
+        length(couch_task_status:all()),
+        0,
+        "Third task finished."
+    ),
+
+    erlang:monitor(process, TaskStatusPid),
+    couch_task_status:stop(),
+    receive
+        {'DOWN', _, _, TaskStatusPid, _} ->
+            ok
+    after
+        1000 ->
+            throw(timeout_error)
+    end,
+
+    ok.