Bug fix (proper extending of tests when dbms is known)

This commit is contained in:
Miroslav Stampar
2014-08-30 21:34:23 +02:00
parent e501b2a80b
commit 1a9a331422
3 changed files with 9 additions and 14 deletions

View File

@@ -106,6 +106,10 @@ def checkSqlInjection(place, parameter, value):
msg = "do you want to include all tests for '%s' " % _
msg += "extending provided level (%d) and risk (%s)? [Y/n]" % (conf.level, conf.risk)
kb.extendTests = [] if readInput(msg, default='Y').upper() != 'Y' else (Backend.getErrorParsedDBMSes() or [kb.heuristicDbms])
elif kb.extendTests is None:
msg = "do you want to include all tests for '%s' " % conf.dbms
msg += "extending provided level (%d) and risk (%s)? [Y/n]" % (conf.level, conf.risk)
kb.extendTests = [] if readInput(msg, default='Y').upper() != 'Y' else ([conf.dbms])
title = test.title
stype = test.stype