From 115b766dfab89cb9c8b6762968100716043ff366 Mon Sep 17 00:00:00 2001
From: Tryn Mirell <tryn@basho.com>
Date: Mon, 2 Apr 2012 07:38:07 -0500
Subject: [PATCH] Initial eunit test for riaknostic_check_ring using meck

---
 src/riaknostic_check_ring_membership.erl | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/src/riaknostic_check_ring_membership.erl b/src/riaknostic_check_ring_membership.erl
index 7363bf4..8993fb1 100644
--- a/src/riaknostic_check_ring_membership.erl
+++ b/src/riaknostic_check_ring_membership.erl
@@ -31,6 +31,8 @@
          check/0,
          format/1]).
 
+-include_lib("eunit/include/eunit.hrl").
+
 -spec description() -> string().
 description() ->
     "Cluster membership validity".
@@ -51,6 +53,14 @@ check() ->
             [{warning, {not_ring_member, NodeName}}]
     end.
 
+check_test() ->
+    meck:new(riaknostic_node, [passthrough]),
+    meck:expect(riaknostic_node, stats, fun() -> [{ring_members, ["riak@127.0.0.1"]}, {nodename, ["notmember@127.0.0.1"]}] end),
+    ?assert(meck:validate(riaknostic_node)),
+    ?assertEqual([{warning, {not_ring_member, ["notmember@127.0.0.1"]}}], check()),
+    ?assertNotEqual([{warning, {not_ring_member, ["notequal@127.0.0.1"]}}], check()),
+    meck:unload(riaknostic_node).
+
 -spec format(term()) -> {io:format(), [term()]}.
 format({not_ring_member, Nodename}) ->
     {"Local node ~w is not a member of the ring. Please check that the -name setting in vm.args is correct.", [Nodename]}.