channel_test_peer.h 1.2 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344
  1. //
  2. //
  3. // Copyright 2020 gRPC authors.
  4. //
  5. // Licensed under the Apache License, Version 2.0 (the "License");
  6. // you may not use this file except in compliance with the License.
  7. // You may obtain a copy of the License at
  8. //
  9. // http://www.apache.org/licenses/LICENSE-2.0
  10. //
  11. // Unless required by applicable law or agreed to in writing, software
  12. // distributed under the License is distributed on an "AS IS" BASIS,
  13. // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14. // See the License for the specific language governing permissions and
  15. // limitations under the License.
  16. //
  17. //
  18. #ifndef GRPCPP_TEST_CHANNEL_TEST_PEER_H
  19. #define GRPCPP_TEST_CHANNEL_TEST_PEER_H
  20. #include <grpcpp/channel.h>
  21. namespace grpc {
  22. namespace testing {
  23. /// A test-only class to access private members of Channel.
  24. class ChannelTestPeer {
  25. public:
  26. explicit ChannelTestPeer(Channel* channel) : channel_(channel) {}
  27. /// Provide the gRPC Core channel
  28. grpc_channel* channel() const { return channel_->c_channel_; }
  29. int registered_calls() const;
  30. int registration_attempts() const;
  31. private:
  32. Channel* channel_; // not owned
  33. };
  34. } // namespace testing
  35. } // namespace grpc
  36. #endif // GRPCPP_TEST_CHANNEL_TEST_PEER_H