40 lines
1.2 KiB
Python
40 lines
1.2 KiB
Python
import unittest
|
|
from main import *
|
|
from unittest.mock import Mock
|
|
from pathlib import Path
|
|
|
|
# test saving file
|
|
class savingFileTest(unittest.TestCase):
|
|
|
|
def test_text_file(self):
|
|
print('Testing save_input method \n')
|
|
old_file_content = 'testing\n\n\nmore testing'
|
|
with open('test.txt', 'w') as file:
|
|
file.write(old_file_content)
|
|
|
|
script_path = os.path.dirname(os.path.realpath(__file__))
|
|
test_file_path = script_path + '\\test.txt'
|
|
newpath = save_input(test_file_path)
|
|
|
|
new_file_content = ''
|
|
with open(newpath, 'r') as new_file:
|
|
for line in new_file:
|
|
new_file_content += line
|
|
|
|
self.assertEqual(old_file_content, new_file_content)
|
|
|
|
|
|
# test using mock checking if a file with analysis had been generated
|
|
class generetingOutputFile(unittest.TestCase):
|
|
|
|
def test_output_file(self):
|
|
|
|
print('Moock testing outputfile')
|
|
m = Mock()
|
|
m.output_file_path = scriptPath + '\\files\\output\\test.pdf'
|
|
|
|
assert Path(m.output_file_path).is_file()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main() |