From 2a65d80deab25f9c0520ae5788adbdc909205e1c Mon Sep 17 00:00:00 2001
From: Michael Pivato <pivmy004@mymail.unisa.edu.au>
Date: Thu, 7 Feb 2019 07:03:36 +1030
Subject: [PATCH] Add message and ballot voter tests.

---
 tests/test_ballot_voter.py |  8 ++++++++
 tests/test_messages.py     | 33 +++++++++++++++++++++++++++++++++
 2 files changed, 41 insertions(+)
 create mode 100644 tests/test_ballot_voter.py
 create mode 100644 tests/test_messages.py

diff --git a/tests/test_ballot_voter.py b/tests/test_ballot_voter.py
new file mode 100644
index 0000000..c5aec10
--- /dev/null
+++ b/tests/test_ballot_voter.py
@@ -0,0 +1,8 @@
+import unittest
+from DecisionSystem.CentralisedDecision.ballotvoter import BallotVoter
+
+class TestBallotVoter(unittest.TestCase):
+    def setUp(self):
+        pass
+
+    
\ No newline at end of file
diff --git a/tests/test_messages.py b/tests/test_messages.py
new file mode 100644
index 0000000..2961ce6
--- /dev/null
+++ b/tests/test_messages.py
@@ -0,0 +1,33 @@
+import unittest
+from DecisionSystem.messages import Message, deserialise, RequestLeader
+
+class TestMessages(unittest.TestCase):
+    def test_base_empty(self):
+        self.assertEqual(self.m.sender, "")
+        self.assertEqual(self.m.data, {})
+        self.assertIsNone(self.m.type)
+
+    def test_base_set(self):
+        self.m2.type = "Michael"
+        self.assertEqual(self.m2.type, "Michael")
+        self.assertEqual(self.m2.data, self.data)
+        self.assertEqual(self.m2.sender, 33)
+
+    def test_serialistion(self):
+        serialised = self.m2.serialise()
+        self.m3 = deserialise(serialised)
+        self.assertEqual(self.m2.sender, self.m3.sender)
+        self.assertEqual(self.m2.data, self.m3.data)
+
+    def test_RequestLeader(self):
+        self.m3 = RequestLeader()
+        self.assertEqual(self.m3.type, "RequestLeader")
+
+    def setUp(self):
+        self.m = Message()
+        self.data = {"hi": 3}
+        self.sender = 33
+        self.m2 = Message(self.sender,self.data)
+
+if __name__ == '__main__':
+    unittest.main()
\ No newline at end of file