self_encryption 0.35.0

Self encrypting files (convergent encryption plus obfuscation)
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69

import pytest
from click.testing import CliRunner
from self_encryption.cli import cli
import tempfile
import os
from pathlib import Path

@pytest.fixture
def runner():
    return CliRunner()

@pytest.fixture
def test_data():
    return b"Hello, World!" * 1000  # Make it large enough to encrypt

def test_encrypt_decrypt_flow(runner):
    with tempfile.TemporaryDirectory() as tmpdir:
        input_file = Path(tmpdir) / "input.dat"
        input_file.write_bytes(b"Hello, World!" * 100)  # Make it large enough
        chunks_dir = Path(tmpdir) / "chunks"
        chunks_dir.mkdir()
        data_map_file = Path(tmpdir) / "data_map.json"
        
        # Test encryption
        result = runner.invoke(cli, ['encrypt-file', str(input_file), str(chunks_dir)])
        print(f"Encryption error output: {result.output}")
        assert result.exit_code == 0
        
        # Save data map to file
        data_map_file.write_text(result.output)
        
        output_file = Path(tmpdir) / "output.dat"
        
        # Test decryption
        result = runner.invoke(cli, [
            'decrypt-file',
            str(data_map_file),
            str(chunks_dir),
            str(output_file)
        ])
        assert result.exit_code == 0
        
        # Verify content
        assert output_file.read_bytes() == b"Hello, World!" * 100

def test_streaming_decrypt(runner, test_data):
    with tempfile.TemporaryDirectory() as tmpdir:
        input_file = Path(tmpdir) / "input.dat"
        input_file.write_bytes(test_data)
        chunks_dir = Path(tmpdir) / "chunks"
        chunks_dir.mkdir()
        data_map_file = Path(tmpdir) / "data_map.json"
        output_file = Path(tmpdir) / "output.dat"
        
        # Encrypt
        result = runner.invoke(cli, ['encrypt-file', str(input_file), str(chunks_dir)])
        assert result.exit_code == 0
        data_map_file.write_text(result.output)
        
        # Test streaming decryption
        result = runner.invoke(cli, [
            'decrypt-file',
            '--streaming',
            str(data_map_file),
            str(chunks_dir),
            str(output_file)
        ])
        assert result.exit_code == 0
        assert output_file.read_bytes() == test_data