main_one_unit_tests.py

from unittest.mock import patch, MagicMock

from lessons.unit_testing.main_one import FileAnalysis
import unittest


class FileAnalysisUnitTest(unittest.TestCase):
    @patch("lessons.unit_testing.main_one.pefile")
    def test_init_ok(self, pefile_mock: MagicMock):
        # with patch("lessons.unit_testing.main_one.pefile"):  # now it's a decorator
        file_test = FileAnalysis(path_to_file="test_path")

        pefile_mock.PE.assert_called_with('test_path')
        self.assertEqual(file_test.magic_number, 0x0)
        # assert file_test.magic_number == 0x0
        self.assertEqual(file_test.signature, 0x0)
        self.assertEqual(file_test.file_size, 0)
        # TODO check all the others

    def test_getting_mz_pe_value(self):
        with patch("lessons.unit_testing.main_one.pefile"):
            file_test = FileAnalysis(path_to_file="test_path")

        file_test.parsed_file.DOS_HEADER.e_magic = 0x123  # this is 291 in dec
        file_test.parsed_file.NT_HEADERS.Signature = 0x321  # this is 801 in dec
        file_test.get_mz_pe_value()

        # TODO make this work
        # self.assertEqual(file_test.magic_number, 291)
        # self.assertEqual(file_test.signature, 801)

        # Magic number is kept as a string. Why?
        # assert file_test.magic_number == 0x123
        # assert file_test.signature == 0x321