planning_test_base.h

浏览该文件的文档.

/******************************************************************************
 * Copyright 2017 The Apollo Authors. All Rights Reserved.
 *
 * 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
 *
 * http://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 <memory>
#include <set>
#include <string>
 
#include "gtest/gtest.h"
 
// TODO(all) #include "modules/planning/planning_component/navi_planning.h"
#include "modules/planning/planning_component/on_lane_planning.h"
#include "modules/planning/planning_component/planning_base.h"
 
namespace apollo {
namespace planning {
 
#define RUN_GOLDEN_TEST(sub_case_num)                                      \
  {                                                                        \
    const ::testing::TestInfo* const test_info =                           \
        ::testing::UnitTest::GetInstance()->current_test_info();           \
    bool no_trajectory_point = false;                                      \
    bool run_planning_success =                                            \
        RunPlanning(test_info->name(), sub_case_num, no_trajectory_point); \
    EXPECT_TRUE(run_planning_success);                                     \
  }
 
#define RUN_GOLDEN_TEST_DECISION(sub_case_num)                             \
  {                                                                        \
    const ::testing::TestInfo* const test_info =                           \
        ::testing::UnitTest::GetInstance()->current_test_info();           \
    bool no_trajectory_point = true;                                       \
    bool run_planning_success =                                            \
        RunPlanning(test_info->name(), sub_case_num, no_trajectory_point); \
    EXPECT_TRUE(run_planning_success);                                     \
  }
 
#define TMAIN                                            \
  int main(int argc, char** argv) {                      \
    ::apollo::cyber::Init("planning_test");              \
    ::testing::InitGoogleTest(&argc, argv);              \
    ::google::ParseCommandLineFlags(&argc, &argv, true); \
    return RUN_ALL_TESTS();                              \
  }
 
#define ENABLE_RULE(RULE_ID) this->rule_enabled_.emplace(RULE_ID)
 
DECLARE_string(test_routing_response_file);
DECLARE_string(test_localization_file);
DECLARE_string(test_chassis_file);
DECLARE_string(test_data_dir);
DECLARE_string(test_prediction_file);
DECLARE_string(test_traffic_light_file);
DECLARE_string(test_relative_map_file);
DECLARE_string(test_previous_planning_file);
 
class PlanningTestBase : public ::testing::Test {
 public:
  virtual ~PlanningTestBase() = default;
 
  static void SetUpTestCase();
  virtual void SetUp();
  void UpdateData();
 
  bool RunPlanning(const std::string& test_case_name, int case_num,
                   bool no_trajectory_point);
 
  std::shared_ptr<TrafficRule> GetTrafficRuleConfig(const std::string& rule_id);
 
 protected:
  void TrimPlanning(ADCTrajectory* origin, bool no_trajectory_point);
  bool FeedTestData();
  bool IsValidTrajectory(const ADCTrajectory& trajectory);
 
 protected:
  std::unique_ptr<PlanningBase> planning_ = nullptr;
  std::set<std::string> rule_enabled_;
  ADCTrajectory adc_trajectory_;
  LocalView local_view_;
  PlanningConfig config_;
  std::shared_ptr<DependencyInjector> injector_;
};
 
}  // namespace planning
}  // namespace apollo