def test_decrypt_payload_with_wrong_protocol_raises_MKFetcherError(self, file_cache):
settings = {"use_regular": "enforce"}
output = AgentRawData(b"the first two bytes are not a number")
fetcher = TCPFetcher(
file_cache,
family=socket.AF_INET,
address=("1.2.3.4", 0),
timeout=0.0,
encryption_settings=settings,
use_only_cache=False,
)
with pytest.raises(MKFetcherError):
fetcher._decrypt(output)
def test_decrypt_plaintext_with_enforce_raises_MKFetcherError(self, file_cache):
settings = {"use_regular": "enforce"}
output = AgentRawData(b"<<<section:sep(0)>>>\nbody\n")
fetcher = TCPFetcher(
file_cache,
family=socket.AF_INET,
address=("1.2.3.4", 0),
timeout=0.0,
encryption_settings=settings,
use_only_cache=False,
)
with pytest.raises(MKFetcherError):
fetcher._decrypt(output)
def test_decrypt_plaintext_is_noop(self, file_cache):
settings = {"use_regular": "allow"}
output = AgentRawData(b"<<<section:sep(0)>>>\nbody\n")
fetcher = TCPFetcher(
file_cache,
family=socket.AF_INET,
address=("1.2.3.4", 0),
timeout=0.0,
encryption_settings=settings,
use_only_cache=False,
)
assert fetcher._decrypt(output) == output
def test_decrypt_plaintext_is_noop(self, file_cache: DefaultAgentFileCache) -> None:
settings = {"use_regular": "allow"}
output = b"<<<section:sep(0)>>>\nbody\n"
fetcher = TCPFetcher(
file_cache,
family=socket.AF_INET,
address=("1.2.3.4", 0),
host_name=HostName("irrelevant_for_this_test"),
timeout=0.0,
encryption_settings=settings,
use_only_cache=False,
)
assert fetcher._decrypt(TransportProtocol(output[:2]), AgentRawData(output[2:])) == output