def test__create_archive_with_an_export_file(mocker): export = Export() archive_path = None with TemporaryDirectory() as temp_dir, NamedTemporaryFile() as export_file: archive_path = export._create_archive(temp_dir, "mock.sd-export", {}, [export_file.name]) assert archive_path == os.path.join(temp_dir, "mock.sd-export") assert os.path.exists(archive_path) # sanity check assert not os.path.exists(archive_path)
def test__run_printer_preflight_raises_ExportError_if_not_empty_string(mocker): """ Ensure ExportError is raised if _run_disk_test returns anything other than 'USB_CONNECTED'. """ export = Export() export._create_archive = mocker.MagicMock(return_value="mock_archive_path") export._export_archive = mocker.MagicMock( return_value="SOMETHING_OTHER_THAN_EMPTY_STRING") with pytest.raises(ExportError): export._run_printer_preflight("mock_archive_dir")
def test__run_usb_test_raises_ExportError_if_not_USB_CONNECTED(mocker): ''' Ensure ExportError is raised if _run_disk_test returns anything other than 'USB_CONNECTED'. ''' export = Export() export._create_archive = mocker.MagicMock(return_value='mock_archive_path') export._export_archive = mocker.MagicMock( return_value='SOMETHING_OTHER_THAN_USB_CONNECTED') with pytest.raises(ExportError): export._run_usb_test('mock_archive_dir')
def test__run_disk_test_raises_ExportError_if_not_USB_ENCRYPTED(mocker): """ Ensure ExportError is raised if _run_disk_test returns anything other than 'USB_ENCRYPTED'. """ export = Export() export._create_archive = mocker.MagicMock(return_value="mock_archive_path") export._export_archive = mocker.MagicMock( return_value="SOMETHING_OTHER_THAN_USB_ENCRYPTED") with pytest.raises(ExportError): export._run_disk_test("mock_archive_dir")
def test__run_print_raises_ExportError_if_not_empty_string(mocker): ''' Ensure ExportError is raised if _run_print returns anything other than ''. ''' export = Export() export._create_archive = mocker.MagicMock(return_value='mock_archive_path') export._export_archive = mocker.MagicMock( return_value='SOMETHING_OTHER_THAN_EMPTY_STRING') with pytest.raises(ExportError): export._run_print('mock_archive_dir', ['mock_filepath'])
def test__create_archive(mocker): """ Ensure _create_archive creates an archive in the supplied directory. """ export = Export() archive_path = None with TemporaryDirectory() as temp_dir: archive_path = export._create_archive(temp_dir, "mock.sd-export", {}) assert archive_path == os.path.join(temp_dir, "mock.sd-export") assert os.path.exists(archive_path) # sanity check assert not os.path.exists(archive_path)
def test__run_disk_export_raises_ExportError_if_not_empty_string(mocker): """ Ensure ExportError is raised if _run_disk_test returns anything other than ''. """ export = Export() export._create_archive = mocker.MagicMock(return_value="mock_archive_path") export._export_archive = mocker.MagicMock( return_value="SOMETHING_OTHER_THAN_EMPTY_STRING") with pytest.raises(ExportError): export._run_disk_export("mock_archive_dir", ["mock_filepath"], "mock_passphrase")
def test__create_archive_with_multiple_export_files(mocker): """ Ensure an archive """ export = Export() archive_path = None with TemporaryDirectory() as temp_dir, NamedTemporaryFile( ) as export_file_one, NamedTemporaryFile() as export_file_two: # noqa archive_path = export._create_archive( temp_dir, "mock.sd-export", {}, [export_file_one.name, export_file_two.name]) assert archive_path == os.path.join(temp_dir, "mock.sd-export") assert os.path.exists(archive_path) # sanity check assert not os.path.exists(archive_path)
def test__run_printer_preflight(mocker): """ Ensure _export_archive and _create_archive are called with the expected parameters, _export_archive is called with the return value of _create_archive, and _run_disk_test returns without error if 'USB_CONNECTED' is the return value of _export_archive. """ export = Export() export._create_archive = mocker.MagicMock(return_value="mock_archive_path") export._export_archive = mocker.MagicMock(return_value="") export._run_printer_preflight("mock_archive_dir") export._export_archive.assert_called_once_with("mock_archive_path") export._create_archive.assert_called_once_with( "mock_archive_dir", "printer-preflight.sd-export", {"device": "printer-preflight"})
def test__run_disk_test(mocker): """ Ensure _export_archive and _create_archive are called with the expected parameters, _export_archive is called with the return value of _create_archive, and _run_disk_test returns without error if 'USB_ENCRYPTED' is the ouput status of _export_archive. """ export = Export() export._create_archive = mocker.MagicMock(return_value="mock_archive_path") export._export_archive = mocker.MagicMock(return_value="USB_ENCRYPTED") export._run_disk_test("mock_archive_dir") export._export_archive.assert_called_once_with("mock_archive_path") export._create_archive.assert_called_once_with("mock_archive_dir", "disk-test.sd-export", {"device": "disk-test"})
def test__run_print(mocker): """ Ensure _export_archive and _create_archive are called with the expected parameters and _export_archive is called with the return value of _create_archive. """ export = Export() export._create_archive = mocker.MagicMock(return_value="mock_archive_path") export._export_archive = mocker.MagicMock(return_value="") export._run_print("mock_archive_dir", ["mock_filepath"]) export._export_archive.assert_called_once_with("mock_archive_path") export._create_archive.assert_called_once_with("mock_archive_dir", "print_archive.sd-export", {"device": "printer"}, ["mock_filepath"])
def test__run_printer_preflight(mocker): ''' Ensure _export_archive and _create_archive are called with the expected parameters, _export_archive is called with the return value of _create_archive, and _run_disk_test returns without error if 'USB_CONNECTED' is the return value of _export_archive. ''' export = Export() export._create_archive = mocker.MagicMock(return_value='mock_archive_path') export._export_archive = mocker.MagicMock(return_value='') export._run_printer_preflight('mock_archive_dir') export._export_archive.assert_called_once_with('mock_archive_path') export._create_archive.assert_called_once_with( 'mock_archive_dir', 'printer-preflight.sd-export', {'device': 'printer-preflight'})
def test__run_disk_test(mocker): ''' Ensure _export_archive and _create_archive are called with the expected parameters, _export_archive is called with the return value of _create_archive, and _run_disk_test returns without error if 'USB_ENCRYPTED' is the ouput status of _export_archive. ''' export = Export() export._create_archive = mocker.MagicMock(return_value='mock_archive_path') export._export_archive = mocker.MagicMock(return_value='USB_ENCRYPTED') export._run_disk_test('mock_archive_dir') export._export_archive.assert_called_once_with('mock_archive_path') export._create_archive.assert_called_once_with('mock_archive_dir', 'disk-test.sd-export', {'device': 'disk-test'})
def test__run_print(mocker): ''' Ensure _export_archive and _create_archive are called with the expected parameters and _export_archive is called with the return value of _create_archive. ''' export = Export() export._create_archive = mocker.MagicMock(return_value='mock_archive_path') export._export_archive = mocker.MagicMock(return_value='') export._run_print('mock_archive_dir', ['mock_filepath']) export._export_archive.assert_called_once_with('mock_archive_path') export._create_archive.assert_called_once_with('mock_archive_dir', 'print_archive.sd-export', { 'device': 'printer', }, ['mock_filepath'])
def test__run_disk_export(mocker): ''' Ensure _export_archive and _create_archive are called with the expected parameters, _export_archive is called with the return value of _create_archive, and _run_disk_test returns without error if '' is the ouput status of _export_archive. ''' export = Export() export._create_archive = mocker.MagicMock(return_value='mock_archive_path') export._export_archive = mocker.MagicMock(return_value='') export._run_disk_export('mock_archive_dir', ['mock_filepath'], 'mock_passphrase') export._export_archive.assert_called_once_with('mock_archive_path') export._create_archive.assert_called_once_with( 'mock_archive_dir', 'archive.sd-export', { 'encryption_key': 'mock_passphrase', 'device': 'disk', 'encryption_method': 'luks' }, ['mock_filepath'])
def test__run_disk_export(mocker): """ Ensure _export_archive and _create_archive are called with the expected parameters, _export_archive is called with the return value of _create_archive, and _run_disk_test returns without error if '' is the ouput status of _export_archive. """ export = Export() export._create_archive = mocker.MagicMock(return_value="mock_archive_path") export._export_archive = mocker.MagicMock(return_value="") export._run_disk_export("mock_archive_dir", ["mock_filepath"], "mock_passphrase") export._export_archive.assert_called_once_with("mock_archive_path") export._create_archive.assert_called_once_with( "mock_archive_dir", "archive.sd-export", { "encryption_key": "mock_passphrase", "device": "disk", "encryption_method": "luks" }, ["mock_filepath"], )