| /* Distributed under the OSI-approved BSD 3-Clause License. See accompanying |
| file Copyright.txt or https://cmake.org/licensing for details. */ |
| #include "cmCreateTestSourceList.h" |
| |
| #include <algorithm> |
| |
| #include "cmMakefile.h" |
| #include "cmSourceFile.h" |
| #include "cmSystemTools.h" |
| |
| class cmExecutionStatus; |
| |
| // cmCreateTestSourceList |
| bool cmCreateTestSourceList::InitialPass(std::vector<std::string> const& args, |
| cmExecutionStatus&) |
| { |
| if (args.size() < 3) { |
| this->SetError("called with wrong number of arguments."); |
| return false; |
| } |
| |
| std::vector<std::string>::const_iterator i = args.begin(); |
| std::string extraInclude; |
| std::string function; |
| std::vector<std::string> tests; |
| // extract extra include and function ot |
| for (; i != args.end(); i++) { |
| if (*i == "EXTRA_INCLUDE") { |
| ++i; |
| if (i == args.end()) { |
| this->SetError("incorrect arguments to EXTRA_INCLUDE"); |
| return false; |
| } |
| extraInclude = "#include \""; |
| extraInclude += *i; |
| extraInclude += "\"\n"; |
| } else if (*i == "FUNCTION") { |
| ++i; |
| if (i == args.end()) { |
| this->SetError("incorrect arguments to FUNCTION"); |
| return false; |
| } |
| function = *i; |
| function += "(&ac, &av);\n"; |
| } else { |
| tests.push_back(*i); |
| } |
| } |
| i = tests.begin(); |
| |
| // Name of the source list |
| |
| const char* sourceList = i->c_str(); |
| ++i; |
| |
| // Name of the test driver |
| // make sure they specified an extension |
| if (cmSystemTools::GetFilenameExtension(*i).size() < 2) { |
| this->SetError( |
| "You must specify a file extension for the test driver file."); |
| return false; |
| } |
| std::string driver = this->Makefile->GetCurrentBinaryDirectory(); |
| driver += "/"; |
| driver += *i; |
| ++i; |
| |
| std::string configFile = cmSystemTools::GetCMakeRoot(); |
| |
| configFile += "/Templates/TestDriver.cxx.in"; |
| // Create the test driver file |
| |
| std::vector<std::string>::const_iterator testsBegin = i; |
| std::vector<std::string> tests_func_name; |
| |
| // The rest of the arguments consist of a list of test source files. |
| // Sadly, they can be in directories. Let's find a unique function |
| // name for the corresponding test, and push it to the tests_func_name |
| // list. |
| // For the moment: |
| // - replace spaces ' ', ':' and '/' with underscores '_' |
| std::string forwardDeclareCode; |
| for (i = testsBegin; i != tests.end(); ++i) { |
| if (*i == "EXTRA_INCLUDE") { |
| break; |
| } |
| std::string func_name; |
| if (!cmSystemTools::GetFilenamePath(*i).empty()) { |
| func_name = cmSystemTools::GetFilenamePath(*i) + "/" + |
| cmSystemTools::GetFilenameWithoutLastExtension(*i); |
| } else { |
| func_name = cmSystemTools::GetFilenameWithoutLastExtension(*i); |
| } |
| cmSystemTools::ConvertToUnixSlashes(func_name); |
| std::replace(func_name.begin(), func_name.end(), ' ', '_'); |
| std::replace(func_name.begin(), func_name.end(), '/', '_'); |
| std::replace(func_name.begin(), func_name.end(), ':', '_'); |
| tests_func_name.push_back(func_name); |
| forwardDeclareCode += "int "; |
| forwardDeclareCode += func_name; |
| forwardDeclareCode += "(int, char*[]);\n"; |
| } |
| |
| std::string functionMapCode; |
| int numTests = 0; |
| std::vector<std::string>::iterator j; |
| for (i = testsBegin, j = tests_func_name.begin(); i != tests.end(); |
| ++i, ++j) { |
| std::string func_name; |
| if (!cmSystemTools::GetFilenamePath(*i).empty()) { |
| func_name = cmSystemTools::GetFilenamePath(*i) + "/" + |
| cmSystemTools::GetFilenameWithoutLastExtension(*i); |
| } else { |
| func_name = cmSystemTools::GetFilenameWithoutLastExtension(*i); |
| } |
| functionMapCode += " {\n" |
| " \""; |
| functionMapCode += func_name; |
| functionMapCode += "\",\n" |
| " "; |
| functionMapCode += *j; |
| functionMapCode += "\n" |
| " },\n"; |
| numTests++; |
| } |
| if (!extraInclude.empty()) { |
| this->Makefile->AddDefinition("CMAKE_TESTDRIVER_EXTRA_INCLUDES", |
| extraInclude); |
| } |
| if (!function.empty()) { |
| this->Makefile->AddDefinition("CMAKE_TESTDRIVER_ARGVC_FUNCTION", function); |
| } |
| this->Makefile->AddDefinition("CMAKE_FORWARD_DECLARE_TESTS", |
| forwardDeclareCode); |
| this->Makefile->AddDefinition("CMAKE_FUNCTION_TABLE_ENTIRES", |
| functionMapCode); |
| bool res = true; |
| if (!this->Makefile->ConfigureFile(configFile, driver, false, true, false)) { |
| res = false; |
| } |
| |
| // Construct the source list. |
| std::string sourceListValue; |
| { |
| cmSourceFile* sf = this->Makefile->GetOrCreateSource(driver); |
| sf->SetProperty("ABSTRACT", "0"); |
| sourceListValue = args[1]; |
| } |
| for (i = testsBegin; i != tests.end(); ++i) { |
| cmSourceFile* sf = this->Makefile->GetOrCreateSource(*i); |
| sf->SetProperty("ABSTRACT", "0"); |
| sourceListValue += ";"; |
| sourceListValue += *i; |
| } |
| |
| this->Makefile->AddDefinition(sourceList, sourceListValue); |
| return res; |
| } |