From b7fc90dca7bed9f5dffe615a4fdba1dac9ea57d5 Mon Sep 17 00:00:00 2001
From: Ned Batchelder <ned@nedbatchelder.com>
Date: Thu, 28 Oct 2021 06:17:58 -0400
Subject: test: add a annotate test, and simplify the logic

---
 tests/test_annotate.py | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

(limited to 'tests/test_annotate.py')

diff --git a/tests/test_annotate.py b/tests/test_annotate.py
index de6edcd0..f85450c7 100644
--- a/tests/test_annotate.py
+++ b/tests/test_annotate.py
@@ -105,3 +105,23 @@ class AnnotationGoldTest(CoverageTest):
         self.start_import_stop(cov, "white")
         cov.annotate()
         compare(gold_path("annotate/annotate"), ".", "*,cover")
+
+    def test_missing_after_else(self):
+        self.make_file("mae.py", """\
+            def f(x):
+                if x == 1:
+                    print("1")
+                else:
+                    print("2")
+
+            if f(1):
+                print("nope")
+            if f(2):
+                print("nope")
+            """)
+
+        cov = coverage.Coverage()
+        self.start_import_stop(cov, "mae")
+        cov.annotate()
+        assert self.stdout() == "1\n2\n"
+        compare(gold_path("annotate/mae"), ".", "*,cover")
-- 
cgit v1.2.1