// Copyright 2014 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 "platform/TracedValue.h"

#include "base/json/json_reader.h"
#include "base/values.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace blink {

scoped_ptr<base::Value> parseTracedValue(PassRefPtr<TracedValue> value)
{
    base::JSONReader reader;
    CString utf8 = value->toString().utf8();
    return reader.Read(utf8.data());
}

TEST(TracedValueTest, FlatDictionary)
{
    RefPtr<TracedValue> value = TracedValue::create();
    value->setInteger("int", 2014);
    value->setDouble("double", 0.0);
    value->setBoolean("bool", true);
    value->setString("string", "string");

    scoped_ptr<base::Value> parsed = parseTracedValue(value);
    base::DictionaryValue* dictionary;
    ASSERT_TRUE(parsed->GetAsDictionary(&dictionary));
    int intValue;
    EXPECT_TRUE(dictionary->GetInteger("int", &intValue));
    EXPECT_EQ(2014, intValue);
    double doubleValue;
    EXPECT_TRUE(dictionary->GetDouble("double", &doubleValue));
    EXPECT_EQ(0.0, doubleValue);
    std::string stringValue;
    EXPECT_TRUE(dictionary->GetString("string", &stringValue));
    EXPECT_EQ("string", stringValue);
}

TEST(TracedValueTest, Hierarchy)
{
    RefPtr<TracedValue> value = TracedValue::create();
    value->setInteger("i0", 2014);
    value->beginDictionary("dict1");
    value->setInteger("i1", 2014);
    value->beginDictionary("dict2");
    value->setBoolean("b2", false);
    value->endDictionary();
    value->setString("s1", "foo");
    value->endDictionary();
    value->setDouble("d0", 0.0);
    value->setBoolean("b0", true);
    value->beginArray("a1");
    value->pushInteger(1);
    value->pushBoolean(true);
    value->beginDictionary();
    value->setInteger("i2", 3);
    value->endDictionary();
    value->endArray();
    value->setString("s0", "foo");

    scoped_ptr<base::Value> parsed = parseTracedValue(value);
    base::DictionaryValue* dictionary;
    ASSERT_TRUE(parsed->GetAsDictionary(&dictionary));
    int i0;
    EXPECT_TRUE(dictionary->GetInteger("i0", &i0));
    EXPECT_EQ(2014, i0);
    int i1;
    EXPECT_TRUE(dictionary->GetInteger("dict1.i1", &i1));
    EXPECT_EQ(2014, i1);
    bool b2;
    EXPECT_TRUE(dictionary->GetBoolean("dict1.dict2.b2", &b2));
    EXPECT_FALSE(b2);
    std::string s1;
    EXPECT_TRUE(dictionary->GetString("dict1.s1", &s1));
    EXPECT_EQ("foo", s1);
    double d0;
    EXPECT_TRUE(dictionary->GetDouble("d0", &d0));
    EXPECT_EQ(0.0, d0);
    bool b0;
    EXPECT_TRUE(dictionary->GetBoolean("b0", &b0));
    EXPECT_TRUE(b0);
    base::ListValue* a1;
    EXPECT_TRUE(dictionary->GetList("a1", &a1));
    int a1i0;
    EXPECT_TRUE(a1->GetInteger(0, &a1i0));
    EXPECT_EQ(1, a1i0);
    bool a1b1;
    EXPECT_TRUE(a1->GetBoolean(1, &a1b1));
    EXPECT_TRUE(a1b1);
    base::DictionaryValue* a1d2;
    EXPECT_TRUE(a1->GetDictionary(2, &a1d2));
    int i2;
    EXPECT_TRUE(a1d2->GetInteger("i2", &i2));
    EXPECT_EQ(3, i2);
    std::string s0;
    EXPECT_TRUE(dictionary->GetString("s0", &s0));
    EXPECT_EQ("foo", s0);
}

TEST(TracedValueTest, Escape)
{
    RefPtr<TracedValue> value = TracedValue::create();
    value->setString("s0", "value0\\");
    value->setString("s1", "value\n1");
    value->setString("s2", "\"value2\"");
    value->setString("s3\\", "value3");
    value->setString("\"s4\"", "value4");

    scoped_ptr<base::Value> parsed = parseTracedValue(value);
    base::DictionaryValue* dictionary;
    ASSERT_TRUE(parsed->GetAsDictionary(&dictionary));
    std::string s0;
    EXPECT_TRUE(dictionary->GetString("s0", &s0));
    EXPECT_EQ("value0\\", s0);
    std::string s1;
    EXPECT_TRUE(dictionary->GetString("s1", &s1));
    EXPECT_EQ("value\n1", s1);
    std::string s2;
    EXPECT_TRUE(dictionary->GetString("s2", &s2));
    EXPECT_EQ("\"value2\"", s2);
    std::string s3;
    EXPECT_TRUE(dictionary->GetString("s3\\", &s3));
    EXPECT_EQ("value3", s3);
    std::string s4;
    EXPECT_TRUE(dictionary->GetString("\"s4\"", &s4));
    EXPECT_EQ("value4", s4);
}

} // namespace blink