mirror of
https://github.com/fnproject/fn.git
synced 2022-10-28 21:29:17 +03:00
* add jaeger support, link hot container & req span * adds jaeger support now with FN_JAEGER_URL, there's a simple tutorial in the operating/metrics.md file now and it's pretty easy to get up and running. * links a hot request span to a hot container span. when we change this to sample at a lower ratio we'll need to finagle the hot container span to always sample or something, otherwise we'll hide that info. at least, since we're sampling at 100% for now if this is flipped on, can see freeze/unfreeze etc. if they hit. this is useful for debugging. note that zipkin's exporter does not follow the link at all, hence jaeger... and they're backed by the Cloud Empire now (CNCF) so we'll probably use it anyway. * vendor: add thrift for jaeger
387 lines
10 KiB
C++
387 lines
10 KiB
C++
/*
|
|
* Licensed to the Apache Software Foundation (ASF) under one
|
|
* or more contributor license agreements. See the NOTICE file
|
|
* distributed with this work for additional information
|
|
* regarding copyright ownership. The ASF licenses this file
|
|
* to you under the Apache License, Version 2.0 (the
|
|
* "License"); you may not use this file except in compliance
|
|
* with the License. You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing,
|
|
* software distributed under the License is distributed on an
|
|
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
|
* KIND, either express or implied. See the License for the
|
|
* specific language governing permissions and limitations
|
|
* under the License.
|
|
*
|
|
* Contains some contributions under the Thrift Software License.
|
|
* Please see doc/old-thrift-license.txt in the Thrift distribution for
|
|
* details.
|
|
*/
|
|
|
|
#include <map>
|
|
#include <thrift/protocol/TDebugProtocol.h>
|
|
#include <thrift/protocol/TBinaryProtocol.h>
|
|
#include <thrift/transport/TBufferTransports.h>
|
|
#include "gen-cpp/OptionalRequiredTest_types.h"
|
|
|
|
#define BOOST_TEST_MODULE OptionalRequiredTest
|
|
#include <boost/test/unit_test.hpp>
|
|
|
|
using namespace thrift::test;
|
|
using namespace apache::thrift;
|
|
using namespace apache::thrift::transport;
|
|
using namespace apache::thrift::protocol;
|
|
|
|
/*
|
|
template<typename Struct>
|
|
void trywrite(const Struct& s, bool should_work) {
|
|
bool worked;
|
|
try {
|
|
TBinaryProtocol protocol(stdcxx::shared_ptr<TTransport>(new TMemoryBuffer));
|
|
s.write(&protocol);
|
|
worked = true;
|
|
} catch (TProtocolException & ex) {
|
|
worked = false;
|
|
}
|
|
BOOST_CHECK(worked == should_work);
|
|
}
|
|
*/
|
|
|
|
template <typename Struct1, typename Struct2>
|
|
void write_to_read(const Struct1& w, Struct2& r) {
|
|
TBinaryProtocol protocol(stdcxx::shared_ptr<TTransport>(new TMemoryBuffer));
|
|
w.write(&protocol);
|
|
r.read(&protocol);
|
|
}
|
|
|
|
BOOST_AUTO_TEST_CASE(test_optional_required_1) {
|
|
OldSchool o;
|
|
|
|
const std::string expected_result(
|
|
"OldSchool {\n"
|
|
" 01: im_int (i16) = 0,\n"
|
|
" 02: im_str (string) = \"\",\n"
|
|
" 03: im_big (list) = list<map>[0] {\n"
|
|
" },\n"
|
|
"}");
|
|
const std::string result(apache::thrift::ThriftDebugString(o));
|
|
|
|
BOOST_CHECK_MESSAGE(!expected_result.compare(result),
|
|
"Expected:\n" << expected_result << "\nGotten:\n" << result);
|
|
}
|
|
|
|
BOOST_AUTO_TEST_CASE(test_optional_required_2_1) {
|
|
Simple s;
|
|
|
|
const std::string expected_result(
|
|
"Simple {\n"
|
|
" 01: im_default (i16) = 0,\n"
|
|
" 02: im_required (i16) = 0,\n"
|
|
"}");
|
|
const std::string result(apache::thrift::ThriftDebugString(s));
|
|
|
|
BOOST_CHECK_MESSAGE(!expected_result.compare(result),
|
|
"Expected:\n" << expected_result << "\nGotten:\n" << result);
|
|
}
|
|
|
|
BOOST_AUTO_TEST_CASE(test_optional_required_2_2) {
|
|
Simple s;
|
|
s.im_optional = 10;
|
|
|
|
const std::string expected_result(
|
|
"Simple {\n"
|
|
" 01: im_default (i16) = 0,\n"
|
|
" 02: im_required (i16) = 0,\n"
|
|
"}");
|
|
const std::string result(apache::thrift::ThriftDebugString(s));
|
|
|
|
BOOST_CHECK_MESSAGE(!expected_result.compare(result),
|
|
"Expected:\n" << expected_result << "\nGotten:\n" << result);
|
|
}
|
|
|
|
BOOST_AUTO_TEST_CASE(test_optional_required_2_3) {
|
|
Simple s;
|
|
s.im_optional = 10;
|
|
s.__isset.im_optional = true;
|
|
|
|
const std::string expected_result(
|
|
"Simple {\n"
|
|
" 01: im_default (i16) = 0,\n"
|
|
" 02: im_required (i16) = 0,\n"
|
|
" 03: im_optional (i16) = 10,\n"
|
|
"}");
|
|
const std::string result(apache::thrift::ThriftDebugString(s));
|
|
|
|
BOOST_CHECK_MESSAGE(!expected_result.compare(result),
|
|
"Expected:\n" << expected_result << "\nGotten:\n" << result);
|
|
}
|
|
|
|
BOOST_AUTO_TEST_CASE(test_optional_required_2_4) {
|
|
Simple s;
|
|
s.__isset.im_optional = true;
|
|
|
|
const std::string expected_result(
|
|
"Simple {\n"
|
|
" 01: im_default (i16) = 0,\n"
|
|
" 02: im_required (i16) = 0,\n"
|
|
" 03: im_optional (i16) = 0,\n"
|
|
"}");
|
|
const std::string result(apache::thrift::ThriftDebugString(s));
|
|
|
|
BOOST_CHECK_MESSAGE(!expected_result.compare(result),
|
|
"Expected:\n" << expected_result << "\nGotten:\n" << result);
|
|
}
|
|
|
|
BOOST_AUTO_TEST_CASE(test_optional_required_2_5) {
|
|
Simple s;
|
|
s.__isset.im_optional = true;
|
|
s.im_optional = 10;
|
|
|
|
const std::string expected_result(
|
|
"Simple {\n"
|
|
" 01: im_default (i16) = 0,\n"
|
|
" 02: im_required (i16) = 0,\n"
|
|
" 03: im_optional (i16) = 10,\n"
|
|
"}");
|
|
const std::string result(apache::thrift::ThriftDebugString(s));
|
|
|
|
BOOST_CHECK_MESSAGE(!expected_result.compare(result),
|
|
"Expected:\n" << expected_result << "\nGotten:\n" << result);
|
|
}
|
|
|
|
BOOST_AUTO_TEST_CASE(test_optional_required_3) {
|
|
// assign/copy-construct with non-required fields
|
|
|
|
Simple s1, s2;
|
|
s1.__isset.im_default = true;
|
|
s1.__set_im_optional(10);
|
|
BOOST_CHECK(s1.__isset.im_default);
|
|
BOOST_CHECK(s1.__isset.im_optional);
|
|
|
|
s2 = s1;
|
|
|
|
BOOST_CHECK(s2.__isset.im_default);
|
|
BOOST_CHECK(s2.__isset.im_optional);
|
|
|
|
Simple s3(s1);
|
|
|
|
BOOST_CHECK(s3.__isset.im_default);
|
|
BOOST_CHECK(s3.__isset.im_optional);
|
|
}
|
|
|
|
BOOST_AUTO_TEST_CASE(test_optional_required_4) {
|
|
// Write-to-read with optional fields.
|
|
|
|
Simple s1, s2, s3;
|
|
s1.im_optional = 10;
|
|
BOOST_CHECK(!s1.__isset.im_default);
|
|
// BOOST_CHECK(!s1.__isset.im_required); // Compile error.
|
|
BOOST_CHECK(!s1.__isset.im_optional);
|
|
|
|
write_to_read(s1, s2);
|
|
|
|
BOOST_CHECK(s2.__isset.im_default);
|
|
// BOOST_CHECK( s2.__isset.im_required); // Compile error.
|
|
BOOST_CHECK(!s2.__isset.im_optional);
|
|
BOOST_CHECK(s3.im_optional == 0);
|
|
|
|
s1.__isset.im_optional = true;
|
|
write_to_read(s1, s3);
|
|
|
|
BOOST_CHECK(s3.__isset.im_default);
|
|
// BOOST_CHECK( s3.__isset.im_required); // Compile error.
|
|
BOOST_CHECK(s3.__isset.im_optional);
|
|
BOOST_CHECK(s3.im_optional == 10);
|
|
}
|
|
|
|
BOOST_AUTO_TEST_CASE(test_optional_required_5) {
|
|
// Writing between optional and default.
|
|
|
|
Tricky1 t1;
|
|
Tricky2 t2;
|
|
|
|
t2.im_optional = 10;
|
|
write_to_read(t2, t1);
|
|
write_to_read(t1, t2);
|
|
BOOST_CHECK(!t1.__isset.im_default);
|
|
BOOST_CHECK(t2.__isset.im_optional);
|
|
BOOST_CHECK(t1.im_default == t2.im_optional);
|
|
BOOST_CHECK(t1.im_default == 0);
|
|
}
|
|
|
|
BOOST_AUTO_TEST_CASE(test_optional_required_6) {
|
|
// Writing between default and required.
|
|
|
|
Tricky1 t1;
|
|
Tricky3 t3;
|
|
write_to_read(t1, t3);
|
|
write_to_read(t3, t1);
|
|
BOOST_CHECK(t1.__isset.im_default);
|
|
}
|
|
|
|
BOOST_AUTO_TEST_CASE(test_optional_required_7) {
|
|
// Writing between optional and required.
|
|
|
|
Tricky2 t2;
|
|
Tricky3 t3;
|
|
t2.__isset.im_optional = true;
|
|
write_to_read(t2, t3);
|
|
write_to_read(t3, t2);
|
|
}
|
|
|
|
BOOST_AUTO_TEST_CASE(test_optional_required_8) {
|
|
// Mu-hu-ha-ha-ha!
|
|
|
|
Tricky2 t2;
|
|
Tricky3 t3;
|
|
try {
|
|
write_to_read(t2, t3);
|
|
abort();
|
|
} catch (const TProtocolException&) {
|
|
}
|
|
|
|
write_to_read(t3, t2);
|
|
BOOST_CHECK(t2.__isset.im_optional);
|
|
}
|
|
|
|
BOOST_AUTO_TEST_CASE(test_optional_required_9) {
|
|
Complex c;
|
|
|
|
const std::string expected_result(
|
|
"Complex {\n"
|
|
" 01: cp_default (i16) = 0,\n"
|
|
" 02: cp_required (i16) = 0,\n"
|
|
" 04: the_map (map) = map<i16,struct>[0] {\n"
|
|
" },\n"
|
|
" 05: req_simp (struct) = Simple {\n"
|
|
" 01: im_default (i16) = 0,\n"
|
|
" 02: im_required (i16) = 0,\n"
|
|
" },\n"
|
|
"}");
|
|
const std::string result(apache::thrift::ThriftDebugString(c));
|
|
|
|
BOOST_CHECK_MESSAGE(!expected_result.compare(result),
|
|
"Expected:\n" << expected_result << "\nGotten:\n" << result);
|
|
}
|
|
|
|
BOOST_AUTO_TEST_CASE(test_optional_required_10) {
|
|
Tricky1 t1;
|
|
Tricky2 t2;
|
|
// Compile error.
|
|
//(void)(t1 == t2);
|
|
}
|
|
|
|
BOOST_AUTO_TEST_CASE(test_optional_required_11) {
|
|
OldSchool o1, o2, o3;
|
|
BOOST_CHECK(o1 == o2);
|
|
o1.im_int = o2.im_int = 10;
|
|
BOOST_CHECK(o1 == o2);
|
|
o1.__isset.im_int = true;
|
|
o2.__isset.im_int = false;
|
|
BOOST_CHECK(o1 == o2);
|
|
o1.im_int = 20;
|
|
o1.__isset.im_int = false;
|
|
BOOST_CHECK(o1 != o2);
|
|
o1.im_int = 10;
|
|
BOOST_CHECK(o1 == o2);
|
|
o1.im_str = o2.im_str = "foo";
|
|
BOOST_CHECK(o1 == o2);
|
|
o1.__isset.im_str = o2.__isset.im_str = true;
|
|
BOOST_CHECK(o1 == o2);
|
|
std::map<int32_t, std::string> mymap;
|
|
mymap[1] = "bar";
|
|
mymap[2] = "baz";
|
|
o1.im_big.push_back(std::map<int32_t, std::string>());
|
|
BOOST_CHECK(o1 != o2);
|
|
o2.im_big.push_back(std::map<int32_t, std::string>());
|
|
BOOST_CHECK(o1 == o2);
|
|
o2.im_big.push_back(mymap);
|
|
BOOST_CHECK(o1 != o2);
|
|
o1.im_big.push_back(mymap);
|
|
BOOST_CHECK(o1 == o2);
|
|
|
|
TBinaryProtocol protocol(stdcxx::shared_ptr<TTransport>(new TMemoryBuffer));
|
|
o1.write(&protocol);
|
|
|
|
o1.im_big.push_back(mymap);
|
|
mymap[3] = "qux";
|
|
o2.im_big.push_back(mymap);
|
|
BOOST_CHECK(o1 != o2);
|
|
o1.im_big.back()[3] = "qux";
|
|
BOOST_CHECK(o1 == o2);
|
|
|
|
o3.read(&protocol);
|
|
o3.im_big.push_back(mymap);
|
|
BOOST_CHECK(o1 == o3);
|
|
|
|
const std::string expected_result(
|
|
"OldSchool {\n"
|
|
" 01: im_int (i16) = 10,\n"
|
|
" 02: im_str (string) = \"foo\",\n"
|
|
" 03: im_big (list) = list<map>[3] {\n"
|
|
" [0] = map<i32,string>[0] {\n"
|
|
" },\n"
|
|
" [1] = map<i32,string>[2] {\n"
|
|
" 1 -> \"bar\",\n"
|
|
" 2 -> \"baz\",\n"
|
|
" },\n"
|
|
" [2] = map<i32,string>[3] {\n"
|
|
" 1 -> \"bar\",\n"
|
|
" 2 -> \"baz\",\n"
|
|
" 3 -> \"qux\",\n"
|
|
" },\n"
|
|
" },\n"
|
|
"}");
|
|
const std::string result(apache::thrift::ThriftDebugString(o3));
|
|
|
|
BOOST_CHECK_MESSAGE(!expected_result.compare(result),
|
|
"Expected:\n" << expected_result << "\nGotten:\n" << result);
|
|
}
|
|
|
|
BOOST_AUTO_TEST_CASE(test_optional_required_12) {
|
|
Tricky2 t1, t2;
|
|
BOOST_CHECK(t1.__isset.im_optional == false);
|
|
BOOST_CHECK(t2.__isset.im_optional == false);
|
|
BOOST_CHECK(t1 == t2);
|
|
t1.im_optional = 5;
|
|
BOOST_CHECK(t1 == t2);
|
|
t2.im_optional = 5;
|
|
BOOST_CHECK(t1 == t2);
|
|
t1.__isset.im_optional = true;
|
|
BOOST_CHECK(t1 != t2);
|
|
t2.__isset.im_optional = true;
|
|
BOOST_CHECK(t1 == t2);
|
|
t1.im_optional = 10;
|
|
BOOST_CHECK(t1 != t2);
|
|
t2.__isset.im_optional = false;
|
|
BOOST_CHECK(t1 != t2);
|
|
}
|
|
|
|
BOOST_AUTO_TEST_CASE(test_optional_required_13) {
|
|
OptionalDefault t1, t2;
|
|
|
|
BOOST_CHECK(t1.__isset.opt_int == true);
|
|
BOOST_CHECK(t1.__isset.opt_str == true);
|
|
BOOST_CHECK(t1.opt_int == t2.opt_int);
|
|
BOOST_CHECK(t1.opt_str == t2.opt_str);
|
|
|
|
write_to_read(t1, t2);
|
|
BOOST_CHECK(t2.__isset.opt_int == true);
|
|
BOOST_CHECK(t2.__isset.opt_str == true);
|
|
BOOST_CHECK(t1.opt_int == t2.opt_int);
|
|
BOOST_CHECK(t1.opt_str == t2.opt_str);
|
|
|
|
const std::string expected_result(
|
|
"OptionalDefault {\n"
|
|
" 01: opt_int (i16) = 1234,\n"
|
|
" 02: opt_str (string) = \"default\",\n"
|
|
"}");
|
|
const std::string result(apache::thrift::ThriftDebugString(t2));
|
|
|
|
BOOST_CHECK_MESSAGE(!expected_result.compare(result),
|
|
"Expected:\n" << expected_result << "\nGotten:\n" << result);
|
|
}
|