| OLD | NEW | 
|   1 /* |   1 /* | 
|   2  *  Copyright 2016 The WebRTC project authors. All Rights Reserved. |   2  *  Copyright 2016 The WebRTC project authors. All Rights Reserved. | 
|   3  * |   3  * | 
|   4  *  Use of this source code is governed by a BSD-style license |   4  *  Use of this source code is governed by a BSD-style license | 
|   5  *  that can be found in the LICENSE file in the root of the source |   5  *  that can be found in the LICENSE file in the root of the source | 
|   6  *  tree. An additional intellectual property rights grant can be found |   6  *  tree. An additional intellectual property rights grant can be found | 
|   7  *  in the file PATENTS.  All contributing project authors may |   7  *  in the file PATENTS.  All contributing project authors may | 
|   8  *  be found in the AUTHORS file in the root of the source tree. |   8  *  be found in the AUTHORS file in the root of the source tree. | 
|   9  */ |   9  */ | 
|  10  |  10  | 
|  11 #ifndef PC_TEST_MOCK_DATACHANNEL_H_ |  11 #ifndef PC_TEST_MOCK_DATACHANNEL_H_ | 
|  12 #define PC_TEST_MOCK_DATACHANNEL_H_ |  12 #define PC_TEST_MOCK_DATACHANNEL_H_ | 
|  13  |  13  | 
|  14 #include "pc/datachannel.h" |  14 #include "pc/datachannel.h" | 
|  15 #include "test/gmock.h" |  15 #include "test/gmock.h" | 
|  16  |  16  | 
|  17 namespace webrtc { |  17 namespace webrtc { | 
|  18  |  18  | 
|  19 class MockDataChannel : public rtc::RefCountedObject<DataChannel> { |  19 class MockDataChannel : public DataChannel { | 
|  20  public: |  20  public: | 
|  21   MockDataChannel(int id, DataState state) |  21   MockDataChannel(int id, DataState state) | 
|  22       : MockDataChannel(id, "MockDataChannel", state, "udp", 0, 0, 0, 0) { |  22       : MockDataChannel(id, "MockDataChannel", state, "udp", 0, 0, 0, 0) { | 
|  23   } |  23   } | 
|  24   MockDataChannel( |  24   MockDataChannel( | 
|  25       int id, |  25       int id, | 
|  26       const std::string& label, |  26       const std::string& label, | 
|  27       DataState state, |  27       DataState state, | 
|  28       const std::string& protocol, |  28       const std::string& protocol, | 
|  29       uint32_t messages_sent, |  29       uint32_t messages_sent, | 
|  30       uint64_t bytes_sent, |  30       uint64_t bytes_sent, | 
|  31       uint32_t messages_received, |  31       uint32_t messages_received, | 
|  32       uint64_t bytes_received) |  32       uint64_t bytes_received) | 
|  33       : rtc::RefCountedObject<DataChannel>( |  33       : DataChannel(nullptr, cricket::DCT_NONE, label) { | 
|  34             nullptr, cricket::DCT_NONE, label) { |  | 
|  35     EXPECT_CALL(*this, id()).WillRepeatedly(testing::Return(id)); |  34     EXPECT_CALL(*this, id()).WillRepeatedly(testing::Return(id)); | 
|  36     EXPECT_CALL(*this, state()).WillRepeatedly(testing::Return(state)); |  35     EXPECT_CALL(*this, state()).WillRepeatedly(testing::Return(state)); | 
|  37     EXPECT_CALL(*this, protocol()).WillRepeatedly(testing::Return(protocol)); |  36     EXPECT_CALL(*this, protocol()).WillRepeatedly(testing::Return(protocol)); | 
|  38     EXPECT_CALL(*this, messages_sent()).WillRepeatedly( |  37     EXPECT_CALL(*this, messages_sent()).WillRepeatedly( | 
|  39         testing::Return(messages_sent)); |  38         testing::Return(messages_sent)); | 
|  40     EXPECT_CALL(*this, bytes_sent()).WillRepeatedly( |  39     EXPECT_CALL(*this, bytes_sent()).WillRepeatedly( | 
|  41         testing::Return(bytes_sent)); |  40         testing::Return(bytes_sent)); | 
|  42     EXPECT_CALL(*this, messages_received()).WillRepeatedly( |  41     EXPECT_CALL(*this, messages_received()).WillRepeatedly( | 
|  43         testing::Return(messages_received)); |  42         testing::Return(messages_received)); | 
|  44     EXPECT_CALL(*this, bytes_received()).WillRepeatedly( |  43     EXPECT_CALL(*this, bytes_received()).WillRepeatedly( | 
|  45         testing::Return(bytes_received)); |  44         testing::Return(bytes_received)); | 
|  46   } |  45   } | 
|  47   MOCK_CONST_METHOD0(id, int()); |  46   MOCK_CONST_METHOD0(id, int()); | 
|  48   MOCK_CONST_METHOD0(state, DataState()); |  47   MOCK_CONST_METHOD0(state, DataState()); | 
|  49   MOCK_CONST_METHOD0(protocol, std::string()); |  48   MOCK_CONST_METHOD0(protocol, std::string()); | 
|  50   MOCK_CONST_METHOD0(messages_sent, uint32_t()); |  49   MOCK_CONST_METHOD0(messages_sent, uint32_t()); | 
|  51   MOCK_CONST_METHOD0(bytes_sent, uint64_t()); |  50   MOCK_CONST_METHOD0(bytes_sent, uint64_t()); | 
|  52   MOCK_CONST_METHOD0(messages_received, uint32_t()); |  51   MOCK_CONST_METHOD0(messages_received, uint32_t()); | 
|  53   MOCK_CONST_METHOD0(bytes_received, uint64_t()); |  52   MOCK_CONST_METHOD0(bytes_received, uint64_t()); | 
|  54 }; |  53 }; | 
|  55  |  54  | 
|  56 }  // namespace webrtc |  55 }  // namespace webrtc | 
|  57  |  56  | 
|  58 #endif  // PC_TEST_MOCK_DATACHANNEL_H_ |  57 #endif  // PC_TEST_MOCK_DATACHANNEL_H_ | 
| OLD | NEW |