Fix an incompatibility with pyarmor

2 files changed, 7 insertions, 1 deletions

diff --git a/coverage/context.py b/coverage/context.py
index 45e86a5c..43d2b1cc 100644
--- a/coverage/context.py
+++ b/coverage/context.py
@@ -48,7 +48,7 @@ def qualname_from_frame(frame):
     fname = co.co_name
     method = None
     if co.co_argcount and co.co_varnames[0] == "self":
-        self = frame.f_locals["self"]
+        self = frame.f_locals.get("self", None)
         method = getattr(self, fname, None)
 
     if method is None:
diff --git a/tests/test_context.py b/tests/test_context.py
index 3f80803b..de972819 100644
--- a/tests/test_context.py
+++ b/tests/test_context.py
@@ -5,6 +5,7 @@
 
 import inspect
 import os.path
+from unittest import mock
 
 import coverage
 from coverage.context import qualname_from_frame
@@ -275,3 +276,8 @@ class QualnameTest(CoverageTest):
         # A class with a name like a function shouldn't confuse qualname_from_frame.
         class test_something:               # pylint: disable=unused-variable
             assert get_qualname() is None
+
+    def test_bug_1210(self):
+        co = mock.Mock(co_name="a_co_name", co_argcount=1, co_varnames=["self"])
+        frame = mock.Mock(f_code = co, f_locals={})
+        assert qualname_from_frame(frame) == "unittest.mock.a_co_name"