From d199265e95e275290165161803b233ff57d6919d Mon Sep 17 00:00:00 2001
From: Artem Boldariev <artem@boldariev.com>
Date: Tue, 9 May 2023 20:49:38 +0300
Subject: [PATCH] Add PROXYv2 related tests for TLS DNS transport

This commit adds a set of PROXYv2 related tests to ensure that Stream
DNS over TLS supports this mode.
---
 tests/isc/tlsdns_test.c | 34 ++++++++++++++++++++++++++++++++++
 1 file changed, 34 insertions(+)

diff --git a/tests/isc/tlsdns_test.c b/tests/isc/tlsdns_test.c
index d238c831c1..9f98ed3ebc 100644
--- a/tests/isc/tlsdns_test.c
+++ b/tests/isc/tlsdns_test.c
@@ -138,6 +138,24 @@ ISC_LOOP_TEST_IMPL(tlsdns_recv_send) {
 	}
 }
 
+/* PROXY tests */
+
+ISC_LOOP_TEST_IMPL(proxy_tlsdns_noop) { loop_test_tlsdns_noop(arg); }
+
+ISC_LOOP_TEST_IMPL(proxy_tlsdns_noresponse) {
+	loop_test_tlsdns_noresponse(arg);
+}
+
+ISC_LOOP_TEST_IMPL(proxy_tlsdns_timeout_recovery) {
+	loop_test_tlsdns_timeout_recovery(arg);
+}
+
+ISC_LOOP_TEST_IMPL(proxy_tlsdns_recv_one) { loop_test_tlsdns_recv_one(arg); }
+
+ISC_LOOP_TEST_IMPL(proxy_tlsdns_recv_two) { loop_test_tlsdns_recv_two(arg); }
+
+ISC_LOOP_TEST_IMPL(proxy_tlsdns_recv_send) { loop_test_tlsdns_recv_send(arg); }
+
 ISC_TEST_LIST_START
 
 ISC_TEST_ENTRY_CUSTOM(tlsdns_noop, stream_noop_setup, stream_noop_teardown)
@@ -154,6 +172,22 @@ ISC_TEST_ENTRY_CUSTOM(tlsdns_recv_send, stream_recv_send_setup,
 
 /* FIXME: Re-add the noalpn tests */
 
+/* PROXY */
+
+ISC_TEST_ENTRY_CUSTOM(proxy_tlsdns_noop, proxystream_noop_setup,
+		      proxystream_noop_teardown)
+ISC_TEST_ENTRY_CUSTOM(proxy_tlsdns_noresponse, proxystream_noresponse_setup,
+		      proxystream_noresponse_teardown)
+ISC_TEST_ENTRY_CUSTOM(proxy_tlsdns_timeout_recovery,
+		      proxystream_timeout_recovery_setup,
+		      proxystream_timeout_recovery_teardown)
+ISC_TEST_ENTRY_CUSTOM(proxy_tlsdns_recv_one, proxystream_recv_one_setup,
+		      proxystream_recv_one_teardown)
+ISC_TEST_ENTRY_CUSTOM(proxy_tlsdns_recv_two, proxystream_recv_two_setup,
+		      proxystream_recv_two_teardown)
+ISC_TEST_ENTRY_CUSTOM(proxy_tlsdns_recv_send, proxystream_recv_send_setup,
+		      proxystream_recv_send_teardown)
+
 ISC_TEST_LIST_END
 
 static int