// Copyright (c) 2012 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 "ui/views/controls/table/table_utils.h"

#include "base/strings/string_number_conversions.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "ui/gfx/font_list.h"
#include "ui/views/controls/table/test_table_model.h"

using ui::TableColumn;
using ui::TableModel;

namespace views {

namespace {

std::string IntVectorToString(const std::vector<int>& values) {
  std::string result;
  for (size_t i = 0; i < values.size(); ++i) {
    if (i != 0)
      result += ",";
    result += base::IntToString(values[i]);
  }
  return result;
}

ui::TableColumn CreateTableColumnWithWidth(int width) {
  ui::TableColumn column;
  column.width = width;
  return column;
}

}  // namespace

// Verifies columns with a specified width is honored.
TEST(TableUtilsTest, SetWidthHonored) {
  TestTableModel model(4);
  std::vector<TableColumn> columns;
  columns.push_back(CreateTableColumnWithWidth(20));
  columns.push_back(CreateTableColumnWithWidth(30));
  gfx::FontList font_list;
  std::vector<int> result(CalculateTableColumnSizes(
      100, 0, font_list, font_list, 0, 0, columns, &model));
  EXPECT_EQ("20,30", IntVectorToString(result));

  // Same with some padding, it should be ignored.
  result = CalculateTableColumnSizes(
      100, 0, font_list, font_list, 2, 0, columns, &model);
  EXPECT_EQ("20,30", IntVectorToString(result));

  // Same with not enough space, it shouldn't matter.
  result = CalculateTableColumnSizes(
      10, 0, font_list, font_list, 2, 0, columns, &model);
  EXPECT_EQ("20,30", IntVectorToString(result));
}

// Verifies if no size is specified the last column gets all the available
// space.
TEST(TableUtilsTest, LastColumnGetsAllSpace) {
  TestTableModel model(4);
  std::vector<TableColumn> columns;
  columns.push_back(ui::TableColumn());
  columns.push_back(ui::TableColumn());
  gfx::FontList font_list;
  std::vector<int> result(CalculateTableColumnSizes(
      500, 0, font_list, font_list, 0, 0, columns, &model));
  EXPECT_NE(0, result[0]);
  EXPECT_GE(result[1],
            WidthForContent(font_list, font_list, 0, 0, columns[1], &model));
  EXPECT_EQ(500, result[0] + result[1]);
}

// Verifies a single column with a percent=1 is resized correctly.
TEST(TableUtilsTest, SingleResizableColumn) {
  TestTableModel model(4);
  std::vector<TableColumn> columns;
  columns.push_back(ui::TableColumn());
  columns.push_back(ui::TableColumn());
  columns.push_back(ui::TableColumn());
  columns[2].percent = 1.0f;
  gfx::FontList font_list;
  std::vector<int> result(CalculateTableColumnSizes(
      500, 0, font_list, font_list, 0, 0, columns, &model));
  EXPECT_EQ(result[0],
            WidthForContent(font_list, font_list, 0, 0, columns[0], &model));
  EXPECT_EQ(result[1],
            WidthForContent(font_list, font_list, 0, 0, columns[1], &model));
  EXPECT_EQ(500 - result[0] - result[1], result[2]);

  // The same with a slightly larger width passed in.
  result = CalculateTableColumnSizes(
      1000, 0, font_list, font_list, 0, 0, columns, &model);
  EXPECT_EQ(result[0],
            WidthForContent(font_list, font_list, 0, 0, columns[0], &model));
  EXPECT_EQ(result[1],
            WidthForContent(font_list, font_list, 0, 0, columns[1], &model));
  EXPECT_EQ(1000 - result[0] - result[1], result[2]);

  // Verify padding for the first column is honored.
  result = CalculateTableColumnSizes(
      1000, 10, font_list, font_list, 0, 0, columns, &model);
  EXPECT_EQ(result[0],
            WidthForContent(font_list, font_list, 0, 0, columns[0], &model)
                + 10);
  EXPECT_EQ(result[1],
            WidthForContent(font_list, font_list, 0, 0, columns[1], &model));
  EXPECT_EQ(1000 - result[0] - result[1], result[2]);

  // Just enough space to show the first two columns. Should force last column
  // to min size.
  result = CalculateTableColumnSizes(
      1000, 0, font_list, font_list, 0, 0, columns, &model);
  result = CalculateTableColumnSizes(
      result[0] + result[1], 0, font_list, font_list, 0, 0, columns, &model);
  EXPECT_EQ(result[0],
            WidthForContent(font_list, font_list, 0, 0, columns[0], &model));
  EXPECT_EQ(result[1],
            WidthForContent(font_list, font_list, 0, 0, columns[1], &model));
  EXPECT_EQ(kUnspecifiedColumnWidth, result[2]);
}

}  // namespace views