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','--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','--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):
        outFD1 = open(self.testFilePath1+".out","r")
        outFD1_lines = outFD1.readlines()
        for i in range(len(outFD1_lines)):
            outFD1_lines[i] = outFD1_lines[i].strip()
        self.assertTrue(os.path.isfile(self.testFilePath1+".format"),"Patched .format file is not created?!")
        self.assertEqual(outFD1_lines[0],"Number of Slither results: 9")
        self.assertEqual(outFD1_lines[1],"Number of patches: 8")
        self.assertEqual(outFD1_lines.count("Detector: external-function"), 8)
        self.assertEqual(outFD1_lines.count("Old string:     public returns"), 1)
        self.assertEqual(outFD1_lines.count("New string:     external returns"), 1)
        self.assertEqual(outFD1_lines.count("Location start: 384"), 1)
        self.assertEqual(outFD1_lines.count("Location end: 399"), 1)
        self.assertEqual(outFD1_lines.count("Old string:"), 1)
        self.assertEqual(outFD1_lines.count("New string:  external"), 4)
        self.assertEqual(outFD1_lines.count("Location start: 524"), 1)
        self.assertEqual(outFD1_lines.count("Location end: 525"), 1)
        self.assertEqual(outFD1_lines.count("Old string:  public    returns"), 1)
        self.assertEqual(outFD1_lines.count("New string:  external    returns"), 1)
        self.assertEqual(outFD1_lines.count("Location start: 562"), 1)
        self.assertEqual(outFD1_lines.count("Location end: 581"), 1)
        self.assertEqual(outFD1_lines.count("Old string:  public"), 3)
        self.assertEqual(outFD1_lines.count("Old string:  public mod"), 1)        
        self.assertEqual(outFD1_lines.count("New string:  external"), 4)
        self.assertEqual(outFD1_lines.count("Location start: 642"), 1)
        self.assertEqual(outFD1_lines.count("Location end: 649"), 1)
        self.assertEqual(outFD1_lines.count("Location start: 685"), 1)
        self.assertEqual(outFD1_lines.count("Location end: 692"), 1)
        self.assertEqual(outFD1_lines.count("Location start: 1022"), 1)
        self.assertEqual(outFD1_lines.count("Location end: 1033"), 1)
        self.assertEqual(outFD1_lines.count("Location start: 1142"), 1)
        self.assertEqual(outFD1_lines.count("Location end: 1147"), 1)
        self.assertEqual(outFD1_lines.count("Old string:  mod"), 1)
        self.assertEqual(outFD1_lines.count("New string:  external mod"), 2)
        self.assertEqual(outFD1_lines.count("Location start: 1305"), 1)
        self.assertEqual(outFD1_lines.count("Location end: 1312"), 1)
        outFD1.close()

        outFD2 = open(self.testFilePath2+".out","r")
        outFD2_lines = outFD2.readlines()
        for i in range(len(outFD2_lines)):
            outFD2_lines[i] = outFD2_lines[i].strip()
        self.assertFalse(os.path.isfile(self.testFilePath2+".format"),"Patched .format file _is_ created?!")
        self.assertEqual(outFD2_lines[0],"Number of Slither results: 0")
        self.assertEqual(outFD2_lines[1],"Number of patches: 0")
        outFD2.close()
        
if __name__ == '__main__':
    unittest.main()