def test_get_recommend(cls):
var = Variable('RHOST')
var.set({'func': 'recommend', 'value': '192.168.1.1'})
var.set({'func': 'recommend', 'value': '192.168.1.2'})
# TODO 还要增加 recommend_cmd
recommend = var.get_recommend()
assert len(recommend) == 2
assert recommend == ['192.168.1.1', '192.168.1.2']
def test_select(cls):
var = Variable('RHOST')
var.select = ' 10.10.10.1'
assert var.select == '10.10.10.1'
var = Variable('[-r] RHOST')
var.select = '10.10.10.1'
assert var.select == '-r 10.10.10.1'
var = Variable('[-r] RHOST')
var.select = ''
assert var.if_has == '-r %s'
assert var.select == ''
def create_list():
varlist = VariableList()
varlist.append(Variable('RHOST'))
varlist.append('LHOST')
return varlist
def test_base(cls):
"""测试基本功能"""
var = Variable('RHOST')
assert var.name == 'RHOST'
def test_merge(cls):
var = Variable('RHOST')
var.set({'func': 'desc', 'value': 'remote host'})
var.set({'func': 'recommend', 'value': '192.168.1.1'})
var.set({'func': 'recommend', 'value': '192.168.1.2'})
other_var = Variable('RHOST')
other_var.set({'func': 'desc', 'value': 'rhost'})
other_var.set({'func': 'recommend', 'value': '10.10.10.1'})
var.merge(other_var)
assert var.desc == 'remote host'
assert len(var.recommend) == 3
assert var.recommend[2] == '10.10.10.1'
def test_parse(cls):
var = Variable('[-s] RHOST')
assert var.name == 'RHOST'
assert var.if_has == '-s %s'
def test_set(cls):
var = Variable('RHOST')
var.set({'func': 'desc', 'value': 'remote host'})
assert var.desc == 'remote host'
var.set({'func': 'recommend', 'value': '192.168.1.1'})
assert var.recommend[0] == '192.168.1.1'
var.set({'func': 'recommend', 'value': '192.168.1.2'})
assert var.recommend[1] == '192.168.1.2'
var.set({'func': 'recommend_cmd', 'value': '-r'})
assert var.recommend_cmd[0] == '-r'
var.set({'func': 'recommend_cmd', 'value': '-d'})
assert var.recommend_cmd[1] == '-d'
var.set({'func': 'if_has', 'value': '-r %s'})
assert var.if_has == '-r %s'
var.set({'func': 'refresh'})
assert len(var.recommend) == 0
assert len(var.recommend_cmd) == 0