def test_find_all_include_defs_when_they_exist(self):
all_include_defs = build_file.from_content(
'include_defs("//foo/DEFS")').find_all_include_defs()
self.assertEqual(["//foo/DEFS"],
list(
map(lambda include: include.get_location(),
all_include_defs)))
def test_find_all_include_defs_when_they_exist(self):
all_include_defs = build_file.from_content(
'include_defs("//foo/DEFS")').find_all_include_defs()
self.assertEqual(["//foo/DEFS"],
list(
map(lambda include: include.get_location(),
all_include_defs)))
def test_find_only_include_defs_when_they_exist(self):
content = """
include_defs("//foo/DEFS")
foo("bar")
"""
all_include_defs = build_file.from_content(content).find_all_include_defs()
self.assertEqual(
["//foo/DEFS"],
list(map(lambda include: include.get_location(), all_include_defs)),
)
def test_find_only_include_defs_when_they_exist(self):
content = """
include_defs("//foo/DEFS")
foo("bar")
"""
all_include_defs = build_file.from_content(content).find_all_include_defs()
self.assertEqual(
["//foo/DEFS"],
list(map(lambda include: include.get_location(), all_include_defs)),
)
def test_find_function_calls_by_name(self):
content = """
foo('bar')
bar('foo')
foo('baz')
"""
foo_calls = build_file.from_content(
content)._find_all_function_calls_by_name("foo")
self.assertEqual(2, len(foo_calls))
for call in foo_calls:
self.assertEqual("foo", call.func.id)
def test_find_function_calls_by_name(self):
content = """
foo('bar')
bar('foo')
foo('baz')
"""
foo_calls = build_file.from_content(
content)._find_all_function_calls_by_name('foo')
self.assertEqual(2, len(foo_calls))
for call in foo_calls:
self.assertEqual('foo', call.func.id)
def test_find_used_symbols(self):
content = """
foo = BAR
bar('foo')
def func():
pass
"""
self.assertEqual(
['BAR', 'bar'],
list(
map(lambda n: n.id,
build_file.from_content(content)._find_used_symbols())))
def test_find_used_symbols(self):
content = """
foo = BAR
bar('foo')
def func():
pass
"""
self.assertEqual(
['BAR', 'bar'],
list(
map(lambda n: n.id,
build_file.from_content(content)._find_used_symbols())))
def test_find_exported_symbols(self):
content = """
foo = BAR
bar('foo')
def func():
pass
class Clazz:
def foo():
pass
"""
self.assertEqual(
['foo', 'func', 'Clazz'],
build_file.from_content(content).find_exported_symbols())
def test_find_exported_symbols(self):
content = """
foo = BAR
bar('foo')
def func():
pass
class Clazz:
def foo():
pass
"""
self.assertEqual(
['foo', 'func', 'Clazz'],
build_file.from_content(content).find_exported_symbols())
def test_find_all_include_defs_when_no_includes(self):
self.assertEqual([], build_file.from_content("").find_all_include_defs())
def test_find_all_include_defs_when_no_includes(self):
self.assertEqual([], build_file.from_content("").find_all_include_defs())
