def test_update(self):
test_payload: Payload = Payload("test").with_suffix("<suffix>")
returned_payload: Payload = Payload().update(test_payload)
self.assertIsInstance(returned_payload, Payload)
self.assertEqual(returned_payload.raw, test_payload.raw)
self.assertEqual(returned_payload.final, test_payload.final)
self.assertDictEqual(returned_payload.config, test_payload.config)
def test_get_customized_payload_with_suffix(self):
test_payload = "test_payload"
test_suffix = ['>', '|']
return_expected = [
Payload(test_payload).with_suffix(test_suffix[0]),
Payload(test_payload).with_suffix(test_suffix[1])
]
Payloader.set_suffix(test_suffix)
returned_payloads = Payloader.get_customized_payload(test_payload)
self.assertIsInstance(returned_payloads, list)
assert_payload_list_is_equal(returned_payloads, return_expected)
def test_set_encoder(self):
test_payload = "test_payload"
return_expected = [
Payload(test_payload).with_encoder("746573745f7061796c6f6164",
"Hex")
]
test_encode_manager = EncodeManager()
test_encode_manager.set_encoders(([Hex()], []))
returned_encoded_payload = test_encode_manager.encode(
[Payload(test_payload)])
self.assertIsInstance(returned_encoded_payload, list)
assert_payload_list_is_equal(returned_encoded_payload, return_expected)
def test_get_customized_payload_with_lower(self):
test_payload = "test_payload"
return_expected = [Payload(test_payload).with_case(str.lower, "Lower")]
Payloader.set_lowercase()
returned_payloads = Payloader.get_customized_payload(test_payload)
self.assertIsInstance(returned_payloads, list)
assert_payload_list_is_equal(returned_payloads, return_expected)
def test_with_suffix(self):
test_payload_str = "test"
test_suffix = "<suffix>"
expected_config = {'suffix': test_suffix}
expected_final_payload = f"{test_payload_str}{test_suffix}"
returned_payload: Payload = Payload(test_payload_str).with_suffix(
test_suffix)
self.assertIsInstance(returned_payload, Payload)
self.assertDictEqual(returned_payload.config, expected_config)
self.assertEqual(returned_payload.final, expected_final_payload)
def test_with_encoder(self):
test_payload_str = "test"
test_encoded = "testencoded"
test_encoder = "TestEncoder"
expected_config = {'encoder': test_encoder}
expected_final_payload = test_encoded
returned_payload: Payload = Payload(test_payload_str).with_encoder(
test_encoded, test_encoder)
self.assertIsInstance(returned_payload, Payload)
self.assertDictEqual(returned_payload.config, expected_config)
self.assertEqual(returned_payload.final, expected_final_payload)
def test_with_case(self):
test_payload_str = "test"
test_case_callback = str.upper
test_case_method = "Upper"
expected_config = {'case': test_case_method}
expected_final_payload = test_payload_str.upper()
returned_payload: Payload = Payload(test_payload_str).with_case(
test_case_callback, test_case_method)
self.assertIsInstance(returned_payload, Payload)
self.assertDictEqual(returned_payload.config, expected_config)
self.assertEqual(returned_payload.final, expected_final_payload)
def test_get_customized_payload_without_mutation(self):
test_payload = "test_payload"
return_expected = [Payload(test_payload)]
returned_payloads = Payloader.get_customized_payload(test_payload)
self.assertIsInstance(returned_payloads, list)
assert_payload_list_is_equal(returned_payloads, return_expected)
def test_str(self):
expected_result = "test-payload"
returned_payload = Payload(expected_result)
self.assertEqual(str(returned_payload), expected_result)