TEST(RawResourceTest, RevalidationSucceededForResourceWithoutBody)
{
Resource* resource = RawResource::create(ResourceRequest("data:text/html,"), Resource::Raw);
ResourceResponse response;
response.setHTTPStatusCode(200);
resource->responseReceived(response, nullptr);
resource->finish();
memoryCache()->add(resource);
// Simulate a successful revalidation.
resource->setRevalidatingRequest(ResourceRequest("data:text/html,"));
Persistent<DummyClient> client = new DummyClient;
resource->addClient(client);
ResourceResponse revalidatingResponse;
revalidatingResponse.setHTTPStatusCode(304);
resource->responseReceived(revalidatingResponse, nullptr);
EXPECT_FALSE(resource->isCacheValidator());
EXPECT_EQ(200, resource->response().httpStatusCode());
EXPECT_EQ(nullptr, resource->resourceBuffer());
EXPECT_EQ(memoryCache()->resourceForURL(KURL(ParsedURLString, "data:text/html,")), resource);
memoryCache()->remove(resource);
resource->removeClient(client);
EXPECT_FALSE(resource->hasClientsOrObservers());
EXPECT_FALSE(client->called());
EXPECT_EQ(0u, client->data().size());
}
TEST(RawResourceTest, AddClientDuringCallback)
{
Resource* raw = RawResource::create(ResourceRequest("data:text/html,"), Resource::Raw);
// Create a non-null response.
ResourceResponse response = raw->response();
response.setURL(KURL(ParsedURLString, "http://600.613/"));
raw->setResponse(response);
raw->finish();
EXPECT_FALSE(raw->response().isNull());
Persistent<DummyClient> dummyClient = new DummyClient();
Persistent<AddingClient> addingClient = new AddingClient(dummyClient.get(), raw);
raw->addClient(addingClient);
testing::runPendingTasks();
raw->removeClient(addingClient);
EXPECT_FALSE(dummyClient->called());
EXPECT_FALSE(raw->hasClientsOrObservers());
}
TEST(RawResourceTest, RedirectDuringRevalidation)
{
Resource* resource = RawResource::create(ResourceRequest("https://example.com/1"), Resource::Raw);
ResourceResponse response;
response.setURL(KURL(ParsedURLString, "https://example.com/1"));
response.setHTTPStatusCode(200);
resource->responseReceived(response, nullptr);
const char data[5] = "abcd";
resource->appendData(data, 4);
resource->finish();
memoryCache()->add(resource);
EXPECT_FALSE(resource->isCacheValidator());
EXPECT_EQ("https://example.com/1", resource->resourceRequest().url().getString());
EXPECT_EQ("https://example.com/1", resource->lastResourceRequest().url().getString());
// Simulate a revalidation.
resource->setRevalidatingRequest(ResourceRequest("https://example.com/1"));
EXPECT_TRUE(resource->isCacheValidator());
EXPECT_EQ("https://example.com/1", resource->resourceRequest().url().getString());
EXPECT_EQ("https://example.com/1", resource->lastResourceRequest().url().getString());
Persistent<DummyClient> client = new DummyClient;
resource->addClient(client);
// The revalidating request is redirected.
ResourceResponse redirectResponse;
redirectResponse.setURL(KURL(ParsedURLString, "https://example.com/1"));
redirectResponse.setHTTPHeaderField("location", "https://example.com/2");
redirectResponse.setHTTPStatusCode(308);
ResourceRequest redirectedRevalidatingRequest("https://example.com/2");
resource->willFollowRedirect(redirectedRevalidatingRequest, redirectResponse);
EXPECT_FALSE(resource->isCacheValidator());
EXPECT_EQ("https://example.com/1", resource->resourceRequest().url().getString());
EXPECT_EQ("https://example.com/2", resource->lastResourceRequest().url().getString());
// The final response is received.
ResourceResponse revalidatingResponse;
revalidatingResponse.setURL(KURL(ParsedURLString, "https://example.com/2"));
revalidatingResponse.setHTTPStatusCode(200);
resource->responseReceived(revalidatingResponse, nullptr);
const char data2[4] = "xyz";
resource->appendData(data2, 3);
resource->finish();
EXPECT_FALSE(resource->isCacheValidator());
EXPECT_EQ("https://example.com/1", resource->resourceRequest().url().getString());
EXPECT_EQ("https://example.com/2", resource->lastResourceRequest().url().getString());
EXPECT_FALSE(resource->isCacheValidator());
EXPECT_EQ(200, resource->response().httpStatusCode());
EXPECT_EQ(3u, resource->resourceBuffer()->size());
EXPECT_EQ(memoryCache()->resourceForURL(KURL(ParsedURLString, "https://example.com/1")), resource);
EXPECT_TRUE(client->called());
EXPECT_EQ(1, client->numberOfRedirectsReceived());
EXPECT_EQ("xyz", String(client->data().data(), client->data().size()));
// Test the case where a client is added after revalidation is completed.
Persistent<DummyClient> client2 = new DummyClient;
resource->addClient(client2);
// Because RawResourceClient is added asynchronously,
// |runPendingTasks()| is called to make |client2| to be notified.
testing::runPendingTasks();
EXPECT_TRUE(client2->called());
EXPECT_EQ(1, client2->numberOfRedirectsReceived());
EXPECT_EQ("xyz", String(client2->data().data(), client2->data().size()));
memoryCache()->remove(resource);
resource->removeClient(client);
resource->removeClient(client2);
EXPECT_FALSE(resource->hasClientsOrObservers());
}
TEST(RawResourceTest, RevalidationSucceededUpdateHeaders)
{
Resource* resource = RawResource::create(ResourceRequest("data:text/html,"), Resource::Raw);
ResourceResponse response;
response.setHTTPStatusCode(200);
response.addHTTPHeaderField("keep-alive", "keep-alive value");
response.addHTTPHeaderField("expires", "expires value");
response.addHTTPHeaderField("last-modified", "last-modified value");
response.addHTTPHeaderField("proxy-authenticate", "proxy-authenticate value");
response.addHTTPHeaderField("proxy-connection", "proxy-connection value");
response.addHTTPHeaderField("x-custom", "custom value");
resource->responseReceived(response, nullptr);
resource->finish();
memoryCache()->add(resource);
// Simulate a successful revalidation.
resource->setRevalidatingRequest(ResourceRequest("data:text/html,"));
// Validate that these headers pre-update.
EXPECT_EQ("keep-alive value", resource->response().httpHeaderField("keep-alive"));
EXPECT_EQ("expires value", resource->response().httpHeaderField("expires"));
EXPECT_EQ("last-modified value", resource->response().httpHeaderField("last-modified"));
EXPECT_EQ("proxy-authenticate value", resource->response().httpHeaderField("proxy-authenticate"));
EXPECT_EQ("proxy-authenticate value", resource->response().httpHeaderField("proxy-authenticate"));
EXPECT_EQ("proxy-connection value", resource->response().httpHeaderField("proxy-connection"));
EXPECT_EQ("custom value", resource->response().httpHeaderField("x-custom"));
Persistent<DummyClient> client = new DummyClient;
resource->addClient(client.get());
// Perform a revalidation step.
ResourceResponse revalidatingResponse;
revalidatingResponse.setHTTPStatusCode(304);
// Headers that aren't copied with an 304 code.
revalidatingResponse.addHTTPHeaderField("keep-alive", "garbage");
revalidatingResponse.addHTTPHeaderField("expires", "garbage");
revalidatingResponse.addHTTPHeaderField("last-modified", "garbage");
revalidatingResponse.addHTTPHeaderField("proxy-authenticate", "garbage");
revalidatingResponse.addHTTPHeaderField("proxy-connection", "garbage");
// Header that is updated with 304 code.
revalidatingResponse.addHTTPHeaderField("x-custom", "updated");
resource->responseReceived(revalidatingResponse, nullptr);
// Validate the original response.
EXPECT_EQ(200, resource->response().httpStatusCode());
// Validate that these headers are not updated.
EXPECT_EQ("keep-alive value", resource->response().httpHeaderField("keep-alive"));
EXPECT_EQ("expires value", resource->response().httpHeaderField("expires"));
EXPECT_EQ("last-modified value", resource->response().httpHeaderField("last-modified"));
EXPECT_EQ("proxy-authenticate value", resource->response().httpHeaderField("proxy-authenticate"));
EXPECT_EQ("proxy-authenticate value", resource->response().httpHeaderField("proxy-authenticate"));
EXPECT_EQ("proxy-connection value", resource->response().httpHeaderField("proxy-connection"));
EXPECT_EQ("updated", resource->response().httpHeaderField("x-custom"));
memoryCache()->remove(resource);
resource->removeClient(client);
EXPECT_FALSE(resource->hasClientsOrObservers());
EXPECT_FALSE(client->called());
EXPECT_EQ(0u, client->data().size());
}