<?php
/**
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either express or implied.  See the License for the
 * specific language governing permissions and limitations
 * under the License.
 */

class IoTest extends BaseTest {
  public function testParseHttpResponseBody() {
    $io = new apiCurlIO();

    $rawHeaders = "HTTP/1.1 200 OK\r\n"
        . "Expires: Sun, 22 Jan 2012 09:00:56 GMT\r\n"
        . "Date: Sun, 22 Jan 2012 09:00:56 GMT\r\n"
        . "Content-Type: application/json; charset=UTF-8\r\n";
    $size = strlen($rawHeaders);
    $rawBody = "{}";

    $rawResponse = "$rawHeaders\r\n$rawBody";
    list($headers, $body) = $io->parseHttpResponseBody($rawResponse, $size);
    $this->assertEquals(3, sizeof($headers));
    $this->assertEquals(array(), json_decode($body, true));


    // Test empty bodies.
    $rawResponse = $rawHeaders . "\r\n";
    list($headers, $body) = $io->parseHttpResponseBody($rawResponse, $size);
    $this->assertEquals(3, sizeof($headers));
    $this->assertEquals(null, json_decode($body, true));

    // Test transforms from proxies.
    $rawHeaders = "HTTP/1.1 200 OK\r\nContent-Type: application/json\r\n";
    $size = strlen($rawHeaders);
    $rawBody = "{}";

    $rawResponse = apiCurlIO::CONNECTION_ESTABLISHED
          . "$rawHeaders\r\n$rawBody";
    list($headers, $body) = $io->parseHttpResponseBody($rawResponse, $size);
    $this->assertEquals(1, sizeof($headers));
    $this->assertEquals(array(), json_decode($body, true));
  }

  public function testProcessEntityRequest() {
    $io = new apiCurlIO();
    $req = new apiHttpRequest("http://localhost.com");
    $req->setRequestMethod("POST");

    // Verify that the content-length is calculated.
    $req->setPostBody("{}");
    $io->processEntityRequest($req);
    $this->assertEquals(2, $req->getRequestHeader("content-length"));

    // Test an empty post body.
    $req->setPostBody("");
    $io->processEntityRequest($req);
    $this->assertEquals(0, $req->getRequestHeader("content-length"));

    // Test a null post body.
    $req->setPostBody(null);
    $io->processEntityRequest($req);
    $this->assertEquals(0, $req->getRequestHeader("content-length"));

    // Set an array in the postbody, and verify that it is url-encoded.
    $req->setPostBody(array("a" => "1", "b" => 2));
    $io->processEntityRequest($req);
    $this->assertEquals(7, $req->getRequestHeader("content-length"));
    $this->assertEquals(apiCurlIO::FORM_URLENCODED,
        $req->getRequestHeader("content-type"));
    $this->assertEquals("a=1&b=2", $req->getPostBody());

    // Verify that the content-type isn't reset.
    $payload = array("a" => "1", "b" => 2);
    $req->setPostBody($payload);
    $req->setRequestHeaders(array("content-type" => "multipart/form-data"));
    $io->processEntityRequest($req);
    $this->assertEquals("multipart/form-data",
        $req->getRequestHeader("content-type"));
    $this->assertEquals($payload, $req->getPostBody());
  }

  public function testCacheHit() {
    $io = new apiCurlIO();
    $url = "http://www.googleapis.com";
    // Create a cacheable request/response.
    // Should not be revalidated.
    $cacheReq = new apiHttpRequest($url, "GET");
    $cacheReq->setRequestHeaders(array(
      "Accept" => "*/*",
    ));
    $cacheReq->setResponseBody("{\"a\": \"foo\"}");
    $cacheReq->setResponseHttpCode(200);
    $cacheReq->setResponseHeaders(array(
      "Cache-Control" => "private",
      "ETag" => "\"this-is-an-etag\"",
      "Expires" => "Sun, 22 Jan 2022 09:00:56 GMT",
      "Date: Sun, 1 Jan 2012 09:00:56 GMT",
      "Content-Type" => "application/json; charset=UTF-8",
    ));

    // Populate the cache.
    $io->setCachedRequest($cacheReq);

    // Execute the same mock request, and expect a cache hit.
    $res = $io->makeRequest(new apiHttpRequest($url, "GET"));
    $this->assertEquals("{\"a\": \"foo\"}", $res->getResponseBody());
    $this->assertEquals(200, $res->getResponseHttpCode());
  }

  public function testAuthCache() {
    $io = new apiCurlIO();
    $url = "http://www.googleapis.com/protected/resource";

    // Create a cacheable request/response, but it should not be cached.
    $cacheReq = new apiHttpRequest($url, "GET");
    $cacheReq->setRequestHeaders(array(
      "Accept" => "*/*",
      "Authorization" => "Bearer Foo"
    ));
    $cacheReq->setResponseBody("{\"a\": \"foo\"}");
    $cacheReq->setResponseHttpCode(200);
    $cacheReq->setResponseHeaders(array(
      "Cache-Control" => "private",
      "ETag" => "\"this-is-an-etag\"",
      "Expires" => "Sun, 22 Jan 2022 09:00:56 GMT",
      "Date: Sun, 1 Jan 2012 09:00:56 GMT",
      "Content-Type" => "application/json; charset=UTF-8",
    ));

    $result = $io->setCachedRequest($cacheReq);
    $this->assertFalse($result);
  }
}