def testBuildTrainDbIncremental(self):
originalBranch = config.getBranch()
dbPath = config.getCfDbFile()
db = CFDatabase(dbPath)
db.clean()
self.assertEqual(len(db.getAllFixData()), 0)
del db
config.setBranch('trainDbScriptIncrementalTest')
buildTestDB.main(True)
db = CFDatabase(dbPath)
self.assertEqual(len(db.getAllFixData()), 1)
self.assertEqual(db.getAllFixData()[0][3], 'deadcode.DeadStores')
self.assertEqual(db.getLastCommit(),
'add691cf37da6c9d40666eac1bc8c1afda071c77')
del db
config.setBranch(originalBranch)
buildTestDB.main(False)
db = CFDatabase(dbPath)
self.assertEqual(len(db.getAllFixData()), 2)
self.assertEqual(db.getAllFixData()[0][3], 'deadcode.DeadStores')
self.assertEqual(db.getAllFixData()[1][3], 'core.DivideZero')
self.assertEqual(db.getLastCommit(),
'f2917b938f0ecbc62ad48101d034369a1ae61a19')
def testBuildTrainDbModelData(self):
dbPath = config.getCfDbFile()
db = CFDatabase(dbPath)
db.clean()
self.assertEqual(len(db.getAllFixData()), 0)
del db
buildTestDB.main(True)
db = CFDatabase(dbPath)
self.assertEqual(len(db.getAllFixData()), 2)
self.assertEqual(db.getAllFixData()[0][3], 'deadcode.DeadStores')
self.assertEqual(db.getAllFixData()[1][3], 'core.DivideZero')
self.assertEqual(db.getLastCommit(),
'f2917b938f0ecbc62ad48101d034369a1ae61a19')
modelBuilder = ModelDataBuilder()
bugVector = modelBuilder.getEncodedBugData('core.DivideZero')
dictionary = modelBuilder.getDictionary()
fixVector = modelBuilder.getEncodedFixData('core.DivideZero',
dictionary)
self.assertSequenceEqual(bugVector, [[
1, 2, 3, 4, 5, 6, 1, 7, 3, 8, 9, 10, 11, 2, 12, 13, 1, 4, 14, 15,
11, 15, 16, 4, 17, 18, 3, 4
]])
self.assertSequenceEqual(fixVector, [[
1, 2, 3, 4, 5, 6, 1, 19, 3, 8, 9, 10, 11, 2, 12, 13, 1, 4, 14, 15,
11, 15, 16, 4, 17, 18, 3, 4
]])
def testBuildTrainDbClean(self):
dbPath = config.getCfDbFile()
db = CFDatabase(dbPath)
db.clean()
self.assertEqual(len(db.getAllFixData()), 0)
del db
buildTestDB.main(True)
db = CFDatabase(dbPath)
self.assertEqual(len(db.getAllFixData()), 2)
self.assertEqual(db.getAllFixData()[0][3], 'deadcode.DeadStores')
self.assertEqual(db.getAllFixData()[1][3], 'core.DivideZero')
self.assertEqual(db.getLastCommit(),
'f2917b938f0ecbc62ad48101d034369a1ae61a19')
def prepareDb(self, clean=False):
self.db = CFDatabase(config.getCfDbFile())
if clean:
self.db.clean()
def __init__(self):
self.db = CFDatabase(config.getCfDbFile())
self.lexer = CxxLexer()
self.checkers = Checkers()
def setUpClass(self):
print("Starting up...")
self.db = CFDatabase(config.getCfDbFile())
self.checkers = Checkers()
self.checkerList = ['deadcode.DeadStores']
self.checkerIndex = 0