update_engine: add new headers
Adding new headers requested by OEM partners. Add brackets to if
statements for consistent across UE codebase.
Bug: 355109876
Test: th
Change-Id: I2adc0866587d6b1f13a3dcba15396b900f45b74e
diff --git a/common/http_fetcher_unittest.cc b/common/http_fetcher_unittest.cc
index 06f3e15..b229660 100644
--- a/common/http_fetcher_unittest.cc
+++ b/common/http_fetcher_unittest.cc
@@ -548,6 +548,8 @@
fetcher->SetHeader("User-Agent", "MyTest");
fetcher->SetHeader("user-agent", "Override that header");
fetcher->SetHeader("Authorization", "Basic user:passwd");
+ fetcher->SetHeader("Cache-Control", "testControl");
+ fetcher->SetHeader("Connection", "testConnection");
// Invalid headers.
fetcher->SetHeader("X-Foo", "Invalid\nHeader\nIgnored");
@@ -571,6 +573,8 @@
delegate.data.find("user-agent: Override that header\r\n"));
EXPECT_NE(string::npos,
delegate.data.find("Authorization: Basic user:passwd\r\n"));
+ EXPECT_NE(string::npos, delegate.data.find("Cache-Control: testControl\r\n"));
+ EXPECT_NE(string::npos, delegate.data.find("Connection: testConnection\r\n"));
EXPECT_EQ(string::npos, delegate.data.find("\nAccept:"));
EXPECT_EQ(string::npos, delegate.data.find("X-Foo: Invalid"));