summaryrefslogtreecommitdiff
path: root/tests/message/conftest.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/message/conftest.py')
-rw-r--r--tests/message/conftest.py16
1 files changed, 9 insertions, 7 deletions
diff --git a/tests/message/conftest.py b/tests/message/conftest.py
index 942119076..e30dbb535 100644
--- a/tests/message/conftest.py
+++ b/tests/message/conftest.py
@@ -3,10 +3,12 @@
# pylint: disable=redefined-outer-name
+from typing import Dict, ValuesView
+
import pytest
from pylint.checkers import BaseChecker
-from pylint.message import MessageDefinitionStore, MessageIdStore
+from pylint.message import MessageDefinition, MessageDefinitionStore, MessageIdStore
@pytest.fixture
@@ -20,12 +22,12 @@ def symbol():
@pytest.fixture
-def empty_store():
+def empty_store() -> MessageDefinitionStore:
return MessageDefinitionStore()
@pytest.fixture
-def store():
+def store() -> MessageDefinitionStore:
store_ = MessageDefinitionStore()
class Checker(BaseChecker):
@@ -50,12 +52,12 @@ def store():
@pytest.fixture
-def message_definitions(store):
+def message_definitions(store: MessageDefinitionStore) -> ValuesView[MessageDefinition]:
return store.messages
@pytest.fixture
-def msgids():
+def msgids() -> Dict[str, str]:
return {
"W1234": "warning-symbol",
"W1235": "warning-symbol-two",
@@ -65,12 +67,12 @@ def msgids():
@pytest.fixture
-def empty_msgid_store():
+def empty_msgid_store() -> MessageIdStore:
return MessageIdStore()
@pytest.fixture
-def msgid_store(msgids):
+def msgid_store(msgids: Dict[str, str]) -> MessageIdStore:
msgid_store = MessageIdStore()
for msgid, symbol in msgids.items():
msgid_store.add_msgid_and_symbol(msgid, symbol)