def test_parse_line_from_module_import_item(self):
variable_parser = VariableParser()
line = "from socket import AF_INET"
variable = variable_parser.parse_line(line)
self.assertEqual('AF_INET', variable.symbol)
self.assertTrue(not variable.fields)
self.assertTrue(not variable.methods)
def test_parse_line_import_socket(self):
variable_parser = VariableParser()
line = "import socket"
variable = variable_parser.parse_line(line)
self.assertEqual('socket', variable.symbol)
self.assertTrue('SOL_SOCKET' in variable.fields)
self.assertTrue('setdefaulttimeout' in variable.methods)
def test_parse_file(self):
return
variable_parser = VariableParser()
fd = open("./test/test_input_script.py")
input = fd.read()
fd.close()
variables = variable_parser.parse_input(input)
variable_symbols = []
for variable in variables:
variable_symbols.append(variable.symbol)
self.assertTrue('Threads' in variable_symbols)
self.assertTrue('lock' in variable_symbols)
self.assertTrue('connection' in variable_symbols)
self.assertTrue(not 'run_time' in variable_symbols)
def test_parse_line_import_failed_no_such_package(self):
variable_parser = VariableParser()
line = "import socke"
variable = variable_parser.parse_line(line)
self.assertTrue(not variable)
def test_parse_line_import_failed_bad_input(self):
variable_parser = VariableParser()
line = "import socket 4"
variable = variable_parser.parse_line(line)
self.assertTrue(not variable)
def test_parse_line_assignment(self):
variable_parser = VariableParser()
symbol = 'variable'
line = "variable = 4"
variable = variable_parser.parse_line(line)
self.assertTrue(symbol == variable.symbol)
def test_parse_line_comment_mid_line(self):
variable_parser = VariableParser()
line = "var #="
variable = variable_parser.parse_line(line)
self.assertTrue(not variable)
def test_parse_line_comment(self):
variable_parser = VariableParser()
line = "#var = 4"
variable = variable_parser.parse_line(line)
#print variable.symbol
self.assertTrue(not variable)
def setup(self):
self.parser = VariableParser()
def test_parse_line_class_definition(self):
variable_parser = VariableParser()
line = "class VariableParser(object):"
variable = variable_parser.parse_line(line)
#print variable.symbol
self.assertTrue(not variable)
def test_parse_line_function_defintion(self):
variable_parser = VariableParser()
line = "def parse_line(self, variable):"
variable = variable_parser.parse_line(line)
self.assertTrue(not variable)
def test_parse_line_void_function(self):
variable_parser = VariableParser()
line = "exit(1)"
variable = variable_parser.parse_line(line)
self.assertTrue(not variable)
def test_parse_line_invalid_line(self):
variable_parser = VariableParser()
line = "\t variable1.field + Max(3, 4) "
variable = variable_parser.parse_line(line)
self.assertTrue(not variable)
def test_parse_line_only_whitespace(self):
variable_parser = VariableParser()
line = "\t \t "
variable = variable_parser.parse_line(line)
self.assertTrue(not variable)
def test_parse_line_from_module_import_module(self):
variable_parser = VariableParser()
line = "from socket import os"
variable = variable_parser.parse_line(line)
self.assertEqual('os', variable.symbol)
self.assertTrue('lseek' in variable.methods)