Convert annotate() farm tests to normal tests

1 files changed, 121 insertions, 0 deletions

diff --git a/tests/test_annotate.py b/tests/test_annotate.py
new file mode 100644
index 00000000..634c1c14
--- /dev/null
+++ b/tests/test_annotate.py
@@ -0,0 +1,121 @@
+# coding: utf-8
+# Licensed under the Apache License: http://www.apache.org/licenses/LICENSE-2.0
+# For details: https://github.com/nedbat/coveragepy/blob/master/NOTICE.txt
+
+"""Tests for annotation from coverage.py."""
+
+import coverage
+
+from tests.coveragetest import CoverageTest
+from tests.goldtest import compare, gold_path
+
+
+class AnnotationGoldTest1(CoverageTest):
+
+    def make_multi(self):
+        self.make_file("multi.py", """\
+            # Licensed under the Apache License: http://www.apache.org/licenses/LICENSE-2.0
+            # For details: https://github.com/nedbat/coveragepy/blob/master/NOTICE.txt
+
+            import a.a
+            import b.b
+
+            a.a.a(1)
+            b.b.b(2)
+            """)
+        self.make_file("a/__init__.py")
+        self.make_file("a/a.py", """\
+            # Licensed under the Apache License: http://www.apache.org/licenses/LICENSE-2.0
+            # For details: https://github.com/nedbat/coveragepy/blob/master/NOTICE.txt
+
+            def a(x):
+                if x == 1:
+                    print("x is 1")
+                else:
+                    print("x is not 1")
+            """)
+        self.make_file("b/__init__.py")
+        self.make_file("b/b.py", """\
+            # Licensed under the Apache License: http://www.apache.org/licenses/LICENSE-2.0
+            # For details: https://github.com/nedbat/coveragepy/blob/master/NOTICE.txt
+
+            def b(x):
+                msg = "x is %s" % x
+                print(msg)
+            """)
+
+    def test_multi(self):
+        self.make_multi()
+        cov = coverage.Coverage()
+        self.start_import_stop(cov, "multi")
+        cov.annotate()
+
+        compare(gold_path("annotate/gold_multi"), ".", "*,cover")
+
+    def test_annotate_dir(self):
+        self.make_multi()
+        cov = coverage.Coverage(source=["."])
+        self.start_import_stop(cov, "multi")
+        cov.annotate(directory="out_anno_dir")
+
+        compare(gold_path("annotate/gold_anno_dir"), "out_anno_dir", "*,cover")
+
+    def test_encoding(self):
+        self.make_file("utf8.py", """\
+            # -*- coding: utf-8 -*-
+            # Licensed under the Apache License: http://www.apache.org/licenses/LICENSE-2.0
+            # For details: https://github.com/nedbat/coveragepy/blob/master/NOTICE.txt
+
+            # This comment has an accent: é
+
+            print("spam eggs")
+            """)
+        cov = coverage.Coverage()
+        self.start_import_stop(cov, "utf8")
+        cov.annotate()
+        compare(gold_path("annotate/gold_encodings"), ".", "*,cover")
+
+    def test_white(self):
+        self.make_file("white.py", """\
+            # Licensed under the Apache License: http://www.apache.org/licenses/LICENSE-2.0
+            # For details: https://github.com/nedbat/coveragepy/blob/master/NOTICE.txt
+
+            # A test case sent to me by Steve White
+
+            def f(self):
+                if self==1:
+                    pass
+                elif self.m('fred'):
+                    pass
+                elif (g==1) and (b==2):
+                    pass
+                elif self.m('fred')==True:
+                    pass
+                elif ((g==1) and (b==2))==True:
+                    pass
+                else:
+                    pass
+
+            def g(x):
+                if x == 1:
+                    a = 1
+                else:
+                    a = 2
+
+            g(1)
+
+            def h(x):
+                if 0:   #pragma: no cover
+                    pass
+                if x == 1:
+                    a = 1
+                else:
+                    a = 2
+
+            h(2)
+            """)
+
+        cov = coverage.Coverage()
+        self.start_import_stop(cov, "white")
+        cov.annotate()
+        compare(gold_path("annotate/gold"), ".", "*,cover")