123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869 |
- import io
- import argparse
- import builtins
- import unittest
- import unittest.mock
- import libray
- class TestInterface(unittest.TestCase):
- @unittest.mock.patch('argparse.ArgumentParser.parse_args', return_value=argparse.Namespace())
- def test_decrypt_with_key(self, mock_args):
- io_object = io.BytesIO(b"some initial binary data: \x00\x01")
- os_stat = unittest.mock.Mock()
- os_stat.return_value.st_mode = 33188
- iso_filepath = unittest.mock.Mock()
- iso_filepath.open = unittest.mock.mock_open(read_data=b"some initial binary data: \x00\x01")
- iso_filepath.is_file.return_value = True
- iso_filepath.stat.return_value = unittest.mock.MagicMock()
- iso_filepath.stat.st_size = 1024
- iso_filepath.stat.st_mode = 33188
- iso_filepath.stat.S_ISBLK.return_value = False
- mock_args.iso = 'encrypted_mock.iso'
-
- with unittest.mock.patch('os.stat', os_stat):
- with unittest.mock.patch('builtins.open', iso_filepath.open):
- libray.core.decrypt(mock_args)
|