def test_return_true_when_use_third_package(self):
handler = PackageHandler()
package = 'com.yyx.console'
code = "console.Find()"
result = handler.is_assigned(package, code)
self.assertTrue(result)
def test_return_true_when_use_functional_method(self):
handler = PackageHandler()
package = 'com.yyx.console'
code = "console.Find().get()"
result = handler.is_assigned(package, code)
self.assertTrue(result)
def test_return_true_when_simple_use_builtin_package(self):
handler = PackageHandler()
package = 'fmt'
code = 'fmt.Println("abc")'
result = handler.is_assigned(package, code)
self.assertTrue(result)
def test_scan_used_package_when_block_is_condition_code(self):
handler = PackageHandler()
handler.add_declared("com.yyx.console", "com.yyx.console")
block = Block("if a == 1 {")
block.append(Block("console.Find().get()"))
block.append("}")
handler.scan_used([block])
self.assertEqual(len(list(handler.get_params())), 1)
def test_should_clear_old_package_when_clear_and_scan_used_package(self):
handler = PackageHandler()
handler.add_declared("com.yyx.console", "com.yyx.console")
handler.assignment_manager.add_assigned(
"com.yyx.text", handler.handler_type
)
block = Block("if a == 1 {")
block.append(Block("console.Find().get()"))
block.append("}")
handler.assignment_manager.clear()
handler.scan_used([block])
self.assertEqual(handler.assignment_manager.length(), 1)
self.assertTrue(
"com.yyx.text" not in handler.get_params()
)
