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/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: TestTServerTransport()36 TestTServerTransport() : valid_(true) {} close()37 void close() override {} 38 bool valid_; 39 40 protected: acceptImpl()41 shared_ptr<TTransport> acceptImpl() override { 42 return valid_ ? std::make_shared<TestTTransport>() 43 : shared_ptr<TestTTransport>(); 44 } 45 }; 46 BOOST_AUTO_TEST_CASE(test_positive_accept)47BOOST_AUTO_TEST_CASE(test_positive_accept) { 48 TestTServerTransport uut; 49 BOOST_CHECK(uut.accept()); 50 } 51 BOOST_AUTO_TEST_CASE(test_negative_accept)52BOOST_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() 59
