1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
|
// Copyright (c) 2016 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef QUICHE_QUIC_TEST_TOOLS_MOCK_QUIC_SESSION_VISITOR_H_
#define QUICHE_QUIC_TEST_TOOLS_MOCK_QUIC_SESSION_VISITOR_H_
#include "quiche/quic/core/quic_crypto_server_stream_base.h"
#include "quiche/quic/core/quic_time_wait_list_manager.h"
#include "quiche/quic/platform/api/quic_test.h"
namespace quic {
namespace test {
class MockQuicSessionVisitor : public QuicTimeWaitListManager::Visitor {
public:
MockQuicSessionVisitor();
MockQuicSessionVisitor(const MockQuicSessionVisitor&) = delete;
MockQuicSessionVisitor& operator=(const MockQuicSessionVisitor&) = delete;
~MockQuicSessionVisitor() override;
MOCK_METHOD(void, OnConnectionClosed,
(QuicConnectionId connection_id, QuicErrorCode error,
const std::string& error_details, ConnectionCloseSource source),
(override));
MOCK_METHOD(void, OnWriteBlocked, (QuicBlockedWriterInterface*), (override));
MOCK_METHOD(void, OnRstStreamReceived, (const QuicRstStreamFrame& frame),
(override));
MOCK_METHOD(void, OnStopSendingReceived, (const QuicStopSendingFrame& frame),
(override));
MOCK_METHOD(bool, TryAddNewConnectionId,
(const QuicConnectionId& server_connection_id,
const QuicConnectionId& new_connection_id),
(override));
MOCK_METHOD(void, OnConnectionIdRetired,
(const quic::QuicConnectionId& server_connection_id), (override));
MOCK_METHOD(void, OnConnectionAddedToTimeWaitList,
(QuicConnectionId connection_id), (override));
};
class MockQuicCryptoServerStreamHelper
: public QuicCryptoServerStreamBase::Helper {
public:
MockQuicCryptoServerStreamHelper();
MockQuicCryptoServerStreamHelper(const MockQuicCryptoServerStreamHelper&) =
delete;
MockQuicCryptoServerStreamHelper& operator=(
const MockQuicCryptoServerStreamHelper&) = delete;
~MockQuicCryptoServerStreamHelper() override;
MOCK_METHOD(bool, CanAcceptClientHello,
(const CryptoHandshakeMessage& message,
const QuicSocketAddress& client_address,
const QuicSocketAddress& peer_address,
const QuicSocketAddress& self_address, std::string*),
(const, override));
};
} // namespace test
} // namespace quic
#endif // QUICHE_QUIC_TEST_TOOLS_MOCK_QUIC_SESSION_VISITOR_H_
|
