def test_issue16_missing_toplevel_enum(self, nested_enum_mod): node = extract_node(""" import {} as sut sut.UNO """.format(nested_enum_mod)) msg = make_message(node, nested_enum_mod, 'UNO') self.assert_adds_messages(node, msg)
def test_talk_to_me(updater, effective_user): message = make_message('Проверка бота!', effective_user, updater.bot) call_handler(updater, talk_to_me, message) assert message.reply_text.called args, kwargs = message.reply_text.call_args assert args[0] == 'Проверка бота! :panda_face:' assert 'reply_markup' in kwargs
def test_star_import_enum_should_warn(self, enum_mod): node = extract_node(""" from {} import * Variable.should_warn #@ """.format(enum_mod)) msg = make_message(node, enum_mod + '.Variable', 'should_warn') self.assert_adds_messages(node, msg)
def test_issue16_nested_enum_definition_warns(self, nested_enum_mod): node = extract_node(""" import {} as sut sut.Message.should_warn """.format(nested_enum_mod)) msg = make_message(node, nested_enum_mod + '.Message', 'should_warn') self.assert_adds_messages(node, msg)
def test_messages_do_not_have_enum_fields(self, nested_enum_mod): node = extract_node(""" from {} import Message Message.Value("ONE") """.format(nested_enum_mod)) msg = make_message(node.func, 'Message', 'Value') self.assert_adds_messages(node, msg)
def test_enums_do_not_have_message_fields(self, enum_mod): node = extract_node(""" from {} import Variable Variable.ParseFromString("blah") """.format(enum_mod)) msg = make_message(node.func, 'Variable', 'ParseFromString') self.assert_adds_messages(node, msg)
def test_issue21_missing_field_on_nested_enum(self, nested_enum_mod): node = extract_node(""" import {} as sut sut.Message.Inner.NOPE = 123 """.format(nested_enum_mod)) msg = make_message(node.targets[0], 'Inner', 'NOPE') self.assert_adds_messages(node, msg)
def test_issue16_package_missing_toplevel_enum(self, package_nested_enum_mod): node = extract_node(""" import {mod} {mod}.UNO """.format(mod=nested_enum_mod)) msg = make_message(node, nested_enum_mod, 'UNO') self.assert_adds_messages(node, msg)
def test_issue10_imported_message_warns(self, parent_mod): node = extract_node(""" from {} import Parent p = Parent() p.child.should_warn = 123 #@ """.format(parent_mod)) msg = make_message(node.targets[0], 'Child', 'should_warn') self.assert_adds_messages(node, msg)
def test_issue16_package_nested_enum_definition_warns( self, package_nested_enum_mod): node = extract_node(""" import {mod} {mod}.Message.should_warn """.format(mod=package_nested_enum_mod)) msg = make_message(node, 'Message', 'should_warn') self.assert_adds_messages(node, msg)
def test_reexported_protobuf_message_definition_warns(self, outer_mod): node = extract_node(""" import outer t = outer.Test() t.should_warn = 123 #@ """) msg = make_message(node, 'outer.Test', 'should_warn') self.assert_adds_messages(node, msg)
def test_import_enum_missing_attributes_warns(self, enum_mod): node = extract_node(""" from {} import Variable print( Variable.should_warn #@ ) """.format(enum_mod)) msg = make_message(node, enum_mod + '.Variable', 'should_warn') self.assert_adds_messages(node, msg)
def test_missing_field(self, foo_mod): node = astroid.extract_node(""" import {} as mod foo = mod.Foo() foo.should_warn """.format(foo_mod)) message = make_message(node, 'Foo', 'should_warn') with self.assertAddsMessages(message): self.walk(node.root())
def test_issue18_renamed_from_import_no_assertion_error(self, parent_mod): node = extract_node(""" import {0} import {0} as foo p = {0}.Parent() p.should_warn = 123 """.format(parent_mod)) msg = make_message(node.targets[0], 'Parent', 'should_warn') self.assert_adds_messages(node, msg)
def test_wkt_should_still_warn(self, module, wkt, _, error_on_missing_modules): node = astroid.extract_node(""" from google.protobuf.{module} import {wkt} t = {wkt}() t.should_warn = 123 """.format(module=module, wkt=wkt)) msg = make_message(node.targets[0], wkt, 'should_warn') with self.assertAddsMessages(msg): self.walk(node.root())