[libcamera-devel,v7,5/5] lc-compliance: Add list and filter parameters

12559 diff mbox series

diff --git a/src/lc-compliance/main.cpp b/src/lc-compliance/main.cpp
index 8c18845141de..e0bd5892b8e4 100644
--- a/src/lc-compliance/main.cpp
+++ b/src/lc-compliance/main.cpp
@@ -20,6 +20,8 @@  using namespace libcamera;
 
 enum {
 	OptCamera = 'c',
+	OptList = 'l',
+	OptFilter = 'f',
 	OptHelp = 'h',
 };
 
@@ -45,13 +47,17 @@  public:
 	~Harness();
 
 	int init();
-	int run(int argc, char **argv);
+	int run(char *arg0);
 
 private:
+	int initGtestParameters(char *arg0);
 	void listCameras();
 
 	OptionsParser::Options options_;
 	std::unique_ptr<CameraManager> cm_;
+
+	const std::map<std::string, std::string> gtestFlag_ = { { "list", "--gtest_list_tests" },
+								{ "filter", "--gtest_filter" } };
 };
 
 Harness::Harness(const OptionsParser::Options &options)
@@ -76,6 +82,12 @@  int Harness::init()
 		return ret;
 	}
 
+	/*
+	 * No need to initialize the camera if we'll just list tests
+	 */
+	if (options_.isSet(OptList))
+		return 0;
+
 	if (!options_.isSet(OptCamera)) {
 		std::cout << "No camera specified, available cameras:" << std::endl;
 		listCameras();
@@ -97,9 +109,11 @@  int Harness::init()
 	return 0;
 }
 
-int Harness::run(int argc, char **argv)
+int Harness::run(char *arg0)
 {
-	::testing::InitGoogleTest(&argc, argv);
+	int ret = initGtestParameters(arg0);
+	if (ret)
+		return ret;
 
 	testing::UnitTest::GetInstance()->listeners().Append(new ThrowListener);
 
@@ -112,12 +126,59 @@  void Harness::listCameras()
 		std::cout << "- " << cam.get()->id() << std::endl;
 }
 
+int Harness::initGtestParameters(char *arg0)
+{
+	int argc = 0;
+	char **argv;
+	std::string filterParam;
+
+	/*
+	 * +2 to have space for both the 0th argument that is needed but not
+	 * used and the null at the end.
+	 */
+	argv = (char **) malloc((gtestFlag_.size() + 2) * sizeof(char *));
+	if (!argv)
+		return -ENOMEM;
+
+	argv[argc] = arg0;
+	argc++;
+
+	if (options_.isSet(OptList)) {
+		argv[argc] = const_cast<char *>(gtestFlag_.at("list").c_str());
+		argc++;
+	}
+
+	if (options_.isSet(OptFilter)) {
+		/*
+		 * The filter flag needs to be passed as a single parameter, in
+		 * the format --gtest_filter=filterStr
+		 */
+		filterParam = gtestFlag_.at("filter") + "=" +
+			(const std::string&) options_[OptFilter];
+
+		argv[argc] = const_cast<char *>(filterParam.c_str());
+		argc++;
+	}
+
+	argv[argc] = 0;
+
+	::testing::InitGoogleTest(&argc, argv);
+
+	free(argv);
+
+	return 0;
+}
+
 static int parseOptions(int argc, char **argv, OptionsParser::Options *options)
 {
 	OptionsParser parser;
 	parser.addOption(OptCamera, OptionString,
 			 "Specify which camera to operate on, by id", "camera",
 			 ArgumentRequired, "camera");
+	parser.addOption(OptList, OptionNone, "List all tests and exit", "list");
+	parser.addOption(OptFilter, OptionString,
+			 "Specify which tests to run", "filter",
+			 ArgumentRequired, "filter");
 	parser.addOption(OptHelp, OptionNone, "Display this help message",
 			 "help");
 
@@ -147,5 +208,5 @@  int main(int argc, char **argv)
 	if (ret)
 		return ret;
 
-	return harness.run(argc, argv);
+	return harness.run(argv[0]);
 }