| // Copyright 2022 The Centipede Authors. |
| // |
| // Licensed under the Apache License, Version 2.0 (the "License"); |
| // you may not use this file except in compliance with the License. |
| // You may obtain a copy of the License at |
| // |
| // https://www.apache.org/licenses/LICENSE-2.0 |
| // |
| // Unless required by applicable law or agreed to in writing, software |
| // distributed under the License is distributed on an "AS IS" BASIS, |
| // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| // See the License for the specific language governing permissions and |
| // limitations under the License. |
| |
| #include "./centipede/environment.h" |
| |
| #include "gtest/gtest.h" |
| |
| namespace centipede { |
| |
| TEST(Environment, UpdateForExperiment) { |
| Environment env; |
| env.num_threads = 12; |
| env.experiment = "use_cmp_features=false,true:path_level=10,20,30"; |
| |
| auto Experiment = [&](size_t shard_index, bool val1, size_t val2, |
| std::string_view experiment_name, |
| std::string_view experiment_flags) { |
| env.my_shard_index = shard_index; |
| env.UpdateForExperiment(); |
| EXPECT_EQ(env.load_other_shard_frequency, 0); |
| EXPECT_EQ(env.use_cmp_features, val1); |
| EXPECT_EQ(env.path_level, val2); |
| EXPECT_EQ(env.experiment_name, experiment_name); |
| EXPECT_EQ(env.experiment_flags, experiment_flags); |
| }; |
| |
| Experiment(0, false, 10, "E00", "use_cmp_features=false:path_level=10:"); |
| Experiment(1, false, 20, "E01", "use_cmp_features=false:path_level=20:"); |
| Experiment(2, false, 30, "E02", "use_cmp_features=false:path_level=30:"); |
| Experiment(3, true, 10, "E10", "use_cmp_features=true:path_level=10:"); |
| Experiment(4, true, 20, "E11", "use_cmp_features=true:path_level=20:"); |
| Experiment(5, true, 30, "E12", "use_cmp_features=true:path_level=30:"); |
| Experiment(6, false, 10, "E00", "use_cmp_features=false:path_level=10:"); |
| Experiment(7, false, 20, "E01", "use_cmp_features=false:path_level=20:"); |
| Experiment(8, false, 30, "E02", "use_cmp_features=false:path_level=30:"); |
| Experiment(9, true, 10, "E10", "use_cmp_features=true:path_level=10:"); |
| Experiment(10, true, 20, "E11", "use_cmp_features=true:path_level=20:"); |
| Experiment(11, true, 30, "E12", "use_cmp_features=true:path_level=30:"); |
| } |
| |
| TEST(Environment, MakeCoverageReportPath) { |
| // TODO(ussuri): Environment is not test-friendly (initialized through |
| // flags, which are hidden in the .cc). Fix. |
| EXPECT_EQ(Environment{}.MakeCoverageReportPath(), |
| "coverage-report-.000000.txt"); |
| EXPECT_EQ(Environment{}.MakeCoverageReportPath("initial"), |
| "coverage-report-.000000.initial.txt"); |
| } |
| |
| } // namespace centipede |