1 // Copyright 2014 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 "base/test/gtest_util.h"
6
7 #include <stddef.h>
8
9 #include <memory>
10
11 #include "base/files/file_path.h"
12 #include "base/json/json_file_value_serializer.h"
13 #include "base/strings/string_util.h"
14 #include "base/values.h"
15 #include "testing/gtest/include/gtest/gtest.h"
16
17 namespace base {
18
19 TestIdentifier::TestIdentifier() = default;
20
21 TestIdentifier::TestIdentifier(const TestIdentifier& other) = default;
22
FormatFullTestName(const std::string & test_case_name,const std::string & test_name)23 std::string FormatFullTestName(const std::string& test_case_name,
24 const std::string& test_name) {
25 return test_case_name + "." + test_name;
26 }
27
TestNameWithoutDisabledPrefix(const std::string & full_test_name)28 std::string TestNameWithoutDisabledPrefix(const std::string& full_test_name) {
29 std::string test_name_no_disabled(full_test_name);
30 ReplaceSubstringsAfterOffset(&test_name_no_disabled, 0, "DISABLED_", "");
31 return test_name_no_disabled;
32 }
33
GetCompiledInTests()34 std::vector<TestIdentifier> GetCompiledInTests() {
35 testing::UnitTest* const unit_test = testing::UnitTest::GetInstance();
36
37 std::vector<TestIdentifier> tests;
38 for (int i = 0; i < unit_test->total_test_case_count(); ++i) {
39 const testing::TestCase* test_case = unit_test->GetTestCase(i);
40 for (int j = 0; j < test_case->total_test_count(); ++j) {
41 const testing::TestInfo* test_info = test_case->GetTestInfo(j);
42 TestIdentifier test_data;
43 test_data.test_case_name = test_case->name();
44 test_data.test_name = test_info->name();
45 test_data.file = test_info->file();
46 test_data.line = test_info->line();
47 tests.push_back(test_data);
48 }
49 }
50 return tests;
51 }
52
WriteCompiledInTestsToFile(const FilePath & path)53 bool WriteCompiledInTestsToFile(const FilePath& path) {
54 std::vector<TestIdentifier> tests(GetCompiledInTests());
55
56 ListValue root;
57 for (const auto& i : tests) {
58 std::unique_ptr<DictionaryValue> test_info(new DictionaryValue);
59 test_info->SetStringKey("test_case_name", i.test_case_name);
60 test_info->SetStringKey("test_name", i.test_name);
61 test_info->SetStringKey("file", i.file);
62 test_info->SetIntKey("line", i.line);
63 root.Append(std::move(test_info));
64 }
65
66 JSONFileValueSerializer serializer(path);
67 return serializer.Serialize(root);
68 }
69
ReadTestNamesFromFile(const FilePath & path,std::vector<TestIdentifier> * output)70 bool ReadTestNamesFromFile(const FilePath& path,
71 std::vector<TestIdentifier>* output) {
72 JSONFileValueDeserializer deserializer(path);
73 int error_code = 0;
74 std::string error_message;
75 std::unique_ptr<base::Value> value =
76 deserializer.Deserialize(&error_code, &error_message);
77 if (!value.get())
78 return false;
79
80 base::ListValue* tests = nullptr;
81 if (!value->GetAsList(&tests))
82 return false;
83
84 std::vector<base::TestIdentifier> result;
85 for (const auto& i : *tests) {
86 const base::DictionaryValue* test = nullptr;
87 if (!i.GetAsDictionary(&test))
88 return false;
89
90 TestIdentifier test_data;
91
92 if (!test->GetStringASCII("test_case_name", &test_data.test_case_name))
93 return false;
94
95 if (!test->GetStringASCII("test_name", &test_data.test_name))
96 return false;
97
98 if (!test->GetStringASCII("file", &test_data.file))
99 return false;
100
101 if (!test->GetInteger("line", &test_data.line))
102 return false;
103
104 result.push_back(test_data);
105 }
106
107 output->swap(result);
108 return true;
109 }
110
111 } // namespace base
112