Add tests for Race default total votes and add_result method

Author: zstumgoren

elex4/tests/test_models.py

@@ -47,6 +47,16 @@ def test_county_results_access(self):
 
 class TestRace(TestCase):
 
+    def setUp(self):
+        self.smith_result = {
+            'date': '2012-11-06',
+            'candidate': 'Smith, Joe',
+            'party': 'Dem',
+            'office': 'President',
+            'county': 'Fairfax',
+            'votes': 2000,
+        }
+
     def test_clean_office_rep(self):
         race = Race("2012-11-06", "U.S. Rep - 1")
         self.assertEquals(race.office, "U.S. House of Representatives")
@@ -56,3 +66,14 @@ def test_clean_office_other(self):
         race = Race("2012-11-06", "President")
         self.assertEquals(race.office, "President")
         self.assertEquals(race.district, "")
+
+    def test_total_votes_default(self):
+        "Race total votes should default to zero"
+        race = Race("2012-11-06", "President")
+        self.assertEquals(race.total_votes, 0)
+
+    def test_total_votes_update(self):
+        "Race.add_result should update total votes for each result"
+        race = Race("2012-11-06", "President")
+        race.add_result(self.smith_result)
+        self.assertEquals(race.total_votes, 2000)