// Copyright 2015 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "mojo/fetcher/data_fetcher.h"

#include "base/auto_reset.h"
#include "base/bind.h"
#include "base/logging.h"
#include "base/macros.h"
#include "base/memory/scoped_ptr.h"
#include "base/message_loop/message_loop.h"
#include "base/run_loop.h"
#include "mojo/runner/context.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "third_party/mojo/src/mojo/public/cpp/system/data_pipe.h"

namespace mojo {
namespace fetcher {
namespace {

class FetchCallbackHelper {
 public:
  FetchCallbackHelper() : run_loop_(nullptr) {}
  ~FetchCallbackHelper() {}

  shell::Fetcher::FetchCallback GetCallback() {
    return base::Bind(&FetchCallbackHelper::CallbackHandler,
                      base::Unretained(this));
  }

  void WaitForCallback() {
    base::RunLoop run_loop;
    base::AutoReset<base::RunLoop*> auto_reset(&run_loop_, &run_loop);
    run_loop.Run();
  }

  shell::Fetcher* fetcher() const { return fetcher_.get(); }

 private:
  void CallbackHandler(scoped_ptr<shell::Fetcher> fetcher) {
    fetcher_ = fetcher.Pass();
    if (run_loop_)
      run_loop_->Quit();
  }

  // If it is not null, it points to a stack-allocated base::RunLoop instance in
  // WaitForCallback().
  base::RunLoop* run_loop_;
  scoped_ptr<shell::Fetcher> fetcher_;
  DISALLOW_COPY_AND_ASSIGN(FetchCallbackHelper);
};

class DataFetcherTest : public testing::Test {
 public:
  DataFetcherTest() {}
  ~DataFetcherTest() override {}

 protected:
  // Overridden from testing::Test:
  void SetUp() override { runner::Context::EnsureEmbedderIsInitialized(); }

  void TestFetchURL(const std::string& url,
                    uint32_t expected_status_code,
                    const std::string& expected_mime_type,
                    const std::string& expected_body) {
    FetchCallbackHelper helper;
    DataFetcher::Start(GURL(url), helper.GetCallback());
    helper.WaitForCallback();

    ASSERT_TRUE(helper.fetcher());
    URLResponsePtr response = helper.fetcher()->AsURLResponse(nullptr, 0);
    ASSERT_TRUE(response);
    EXPECT_EQ(url, response->url);
    EXPECT_EQ(expected_status_code, response->status_code);

    if (expected_status_code != 200)
      return;

    ASSERT_TRUE(response->body.is_valid());
    EXPECT_EQ(expected_mime_type, response->mime_type);

    uint32_t num_bytes = 0;
    Handle body_handle = response->body.release();
    MojoResult result = MojoReadData(body_handle.value(), nullptr, &num_bytes,
                                     MOJO_READ_DATA_FLAG_QUERY);
    ASSERT_EQ(MOJO_RESULT_OK, result);

    scoped_ptr<char[]> body(new char[num_bytes]);
    result = MojoReadData(body_handle.value(), body.get(), &num_bytes,
                          MOJO_READ_DATA_FLAG_ALL_OR_NONE);
    ASSERT_EQ(MOJO_RESULT_OK, result);
    EXPECT_EQ(expected_body, std::string(body.get(), num_bytes));
  }

 private:
  base::MessageLoop loop_;

  DISALLOW_COPY_AND_ASSIGN(DataFetcherTest);
};

TEST_F(DataFetcherTest, BasicSuccess) {
  TestFetchURL("data:text/html,Hello world", 200, "text/html", "Hello world");
}

TEST_F(DataFetcherTest, BasicFailure) {
  TestFetchURL("about:blank", 400, std::string(), std::string());
  TestFetchURL("data:;base64,aGVs_-_-", 400, std::string(), std::string());
}

}  // namespace
}  // namespace fetcher
}  // namespace mojo