ceph/src/jaegertracing/thrift/lib/cpp/test/TServerTransportTest.cpp

   1 /*
   2  * Licensed to the Apache Software Foundation (ASF) under one
   3  * or more contributor license agreements. See the NOTICE file
   4  * distributed with this work for additional information
   5  * regarding copyright ownership. The ASF licenses this file
   6  * to you under the Apache License, Version 2.0 (the
   7  * "License"); you may not use this file except in compliance
   8  * with the License. You may obtain a copy of the License at
   9  *
  10  *   http://www.apache.org/licenses/LICENSE-2.0
  11  *
  12  * Unless required by applicable law or agreed to in writing,
  13  * software distributed under the License is distributed on an
  14  * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
  15  * KIND, either express or implied. See the License for the
  16  * specific language governing permissions and limitations
  17  * under the License.
  18  */
  19
  20 #include <boost/test/auto_unit_test.hpp>
  21 #include <thrift/transport/TSocket.h>
  22 #include <thrift/transport/TServerTransport.h>
  23 #include <memory>
  24
  25 using apache::thrift::transport::TServerTransport;
  26 using apache::thrift::transport::TTransport;
  27 using apache::thrift::transport::TTransportException;
  28 using std::shared_ptr;
  29
  30 BOOST_AUTO_TEST_SUITE(TServerTransportTest)
  31
  32 class TestTTransport : public TTransport {};
  33
  34 class TestTServerTransport : public TServerTransport {
  35 public:
  36   TestTServerTransport() : valid_(true) {}
  37   void close() override {}
  38   bool valid_;
  39
  40 protected:
  41   shared_ptr<TTransport> acceptImpl() override {
  42     return valid_ ? std::make_shared<TestTTransport>()
  43                   : shared_ptr<TestTTransport>();
  44   }
  45 };
  46
  47 BOOST_AUTO_TEST_CASE(test_positive_accept) {
  48   TestTServerTransport uut;
  49   BOOST_CHECK(uut.accept());
  50 }
  51
  52 BOOST_AUTO_TEST_CASE(test_negative_accept) {
  53   TestTServerTransport uut;
  54   uut.valid_ = false;
  55   BOOST_CHECK_THROW(uut.accept(), TTransportException);
  56 }
  57
  58 BOOST_AUTO_TEST_SUITE_END()