X-Git-Url: https://projects.mako.cc/source/selectricity-live/blobdiff_plain/8ddabb6f9904361d01f65dfafb043c3911d720b0..14836effd31a54f1ff693f423c062691b4f64768:/test/unit/selectricityservice_test.rb

diff --git a/test/unit/selectricityservice_test.rb b/test/unit/selectricityservice_test.rb
index 877f7e9..4fd616f 100644
--- a/test/unit/selectricityservice_test.rb
+++ b/test/unit/selectricityservice_test.rb
@@ -21,20 +21,15 @@ class SelectricityServiceTest < Test::Unit::TestCase
     test_create_quickvote
     election = invoke_delegated :vote, :get_quickvote, "TestVote"
     casted_vote = election.candidate_ids.sort_by {rand} #Shuffles
-    invoke_delegated :vote, :cast_quickvote, "TestVote", 42, [casted_vote]
-    quickvote_votes= invoke_delegated :vote, :get_quickvote_votes, "TestVote"
-    assert_equal quickvote_votes.length, 1
-    assert_equal quickvote_votes[0].vote, casted_vote
+    assert_cast_quickvote_succeeds "TestVote", 42, [casted_vote]
   end
   def test_cast_mass_quickvote
     test_create_quickvote
     election = invoke_delegated :vote, :get_quickvote, "TestVote"
     20.times do |t|
       casted_vote = election.candidate_ids.sort_by {rand}
-      invoke_delegated :vote, :cast_quickvote, "TestVote", t, [casted_vote]
+      assert_cast_quickvote_succeeds "TestVote", t, [casted_vote]
     end
-    quickvote_votes= invoke_delegated :vote, :get_quickvote_votes, "TestVote"
-    assert_equal quickvote_votes.length, 20
   end
   def test_create_mass_quickvote
     10.times do |t|
@@ -100,11 +95,58 @@ class SelectricityServiceTest < Test::Unit::TestCase
     election =  ElectionStruct.new :name => "foobar", :description => "test \x01\x02\x03\x04\x05\x06\x07\x08", :candidate_names => ["Apple", "Orange", "Banana", "Pineapple"]
     assert_create_quickvote_succeeds election
   end
+  def test_quickvote_proper_results
+    election =  ElectionStruct.new :name => "favdev", :description => "Who is your favorite developer?", :candidate_names => ["mako", "jdong", "justin"]
+    assert_create_quickvote_succeeds election
+    reflection = invoke_delegated :vote, :get_quickvote, "favdev"
+    candidates = {}
+    reflection.candidate_names.each_with_index do |name, index|
+      candidates[name] =  reflection.candidate_ids[index]
+    end
+    25.times do |t|
+      vote = [candidates["jdong"], candidates["mako"], candidates["justin"]]
+      assert_cast_quickvote_succeeds "favdev", "1:#{t}", [vote]
+    end
+    5.times do |t|
+      vote = [candidates["mako"], candidates["justin"], candidates["jdong"]]
+      assert_cast_quickvote_succeeds "favdev", "2:#{t}", [vote]
+    end
+    10.times do |t|
+      vote = [candidates["justin"], candidates["mako"], candidates["jdong"]]
+      assert_cast_quickvote_succeeds "favdev", "3:#{t}", [vote]
+    end
+    results=invoke_delegated(:vote, :get_quickvote_results, "favdev")
+    assert_equal results.approval_winners, [candidates["mako"]]
+    assert_equal results.borda_winners, [candidates["jdong"]]
+    assert_equal results.plurality_winners, [candidates["jdong"]]
+    assert_equal results.condorcet_winners, [candidates["jdong"]]
+    assert_equal results.ssd_winners, [candidates["jdong"]]
+    assert_equal results.errors.length, 0
+  end
   private
+  def assert_cast_quickvote_succeeds(shortname, id, vote)
+    old_votes = invoke_delegated :vote, :get_quickvote_votes, shortname
+    assert_nothing_raised do
+      invoke_delegated :vote, :cast_quickvote, shortname, id, vote
+    end
+    new_votes = invoke_delegated :vote, :get_quickvote_votes, shortname
+    assert_equal old_votes.length, new_votes.length-1
+    assert_not_nil(reflection = new_votes.find_all { |v| v.voter_session_id == "XMLRPC:#{id}" })
+    assert_equal reflection.length, 1
+    assert_equal reflection[0].vote, vote[0]
+  end
+  def assert_cast_quickvote_fails(shortname, id, vote)
+    assert_raises Test::Unit::AssertionFailedError do 
+      assert_cast_quickvote_succeeds(shortname, id, vote)
+    end
+  end
   def assert_create_quickvote_succeeds(election)
     # Checks if a created quickvote is identical when retrieved
     old_len=invoke_delegated(:vote,:list_quickvotes).length
-    result = invoke_delegated :vote, :create_quickvote, election
+    result=""
+    assert_nothing_raised do
+      result = invoke_delegated :vote, :create_quickvote, election
+    end
     assert_equal result, ""
     reflection = invoke_delegated :vote, :get_quickvote, election.name
     assert_equal election.description, reflection.description
@@ -113,11 +155,8 @@ class SelectricityServiceTest < Test::Unit::TestCase
     assert_equal(invoke_delegated(:vote,:list_quickvotes).length, old_len+1)
   end
   def assert_create_quickvote_fails(election)
-    # Helper function to check that creating this quickvote fails
-    old_len=invoke_delegated(:vote,:list_quickvotes).length
-    result = invoke_delegated :vote, :create_quickvote, election
-    assert_instance_of String, result
-    assert_not_equal result.length, 0
-    assert_equal(invoke_delegated(:vote,:list_quickvotes).length, old_len)
+    assert_raises Test::Unit::AssertionFailedError do
+      assert_create_quickvote_succeeds election
+    end
   end
 end