void HttpHeaders::set100ContinueExpected(HttpMessage& message, bool set) {
if (set) {
message.setHeader(Names::EXPECT, Values::CONTINUE);
}
else {
message.removeHeader(Names::EXPECT);
}
}
void HttpHeaders::setKeepAlive(HttpMessage& message, bool keepAlive) {
if (message.getProtocolVersion().isKeepAliveDefault()) {
if (keepAlive) {
message.removeHeader(Names::CONNECTION);
}
else {
message.setHeader(Names::CONNECTION, Values::CLOSE);
}
}
else {
if (keepAlive) {
message.setHeader(Names::CONNECTION, Values::KEEP_ALIVE);
}
else {
message.removeHeader(Names::CONNECTION);
}
}
}
/**
* Test header
*/
void testHeader()
{
const char* name1 = "yyy";
const char* value1 = "yyy-value";
const char* valueRef = NULL;
const char* name2 = "yyy1";
const char* value2 = "yyy-value1";
const char* value2a = "yyy-value2";
const char* value2b = "yyy-value3";
UtlBoolean rc;
HttpMessage *msg;
msg = new HttpMessage();
// H E A D E R
int fieldCount = msg->getCountHeaderFields();
CPPUNIT_ASSERT_EQUAL_MESSAGE("field count should be zero",
0, fieldCount);
// add header field name1
msg->addHeaderField(name1, value1);
// get overall header field count
fieldCount = msg->getCountHeaderFields();
CPPUNIT_ASSERT_EQUAL_MESSAGE("field count should be zero", 1,
fieldCount);
// get header field count for name1
fieldCount = msg->getCountHeaderFields(name1);
CPPUNIT_ASSERT_EQUAL_MESSAGE("field count should be zero", 1,
fieldCount);
// get header field by index
valueRef = msg->getHeaderValue(0);
CPPUNIT_ASSERT_MESSAGE("NULL field value", valueRef != NULL);
ASSERT_STR_EQUAL_MESSAGE("incorrect field value", value1,
valueRef);
// get header field by index and name
valueRef = msg->getHeaderValue(0, name1);
CPPUNIT_ASSERT_MESSAGE("NULL field value", valueRef != NULL);
ASSERT_STR_EQUAL_MESSAGE("incorrect field value", value1,
valueRef);
// add header field name2
msg->addHeaderField(name2, value2);
// get header field by name and index
valueRef = msg->getHeaderValue(0, name2);
CPPUNIT_ASSERT_MESSAGE("NULL field value", valueRef != NULL);
ASSERT_STR_EQUAL_MESSAGE("incorrect field value", value2,
valueRef);
// add second header field name2
msg->addHeaderField(name2, value2);
// get header field by name and index
valueRef = msg->getHeaderValue(0, name2);
CPPUNIT_ASSERT_MESSAGE("NULL field value", valueRef != NULL);
ASSERT_STR_EQUAL_MESSAGE("incorrect field value", value2,
valueRef);
// set second header field name2
msg->setHeaderValue(name2, value2b, 1);
// get header field by name and index
valueRef = msg->getHeaderValue(0, name2);
CPPUNIT_ASSERT_MESSAGE("NULL field value", valueRef != NULL);
ASSERT_STR_EQUAL_MESSAGE("incorrect field value", value2,
valueRef);
// insert header field name2 as a second header with name2
msg->insertHeaderField(name2, value2a, 2);
// get header field by name and index
valueRef = msg->getHeaderValue(0, name2);
CPPUNIT_ASSERT_MESSAGE("NULL field value", valueRef != NULL);
ASSERT_STR_EQUAL_MESSAGE("incorrect field value", value2,
valueRef);
// get overall header field count
fieldCount = msg->getCountHeaderFields();
CPPUNIT_ASSERT_EQUAL_MESSAGE("field count should be 4", 4,
fieldCount);
// get name1 header field count
fieldCount = msg->getCountHeaderFields(name1);
CPPUNIT_ASSERT_EQUAL_MESSAGE("field count should be 1", 1,
fieldCount);
// get name2 header field count
fieldCount = msg->getCountHeaderFields(name2);
CPPUNIT_ASSERT_EQUAL_MESSAGE("field count should be 3", 3,
fieldCount);
// get header field by name and index
valueRef = msg->getHeaderValue(0, name2);
CPPUNIT_ASSERT_MESSAGE("NULL field value", valueRef != NULL);
ASSERT_STR_EQUAL_MESSAGE("incorrect field value", value2,
valueRef);
// get header field by name and index
valueRef = msg->getHeaderValue(1, name2);
CPPUNIT_ASSERT_MESSAGE("NULL field value", valueRef != NULL);
ASSERT_STR_EQUAL_MESSAGE("incorrect field value", value2a,
valueRef);
// get header field by index
valueRef = msg->getHeaderValue(3);
CPPUNIT_ASSERT_MESSAGE("NULL field value", valueRef != NULL);
ASSERT_STR_EQUAL_MESSAGE("incorrect field value", value2b,
valueRef);
// remove non-existing header field
rc = msg->removeHeader("non-exist", 36);
CPPUNIT_ASSERT_MESSAGE("incorrect return code", rc == FALSE);
// get overall header field count
fieldCount = msg->getCountHeaderFields();
CPPUNIT_ASSERT_EQUAL_MESSAGE("field count should be 4", 4,
fieldCount);
// remove header field name1
rc = msg->removeHeader(name1, 0);
CPPUNIT_ASSERT_MESSAGE("incorrect return code", rc == TRUE);
// get header field by index
valueRef = msg->getHeaderValue(0);
CPPUNIT_ASSERT_MESSAGE("NULL field value", valueRef != NULL);
ASSERT_STR_EQUAL_MESSAGE("incorrect field value", value2,
valueRef);
// get name1 header field count
fieldCount = msg->getCountHeaderFields(name1);
CPPUNIT_ASSERT_EQUAL_MESSAGE("field count should be 0", 0,
fieldCount);
// remove second header field name2
rc = msg->removeHeader(name2, 1);
CPPUNIT_ASSERT_MESSAGE("incorrect return code", rc == TRUE);
// get overall header field count
fieldCount = msg->getCountHeaderFields();
CPPUNIT_ASSERT_EQUAL_MESSAGE("field count should be 2", 2,
fieldCount);
// get name2 header field count
fieldCount = msg->getCountHeaderFields(name2);
CPPUNIT_ASSERT_EQUAL_MESSAGE("field count should be 2", 2,
fieldCount);
// get header field by index
valueRef = msg->getHeaderValue(0);
CPPUNIT_ASSERT_MESSAGE("NULL field value", valueRef != NULL);
ASSERT_STR_EQUAL_MESSAGE("incorrect field value", value2,
valueRef);
// get header field by name and index
valueRef = msg->getHeaderValue(1, name2);
CPPUNIT_ASSERT_MESSAGE("NULL field value", valueRef != NULL);
ASSERT_STR_EQUAL_MESSAGE("incorrect field value", value2b,
valueRef);
// get non-exist header field by name and index
valueRef = msg->getHeaderValue(2, name2);
CPPUNIT_ASSERT_MESSAGE("non-NULL field value", valueRef == NULL);
// remove header field name2
rc = msg->removeHeader(name2, 1);
CPPUNIT_ASSERT_MESSAGE("incorrect return code", rc == TRUE);
// get overall header field count
fieldCount = msg->getCountHeaderFields();
CPPUNIT_ASSERT_EQUAL_MESSAGE("field count should be 1", 1,
fieldCount);
// get header field by index
valueRef = msg->getHeaderValue(0);
CPPUNIT_ASSERT_MESSAGE("NULL field value", valueRef != NULL);
ASSERT_STR_EQUAL_MESSAGE("incorrect field value", value2,
valueRef);
// remove header field name2
rc = msg->removeHeader(name2, 0);
CPPUNIT_ASSERT_MESSAGE("incorrect return code", rc == TRUE);
// get overall header field count
fieldCount = msg->getCountHeaderFields();
CPPUNIT_ASSERT_EQUAL_MESSAGE("field count should be 0", 0,
fieldCount);
delete msg;
}