1 /*
2  *
3  * Copyright 2017 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 
19 #include "src/core/lib/transport/status_metadata.h"
20 
21 #include <gtest/gtest.h>
22 
23 #include "src/core/lib/transport/static_metadata.h"
24 #include "test/core/util/test_config.h"
25 
26 namespace {
27 
TEST(GetStatusCodeFromMetadata,OK)28 TEST(GetStatusCodeFromMetadata, OK) {
29   EXPECT_EQ(GRPC_STATUS_OK,
30             grpc_get_status_code_from_metadata(GRPC_MDELEM_GRPC_STATUS_0));
31 }
32 
TEST(GetStatusCodeFromMetadata,CANCELLED)33 TEST(GetStatusCodeFromMetadata, CANCELLED) {
34   EXPECT_EQ(GRPC_STATUS_CANCELLED,
35             grpc_get_status_code_from_metadata(GRPC_MDELEM_GRPC_STATUS_1));
36 }
37 
TEST(GetStatusCodeFromMetadata,UNKNOWN)38 TEST(GetStatusCodeFromMetadata, UNKNOWN) {
39   EXPECT_EQ(GRPC_STATUS_UNKNOWN,
40             grpc_get_status_code_from_metadata(GRPC_MDELEM_GRPC_STATUS_2));
41 }
42 
TEST(GetStatusCodeFromMetadata,Other)43 TEST(GetStatusCodeFromMetadata, Other) {
44   grpc_mdelem status_md = grpc_mdelem_from_slices(
45       GRPC_MDSTR_GRPC_STATUS, grpc_slice_from_static_string("10"));
46   EXPECT_EQ(GRPC_STATUS_ABORTED, grpc_get_status_code_from_metadata(status_md));
47   GRPC_MDELEM_UNREF(status_md);
48 }
49 
TEST(GetStatusCodeFromMetadata,Unparseable)50 TEST(GetStatusCodeFromMetadata, Unparseable) {
51   grpc_mdelem status_md = grpc_mdelem_from_slices(
52       GRPC_MDSTR_GRPC_STATUS, grpc_slice_from_static_string("NaN"));
53   EXPECT_EQ(GRPC_STATUS_UNKNOWN, grpc_get_status_code_from_metadata(status_md));
54   GRPC_MDELEM_UNREF(status_md);
55 }
56 
57 }  // namespace
58 
main(int argc,char ** argv)59 int main(int argc, char** argv) {
60   grpc::testing::TestEnvironment env(argc, argv);
61   ::testing::InitGoogleTest(&argc, argv);
62   grpc_init();
63   int ret = RUN_ALL_TESTS();
64   grpc_shutdown();
65   return ret;
66 }
67