proxygen
HTTPTransactionSMTest.cpp
Go to the documentation of this file.
1 /*
2  * Copyright (c) 2015-present, Facebook, Inc.
3  * All rights reserved.
4  *
5  * This source code is licensed under the BSD-style license found in the
6  * LICENSE file in the root directory of this source tree. An additional grant
7  * of patent rights can be found in the PATENTS file in the same directory.
8  *
9  */
10 #include <folly/portability/GTest.h>
11 #include <proxygen/lib/http/session/HTTPTransactionEgressSM.h>
12 #include <proxygen/lib/http/session/HTTPTransactionIngressSM.h>
13 
14 using namespace proxygen;
15 
16 class EgressStateMachineFixture: public ::testing::Test {
17  public:
18  EgressStateMachineFixture(): instance_(
19  HTTPTransactionEgressSM::getNewInstance()) {}
20 
21  void follow(HTTPTransactionEgressSM::Event e) {
22  EXPECT_TRUE(HTTPTransactionEgressSM::transit(instance_, e));
23  }
24 
25  void fail(HTTPTransactionEgressSM::Event e) {
26  EXPECT_FALSE(HTTPTransactionEgressSM::transit(instance_, e));
27  }
28  private:
29  HTTPTransactionEgressSM::State instance_;
30 };
31 
32 class IngressStateMachineFixture: public ::testing::Test {
33  public:
34  IngressStateMachineFixture():
35  instance_(HTTPTransactionIngressSM::getNewInstance()) {}
36 
37  void follow(HTTPTransactionIngressSM::Event e) {
38  EXPECT_TRUE(HTTPTransactionIngressSM::transit(instance_, e));
39  }
40 
41  void fail(HTTPTransactionIngressSM::Event e) {
42  EXPECT_FALSE(HTTPTransactionIngressSM::transit(instance_, e));
43  }
44  private:
45  HTTPTransactionIngressSM::State instance_;
46 };
47 
48 // Egress tests
49 
50 TEST_F(EgressStateMachineFixture, BadEgressTransitions1) {
51  follow(HTTPTransactionEgressSM::Event::sendHeaders);
52  follow(HTTPTransactionEgressSM::Event::sendChunkHeader);
53  follow(HTTPTransactionEgressSM::Event::sendBody);
54  fail(HTTPTransactionEgressSM::Event::sendEOM);
55 }
56 
57 TEST_F(EgressStateMachineFixture, BadEgressTransitions2) {
58  fail(HTTPTransactionEgressSM::Event::sendBody);
59 }
60 
61 TEST_F(EgressStateMachineFixture, BadEgressTransitions3) {
62  follow(HTTPTransactionEgressSM::Event::sendHeaders);
63  follow(HTTPTransactionEgressSM::Event::sendBody);
64  follow(HTTPTransactionEgressSM::Event::sendBody);
65  follow(HTTPTransactionEgressSM::Event::sendEOM);
66  fail(HTTPTransactionEgressSM::Event::sendEOM);
67 }
68 
69 TEST_F(EgressStateMachineFixture, BadEgressTransitions4) {
70  follow(HTTPTransactionEgressSM::Event::sendHeaders);
71  follow(HTTPTransactionEgressSM::Event::sendChunkHeader);
72  follow(HTTPTransactionEgressSM::Event::sendBody);
73  follow(HTTPTransactionEgressSM::Event::sendChunkTerminator);
74  follow(HTTPTransactionEgressSM::Event::sendChunkHeader);
75  fail(HTTPTransactionEgressSM::Event::sendChunkTerminator);
76 }
77 
78 TEST_F(EgressStateMachineFixture, EgressChunkedTransitions) {
79  follow(HTTPTransactionEgressSM::Event::sendHeaders);
80 
81  follow(HTTPTransactionEgressSM::Event::sendChunkHeader);
82  follow(HTTPTransactionEgressSM::Event::sendBody);
83  follow(HTTPTransactionEgressSM::Event::sendChunkTerminator);
84 
85  follow(HTTPTransactionEgressSM::Event::sendChunkHeader);
86  follow(HTTPTransactionEgressSM::Event::sendBody);
87  follow(HTTPTransactionEgressSM::Event::sendChunkTerminator);
88 
89  follow(HTTPTransactionEgressSM::Event::sendChunkHeader);
90  follow(HTTPTransactionEgressSM::Event::sendBody);
91  follow(HTTPTransactionEgressSM::Event::sendChunkTerminator);
92 
93  follow(HTTPTransactionEgressSM::Event::sendTrailers);
94 
95  follow(HTTPTransactionEgressSM::Event::sendEOM);
96  follow(HTTPTransactionEgressSM::Event::eomFlushed);
97 }
98 
99 TEST_F(EgressStateMachineFixture, NormalEgressTransitions) {
100  follow(HTTPTransactionEgressSM::Event::sendHeaders);
101  follow(HTTPTransactionEgressSM::Event::sendBody);
102  follow(HTTPTransactionEgressSM::Event::sendBody);
103  follow(HTTPTransactionEgressSM::Event::sendBody);
104  follow(HTTPTransactionEgressSM::Event::sendBody);
105  follow(HTTPTransactionEgressSM::Event::sendEOM);
106 }
107 
108 TEST_F(EgressStateMachineFixture, NormalEgressTransitionsWithTrailers) {
109  follow(HTTPTransactionEgressSM::Event::sendHeaders);
110  follow(HTTPTransactionEgressSM::Event::sendBody);
111  follow(HTTPTransactionEgressSM::Event::sendBody);
112  follow(HTTPTransactionEgressSM::Event::sendBody);
113  follow(HTTPTransactionEgressSM::Event::sendBody);
114  follow(HTTPTransactionEgressSM::Event::sendTrailers);
115  follow(HTTPTransactionEgressSM::Event::sendEOM);
116  follow(HTTPTransactionEgressSM::Event::eomFlushed);
117 }
118 
119 TEST_F(EgressStateMachineFixture, WeirdEgressTransitions) {
120  follow(HTTPTransactionEgressSM::Event::sendHeaders);
121  follow(HTTPTransactionEgressSM::Event::sendTrailers);
122  follow(HTTPTransactionEgressSM::Event::sendEOM);
123  follow(HTTPTransactionEgressSM::Event::eomFlushed);
124 }
125 
126 // Ingress tests
127 
128 TEST_F(IngressStateMachineFixture, BadIngressTransitions1) {
129  follow(HTTPTransactionIngressSM::Event::onHeaders);
130  follow(HTTPTransactionIngressSM::Event::onChunkHeader);
131  follow(HTTPTransactionIngressSM::Event::onBody);
132  fail(HTTPTransactionIngressSM::Event::onEOM);
133 }
134 
135 TEST_F(IngressStateMachineFixture, BadIngressTransitions2) {
136  fail(HTTPTransactionIngressSM::Event::onBody);
137 }
138 
139 TEST_F(IngressStateMachineFixture, BadIngressTransitions3) {
140  follow(HTTPTransactionIngressSM::Event::onHeaders);
141  follow(HTTPTransactionIngressSM::Event::onBody);
142  follow(HTTPTransactionIngressSM::Event::onBody);
143  follow(HTTPTransactionIngressSM::Event::onEOM);
144  fail(HTTPTransactionIngressSM::Event::onEOM);
145 }
146 
147 TEST_F(IngressStateMachineFixture, BadIngressTransitions4) {
148  follow(HTTPTransactionIngressSM::Event::onHeaders);
149  follow(HTTPTransactionIngressSM::Event::onChunkHeader);
150  follow(HTTPTransactionIngressSM::Event::onBody);
151  follow(HTTPTransactionIngressSM::Event::onChunkComplete);
152  follow(HTTPTransactionIngressSM::Event::onChunkHeader);
153  fail(HTTPTransactionIngressSM::Event::onChunkComplete);
154 }
155 
156 TEST_F(IngressStateMachineFixture, IngressChunkedTransitions) {
157  follow(HTTPTransactionIngressSM::Event::onHeaders);
158 
159  follow(HTTPTransactionIngressSM::Event::onChunkHeader);
160  follow(HTTPTransactionIngressSM::Event::onBody);
161  follow(HTTPTransactionIngressSM::Event::onChunkComplete);
162 
163  follow(HTTPTransactionIngressSM::Event::onChunkHeader);
164  follow(HTTPTransactionIngressSM::Event::onBody);
165  follow(HTTPTransactionIngressSM::Event::onChunkComplete);
166 
167  follow(HTTPTransactionIngressSM::Event::onChunkHeader);
168  follow(HTTPTransactionIngressSM::Event::onBody);
169  follow(HTTPTransactionIngressSM::Event::onChunkComplete);
170 
171  follow(HTTPTransactionIngressSM::Event::onTrailers);
172  follow(HTTPTransactionIngressSM::Event::onEOM);
173 }
174 
175 TEST_F(IngressStateMachineFixture, NormalIngressTransitions) {
176  follow(HTTPTransactionIngressSM::Event::onHeaders);
177  follow(HTTPTransactionIngressSM::Event::onBody);
178  follow(HTTPTransactionIngressSM::Event::onBody);
179  follow(HTTPTransactionIngressSM::Event::onBody);
180  follow(HTTPTransactionIngressSM::Event::onBody);
181  follow(HTTPTransactionIngressSM::Event::onEOM);
182 }
183 
184 TEST_F(IngressStateMachineFixture, WeirdIngressTransitions) {
185  follow(HTTPTransactionIngressSM::Event::onHeaders);
186  follow(HTTPTransactionIngressSM::Event::onTrailers);
187  follow(HTTPTransactionIngressSM::Event::onEOM);
188 }
proxygen::StateMachine::State
typename T::State State
Definition: StateMachine.h:20
EgressStateMachineFixture::follow
void follow(HTTPTransactionEgressSM::Event e)
Definition: HTTPTransactionSMTest.cpp:21
IngressStateMachineFixture::fail
void fail(HTTPTransactionIngressSM::Event e)
Definition: HTTPTransactionSMTest.cpp:41
EgressStateMachineFixture::fail
void fail(HTTPTransactionEgressSM::Event e)
Definition: HTTPTransactionSMTest.cpp:25
proxygen::StateMachine::transit
static bool transit(State &state, Event event)
Definition: StateMachine.h:27
fail
void fail()
Definition: SafeAssertTest.cpp:27
EgressStateMachineFixture::instance_
HTTPTransactionEgressSM::State instance_
Definition: HTTPTransactionSMTest.cpp:29
IngressStateMachineFixture::instance_
HTTPTransactionIngressSM::State instance_
Definition: HTTPTransactionSMTest.cpp:45
proxygen::StateMachine::Event
typename T::Event Event
Definition: StateMachine.h:21
EXPECT_TRUE
#define EXPECT_TRUE(condition)
Definition: gtest.h:1859
proxygen::TEST_F
TEST_F(HeaderTableTests, IndexTranslation)
Definition: HeaderTableTests.cpp:54
EXPECT_FALSE
#define EXPECT_FALSE(condition)
Definition: gtest.h:1862
IngressStateMachineFixture::follow
void follow(HTTPTransactionIngressSM::Event e)
Definition: HTTPTransactionSMTest.cpp:37