class dtls_socket {
public:
    dtls_socket(asio::ip::udp::socket socket, SSL_CTX* ssl_ctx);
    
    auto async_handshake(handshake_type type, handler) -> void;
    auto set_receive_callback(callback) -> void;
    auto set_error_callback(callback) -> void;
    auto start_receive() -> void;
    auto stop_receive() -> void;
    auto async_send(data, handler) -> void;
    auto async_send_to(data, endpoint, handler) -> void;
    auto is_handshake_complete() const -> bool;
};

TEST_F(DtlsSocketTest, ConstructWithValidContext)
TEST_F(DtlsSocketTest, ConstructWithNullContextThrows)

TEST_F(DtlsSocketTest, ClientHandshakeSuccess)
TEST_F(DtlsSocketTest, ServerHandshakeSuccess)
TEST_F(DtlsSocketTest, HandshakeTimeout)
TEST_F(DtlsSocketTest, HandshakeWithInvalidCertificate)
TEST_F(DtlsSocketTest, HandshakePacketLoss)
TEST_F(DtlsSocketTest, HandshakeRetransmission)

TEST_F(DtlsSocketTest, SendAfterHandshake)
TEST_F(DtlsSocketTest, SendBeforeHandshakeFails)
TEST_F(DtlsSocketTest, ReceiveDecryptsCorrectly)
TEST_F(DtlsSocketTest, SendToSpecificEndpoint)
TEST_F(DtlsSocketTest, LargePayloadFragmentation)

TEST_F(DtlsSocketTest, ErrorCallbackOnSslError)
TEST_F(DtlsSocketTest, ErrorCallbackOnNetworkError)
TEST_F(DtlsSocketTest, RecoveryAfterTransientError)

TEST_F(DtlsSocketTest, ConcurrentSendOperations)
TEST_F(DtlsSocketTest, SendDuringReceive)
TEST_F(DtlsSocketTest, StopReceiveDuringCallback)

TEST_F(DtlsIntegrationTest, ClientServerCommunication)
TEST_F(DtlsIntegrationTest, MultipleClientsToServer)
TEST_F(DtlsIntegrationTest, SessionResumption)
TEST_F(DtlsIntegrationTest, ConnectionTimeout)
TEST_F(DtlsIntegrationTest, GracefulShutdown)

TEST_F(DtlsSecurityTest, RejectExpiredCertificate)
TEST_F(DtlsSecurityTest, RejectSelfSignedWhenRequired)
TEST_F(DtlsSecurityTest, VerifyHostname)
TEST_F(DtlsSecurityTest, MinimumTlsVersion)

class MockSslContext {
public:
    static SSL_CTX* create_test_context();
    static std::pair<std::string, std::string> generate_test_cert();
};