Paweł Hajdan, Jr | a4797a3 | 2014-11-25 00:58:51 +0900 | [diff] [blame] | 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 | |
avi | f09d539 | 2015-12-24 12:28:02 +0900 | [diff] [blame] | 7 | #include <stddef.h> |
| 8 | |
dcheng | 7219b06 | 2016-06-11 12:41:48 +0900 | [diff] [blame] | 9 | #include <memory> |
| 10 | |
Paweł Hajdan, Jr | a4797a3 | 2014-11-25 00:58:51 +0900 | [diff] [blame] | 11 | #include "base/files/file_path.h" |
| 12 | #include "base/json/json_file_value_serializer.h" |
stgao | 05f1ea3 | 2016-11-01 06:38:44 +0900 | [diff] [blame] | 13 | #include "base/strings/string_util.h" |
Paweł Hajdan, Jr | a4797a3 | 2014-11-25 00:58:51 +0900 | [diff] [blame] | 14 | #include "base/values.h" |
| 15 | #include "testing/gtest/include/gtest/gtest.h" |
| 16 | |
| 17 | namespace base { |
| 18 | |
phajdan.jr | 700cfba | 2015-07-29 18:49:54 +0900 | [diff] [blame] | 19 | TestIdentifier::TestIdentifier() { |
| 20 | } |
| 21 | |
vmpstr | c8651bf | 2016-02-25 09:50:31 +0900 | [diff] [blame] | 22 | TestIdentifier::TestIdentifier(const TestIdentifier& other) = default; |
| 23 | |
Paweł Hajdan | 1509f2c | 2015-01-23 22:33:49 +0900 | [diff] [blame] | 24 | std::string FormatFullTestName(const std::string& test_case_name, |
| 25 | const std::string& test_name) { |
| 26 | return test_case_name + "." + test_name; |
| 27 | } |
| 28 | |
stgao | 05f1ea3 | 2016-11-01 06:38:44 +0900 | [diff] [blame] | 29 | std::string TestNameWithoutDisabledPrefix(const std::string& full_test_name) { |
| 30 | std::string test_name_no_disabled(full_test_name); |
| 31 | ReplaceSubstringsAfterOffset(&test_name_no_disabled, 0, "DISABLED_", ""); |
| 32 | return test_name_no_disabled; |
| 33 | } |
| 34 | |
phajdan.jr | 700cfba | 2015-07-29 18:49:54 +0900 | [diff] [blame] | 35 | std::vector<TestIdentifier> GetCompiledInTests() { |
Paweł Hajdan, Jr | a4797a3 | 2014-11-25 00:58:51 +0900 | [diff] [blame] | 36 | testing::UnitTest* const unit_test = testing::UnitTest::GetInstance(); |
| 37 | |
phajdan.jr | 700cfba | 2015-07-29 18:49:54 +0900 | [diff] [blame] | 38 | std::vector<TestIdentifier> tests; |
Paweł Hajdan, Jr | a4797a3 | 2014-11-25 00:58:51 +0900 | [diff] [blame] | 39 | for (int i = 0; i < unit_test->total_test_case_count(); ++i) { |
| 40 | const testing::TestCase* test_case = unit_test->GetTestCase(i); |
| 41 | for (int j = 0; j < test_case->total_test_count(); ++j) { |
| 42 | const testing::TestInfo* test_info = test_case->GetTestInfo(j); |
phajdan.jr | 700cfba | 2015-07-29 18:49:54 +0900 | [diff] [blame] | 43 | TestIdentifier test_data; |
| 44 | test_data.test_case_name = test_case->name(); |
| 45 | test_data.test_name = test_info->name(); |
| 46 | test_data.file = test_info->file(); |
| 47 | test_data.line = test_info->line(); |
| 48 | tests.push_back(test_data); |
Paweł Hajdan, Jr | a4797a3 | 2014-11-25 00:58:51 +0900 | [diff] [blame] | 49 | } |
| 50 | } |
| 51 | return tests; |
| 52 | } |
| 53 | |
| 54 | bool WriteCompiledInTestsToFile(const FilePath& path) { |
phajdan.jr | 700cfba | 2015-07-29 18:49:54 +0900 | [diff] [blame] | 55 | std::vector<TestIdentifier> tests(GetCompiledInTests()); |
Paweł Hajdan, Jr | a4797a3 | 2014-11-25 00:58:51 +0900 | [diff] [blame] | 56 | |
| 57 | ListValue root; |
| 58 | for (size_t i = 0; i < tests.size(); ++i) { |
dcheng | e1a4cd5 | 2016-09-09 06:04:33 +0900 | [diff] [blame] | 59 | std::unique_ptr<DictionaryValue> test_info(new DictionaryValue); |
phajdan.jr | 700cfba | 2015-07-29 18:49:54 +0900 | [diff] [blame] | 60 | test_info->SetString("test_case_name", tests[i].test_case_name); |
| 61 | test_info->SetString("test_name", tests[i].test_name); |
| 62 | test_info->SetString("file", tests[i].file); |
| 63 | test_info->SetInteger("line", tests[i].line); |
dcheng | 7219b06 | 2016-06-11 12:41:48 +0900 | [diff] [blame] | 64 | root.Append(std::move(test_info)); |
Paweł Hajdan, Jr | a4797a3 | 2014-11-25 00:58:51 +0900 | [diff] [blame] | 65 | } |
| 66 | |
| 67 | JSONFileValueSerializer serializer(path); |
| 68 | return serializer.Serialize(root); |
| 69 | } |
| 70 | |
Paweł Hajdan | 2273347 | 2015-01-15 22:32:01 +0900 | [diff] [blame] | 71 | bool ReadTestNamesFromFile(const FilePath& path, |
phajdan.jr | 700cfba | 2015-07-29 18:49:54 +0900 | [diff] [blame] | 72 | std::vector<TestIdentifier>* output) { |
prashhir | 58322f4 | 2015-03-05 18:30:57 +0900 | [diff] [blame] | 73 | JSONFileValueDeserializer deserializer(path); |
Paweł Hajdan | 2273347 | 2015-01-15 22:32:01 +0900 | [diff] [blame] | 74 | int error_code = 0; |
| 75 | std::string error_message; |
dcheng | cc8e4d8 | 2016-04-05 06:25:51 +0900 | [diff] [blame] | 76 | std::unique_ptr<base::Value> value = |
olli.raula | 6dc6aca | 2015-10-16 15:16:40 +0900 | [diff] [blame] | 77 | deserializer.Deserialize(&error_code, &error_message); |
Paweł Hajdan | 2273347 | 2015-01-15 22:32:01 +0900 | [diff] [blame] | 78 | if (!value.get()) |
| 79 | return false; |
| 80 | |
| 81 | base::ListValue* tests = nullptr; |
| 82 | if (!value->GetAsList(&tests)) |
| 83 | return false; |
| 84 | |
phajdan.jr | 700cfba | 2015-07-29 18:49:54 +0900 | [diff] [blame] | 85 | std::vector<base::TestIdentifier> result; |
Paweł Hajdan | 2273347 | 2015-01-15 22:32:01 +0900 | [diff] [blame] | 86 | for (base::ListValue::iterator i = tests->begin(); i != tests->end(); ++i) { |
| 87 | base::DictionaryValue* test = nullptr; |
rouslan | 7ce5d82 | 2017-04-11 23:58:00 +0900 | [diff] [blame] | 88 | if (!(*i)->GetAsDictionary(&test)) |
Paweł Hajdan | 2273347 | 2015-01-15 22:32:01 +0900 | [diff] [blame] | 89 | return false; |
| 90 | |
phajdan.jr | 700cfba | 2015-07-29 18:49:54 +0900 | [diff] [blame] | 91 | TestIdentifier test_data; |
| 92 | |
| 93 | if (!test->GetStringASCII("test_case_name", &test_data.test_case_name)) |
Paweł Hajdan | 2273347 | 2015-01-15 22:32:01 +0900 | [diff] [blame] | 94 | return false; |
| 95 | |
phajdan.jr | 700cfba | 2015-07-29 18:49:54 +0900 | [diff] [blame] | 96 | if (!test->GetStringASCII("test_name", &test_data.test_name)) |
Paweł Hajdan | 2273347 | 2015-01-15 22:32:01 +0900 | [diff] [blame] | 97 | return false; |
| 98 | |
phajdan.jr | 700cfba | 2015-07-29 18:49:54 +0900 | [diff] [blame] | 99 | if (!test->GetStringASCII("file", &test_data.file)) |
| 100 | return false; |
| 101 | |
| 102 | if (!test->GetInteger("line", &test_data.line)) |
| 103 | return false; |
| 104 | |
| 105 | result.push_back(test_data); |
Paweł Hajdan | 2273347 | 2015-01-15 22:32:01 +0900 | [diff] [blame] | 106 | } |
| 107 | |
| 108 | output->swap(result); |
| 109 | return true; |
| 110 | } |
| 111 | |
prashhir | 58322f4 | 2015-03-05 18:30:57 +0900 | [diff] [blame] | 112 | } // namespace base |