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())
