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