components/autofill/core/browser/data_driven_test.cc

   1 // Copyright 2013 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "components/autofill/core/browser/data_driven_test.h"
   6
   7 #include "base/files/file_enumerator.h"
   8 #include "base/files/file_util.h"
   9 #include "base/strings/string_util.h"
  10 #include "testing/gtest/include/gtest/gtest.h"
  11
  12 namespace autofill {
  13 namespace {
  14
  15 // Reads |file| into |content|, and converts Windows line-endings to Unix ones.
  16 // Returns true on success.
  17 bool ReadFile(const base::FilePath& file, std::string* content) {
  18   if (!base::ReadFileToString(file, content))
  19     return false;
  20
  21   base::ReplaceSubstringsAfterOffset(content, 0, "\r\n", "\n");
  22   return true;
  23 }
  24
  25 // Write |content| to |file|. Returns true on success.
  26 bool WriteFile(const base::FilePath& file, const std::string& content) {
  27   int write_size = base::WriteFile(file, content.c_str(),
  28                                    static_cast<int>(content.length()));
  29   return write_size == static_cast<int>(content.length());
  30 }
  31
  32 }  // namespace
  33
  34 void DataDrivenTest::RunDataDrivenTest(
  35     const base::FilePath& input_directory,
  36     const base::FilePath& output_directory,
  37     const base::FilePath::StringType& file_name_pattern) {
  38   ASSERT_TRUE(base::DirectoryExists(input_directory));
  39   ASSERT_TRUE(base::DirectoryExists(output_directory));
  40   base::FileEnumerator input_files(input_directory,
  41                                    false,
  42                                    base::FileEnumerator::FILES,
  43                                    file_name_pattern);
  44
  45   for (base::FilePath input_file = input_files.Next();
  46        !input_file.empty();
  47        input_file = input_files.Next()) {
  48     RunOneDataDrivenTest(input_file, output_directory);
  49   }
  50 }
  51
  52 void DataDrivenTest::RunOneDataDrivenTest(
  53     const base::FilePath& test_file_name,
  54     const base::FilePath& output_directory) {
  55   // iOS doesn't get rid of removed test files. TODO(estade): remove this after
  56   // all iOS bots are clobbered.
  57   if (test_file_name.BaseName().value() == FILE_PATH_LITERAL("multimerge.in"))
  58     return;
  59
  60   ASSERT_TRUE(base::DirectoryExists(output_directory));
  61   SCOPED_TRACE(test_file_name.BaseName().value());
  62
  63   std::string input;
  64   ReadFile(test_file_name, &input);
  65
  66   std::string output;
  67   GenerateResults(input, &output);
  68
  69   base::FilePath output_file = output_directory.Append(
  70       test_file_name.BaseName().StripTrailingSeparators().ReplaceExtension(
  71           FILE_PATH_LITERAL(".out")));
  72
  73   std::string output_file_contents;
  74   if (ReadFile(output_file, &output_file_contents))
  75     EXPECT_EQ(output_file_contents, output);
  76   else
  77     ASSERT_TRUE(WriteFile(output_file, output));
  78 }
  79
  80 base::FilePath DataDrivenTest::GetInputDirectory(
  81     const base::FilePath::StringType& test_name) {
  82   base::FilePath dir;
  83   dir = test_data_directory_.AppendASCII("autofill")
  84                             .Append(test_name)
  85                             .AppendASCII("input");
  86   return dir;
  87 }
  88
  89 base::FilePath DataDrivenTest::GetOutputDirectory(
  90     const base::FilePath::StringType& test_name) {
  91   base::FilePath dir;
  92   dir = test_data_directory_.AppendASCII("autofill")
  93                             .Append(test_name)
  94                             .AppendASCII("output");
  95   return dir;
  96 }
  97
  98 DataDrivenTest::DataDrivenTest(const base::FilePath& test_data_directory)
  99     : test_data_directory_(test_data_directory) {
 100 }
 101
 102 DataDrivenTest::~DataDrivenTest() {
 103 }
 104
 105 }  // namespace autofill