add unit tests

Author: Cody D'Ambrosio

tests/test_line.py

@@ -38,5 +38,14 @@ def test_str_directive(self):
         self.assertEqual("ok 44 passing # SKIP a reason", str(result))
 
     def test_str_diagnostics(self):
-        result = Result(False, 43, "failing", diagnostics="# more info")
-        self.assertEqual("not ok 43 failing\n# more info", str(result))
+        result = Result(False, 45, "failing", diagnostics="# more info")
+        self.assertEqual("not ok 45 failing\n# more info", str(result))
+
+    def test_yaml_block(self):
+        raw_yaml_block = """\
+message: test_message
+severity: fail
+"""
+        result = Result(False, 46, "passing", None, None,
+                        raw_yaml_block=raw_yaml_block)
+        self.assertEqual(result.yaml_block["message"], "test_message")

tests/test_tracker.py

@@ -295,6 +295,24 @@ def test_adds_not_ok_with_diagnostics(self):
         line = tracker._test_cases["FakeTestCase"][0]
         self.assertEqual("# more info\n", line.diagnostics)
 
+    def test_adds_ok_with_yaml_block(self):
+        tracker = Tracker()
+        tracker.add_ok("FakeTestCase", "a description", raw_yaml_block="""\
+message: test_message
+severity: pass
+""")
+        line = tracker._test_cases["FakeTestCase"][0]
+        self.assertEqual("test_message", line.yaml_block["message"])
+
+    def test_adds_not_ok_with_yaml_block(self):
+        tracker = Tracker()
+        tracker.add_not_ok("FakeTestCase", "a description", raw_yaml_block="""\
+message: test_message
+severity: fail
+""")
+        line = tracker._test_cases["FakeTestCase"][0]
+        self.assertEqual("test_message", line.yaml_block["message"])
+
     def test_header_displayed_by_default(self):
         tracker = Tracker()
         self.assertTrue(tracker.header)