mirror of https://github.com/crytic/slither
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
72 lines
4.3 KiB
72 lines
4.3 KiB
import unittest
|
|
import subprocess, os, sys
|
|
|
|
class TestExternalFunctions(unittest.TestCase):
|
|
testDataFile1 = "external_function.sol"
|
|
testDataFile2 = "external_function_2.sol"
|
|
testDataDir = "./slither_format/tests/test_data/"
|
|
testFilePath1 = testDataDir+testDataFile1
|
|
testFilePath2 = testDataDir+testDataFile2
|
|
|
|
def setUp(self):
|
|
outFD1 = open(self.testFilePath1+".out","w")
|
|
errFD1 = open(self.testFilePath1+".err","w")
|
|
p1 = subprocess.Popen(['python3', '-m', 'slither_format','--verbose-test','--detect','external-function',self.testFilePath1], stdout=outFD1,stderr=errFD1)
|
|
p1.wait()
|
|
outFD2 = open(self.testFilePath2+".out","w")
|
|
errFD2 = open(self.testFilePath2+".err","w")
|
|
p2 = subprocess.Popen(['python3', '-m', 'slither_format','--verbose-test','--detect','external-function',self.testFilePath2], stdout=outFD2,stderr=errFD2)
|
|
p2.wait()
|
|
outFD1.close()
|
|
errFD1.close()
|
|
outFD2.close()
|
|
errFD2.close()
|
|
|
|
def tearDown(self):
|
|
p1 = subprocess.Popen(['rm','-f',self.testFilePath1+'.out',self.testFilePath1+'.err',self.testFilePath1+'.format'])
|
|
p1.wait()
|
|
p2 = subprocess.Popen(['rm','-f',self.testFilePath2+'.out',self.testFilePath2+'.err',self.testFilePath2+'.format'])
|
|
p2.wait()
|
|
|
|
def test_external_function(self):
|
|
errFD1 = open(self.testFilePath1+".err","r")
|
|
errFD1_lines = errFD1.readlines()
|
|
for i in range(len(errFD1_lines)):
|
|
errFD1_lines[i] = errFD1_lines[i].strip()
|
|
self.assertTrue(os.path.isfile(self.testFilePath1+".format"),"Patched .format file is not created?!")
|
|
self.assertEqual(errFD1_lines[0],"INFO:Slither.Format:Number of Slither results: 9")
|
|
self.assertEqual(errFD1_lines[1],"INFO:Slither.Format:Number of patches: 8")
|
|
self.assertEqual(errFD1_lines.count("INFO:Slither.Format:Detector: external-function"), 8)
|
|
self.assertEqual(errFD1_lines.count("INFO:Slither.Format:Old string: public"), 6)
|
|
self.assertEqual(errFD1_lines.count("INFO:Slither.Format:New string: external"), 6)
|
|
self.assertEqual(errFD1_lines.count("INFO:Slither.Format:Location start: 384"), 1)
|
|
self.assertEqual(errFD1_lines.count("INFO:Slither.Format:Location end: 390"), 1)
|
|
self.assertEqual(errFD1_lines.count("INFO:Slither.Format:Location start: 562"), 1)
|
|
self.assertEqual(errFD1_lines.count("INFO:Slither.Format:Location end: 568"), 1)
|
|
self.assertEqual(errFD1_lines.count("INFO:Slither.Format:Location start: 642"), 1)
|
|
self.assertEqual(errFD1_lines.count("INFO:Slither.Format:Location end: 648"), 1)
|
|
self.assertEqual(errFD1_lines.count("INFO:Slither.Format:Location start: 685"), 1)
|
|
self.assertEqual(errFD1_lines.count("INFO:Slither.Format:Location end: 691"), 1)
|
|
self.assertEqual(errFD1_lines.count("INFO:Slither.Format:Location start: 1022"), 1)
|
|
self.assertEqual(errFD1_lines.count("INFO:Slither.Format:Location end: 1028"), 1)
|
|
self.assertEqual(errFD1_lines.count("INFO:Slither.Format:Location start: 1305"), 1)
|
|
self.assertEqual(errFD1_lines.count("INFO:Slither.Format:Location end: 1311"), 1)
|
|
self.assertEqual(errFD1_lines.count("INFO:Slither.Format:Old string:"), 2)
|
|
self.assertEqual(errFD1_lines.count("INFO:Slither.Format:New string: external"), 2)
|
|
self.assertEqual(errFD1_lines.count("INFO:Slither.Format:Location start: 524"), 1)
|
|
self.assertEqual(errFD1_lines.count("INFO:Slither.Format:Location end: 524"), 1)
|
|
self.assertEqual(errFD1_lines.count("INFO:Slither.Format:Location start: 1142"), 1)
|
|
self.assertEqual(errFD1_lines.count("INFO:Slither.Format:Location end: 1142"), 1)
|
|
errFD1.close()
|
|
|
|
errFD2 = open(self.testFilePath2+".err","r")
|
|
errFD2_lines = errFD2.readlines()
|
|
for i in range(len(errFD2_lines)):
|
|
errFD2_lines[i] = errFD2_lines[i].strip()
|
|
self.assertFalse(os.path.isfile(self.testFilePath2+".format"),"Patched .format file _is_ created?!")
|
|
self.assertEqual(errFD2_lines[0],"INFO:Slither.Format:Number of Slither results: 0")
|
|
self.assertEqual(errFD2_lines[1],"INFO:Slither.Format:Number of patches: 0")
|
|
errFD2.close()
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|
|
|