extracttests: Added a simple program for extracting testcode from our unit testing

This commit is contained in:
Daniel Marjamäki 2009-11-07 18:04:45 +01:00
parent 3bd36226d6
commit b9ebdee2f5
1 changed files with 136 additions and 0 deletions

136
tools/extracttests.cpp Normal file
View File

@ -0,0 +1,136 @@
#include <iostream>
#include <fstream>
#include <sstream>
#include <list>
const std::string ext(".c");
const std::string gcc("gcc");
static std::string str(unsigned int value)
{
std::ostringstream ostr;
ostr << value;
return ostr.str();
}
int main(const int argc, const char * const * const argv)
{
if (argc != 2)
{
std::cerr << "syntax: extracttests testfile" << std::endl;
return 0;
}
std::list<std::string> fnames;
std::string testname;
unsigned int subcount = 0;
std::ifstream f(argv[1]);
std::string line;
while (std::getline(f, line))
{
if (line.compare(0, 9, " void ") == 0)
{
testname = line.substr(9, line.size() - 11);
std::cout << "\"" << testname << "\"" << std::endl;
subcount = 0;
continue;
}
if (line.find("}") != std::string::npos)
{
testname = "";
subcount = 0;
continue;
}
if (!testname.empty() && line.find(" check(\"") != std::string::npos)
{
const std::string fname(testname + str(++subcount));
if (ext == ".c")
{
if (fname == "array_index_21" ||
fname == "array_index_81" ||
fname == "array_index_111" ||
fname == "array_index_121" ||
fname == "array_index_131" ||
fname == "array_index_141" ||
fname == "array_index_151" ||
fname == "array_index_161" ||
fname == "array_index_171" ||
fname == "array_index_172" ||
fname == "array_index_173" ||
fname == "array_index_181" ||
fname == "array_index_201" ||
fname == "buffer_overrun_41" ||
fname == "buffer_overrun_81" ||
fname == "buffer_overrun_91" ||
fname == "buffer_overrun_101" ||
fname == "buffer_overrun_111" ||
fname == "sprintf12" ||
fname == "cin11" ||
fname == "alloc1")
continue;
}
fnames.push_back(fname);
std::ofstream fout((fname + ext).c_str());
fout << "#include <string.h>" << std::endl;
fout << "#include <stdio.h>" << std::endl;
fout << "#include <stdlib.h>" << std::endl;
do
{
std::string::size_type pos = line.find("\"");
if (pos == std::string::npos)
break;
line.erase(0, pos + 1);
pos = line.rfind("\"");
if (pos == std::string::npos)
break;
const bool lastline(line.find(");", pos) != std::string::npos);
line.erase(pos);
pos = 0;
while ((pos = line.find("\\", pos)) != std::string::npos)
{
line.erase(pos, 1);
if (line[pos] == 'n')
line.erase(pos, 1);
else
++pos;
}
fout << line << std::endl;
if (lastline)
break;
}
while (std::getline(f, line));
fout << std::endl;
continue;
}
}
std::ofstream makefile("Makefile");
makefile << "all:";
for (std::list<std::string>::const_iterator it = fnames.begin(); it != fnames.end(); ++it)
makefile << "\t" << *it << ".o";
makefile << std::endl;
for (std::list<std::string>::const_iterator it = fnames.begin(); it != fnames.end(); ++it)
{
makefile << *it << ".o:\t" << *it << ext << std::endl;
makefile << "\t" << gcc << " -c " << *it << ext << std::endl;
}
return 0;
}