test.cpp: run tests with std::istringstream, std::ifstream and file input

Author: firewave

test.cpp

@@ -18,10 +18,18 @@
 
 #include <iostream>
 #include <limits>
+#include <fstream>
 #include <sstream>
 #include <vector>
 #include "simplecpp.h"
 
+enum Input {
+    Stringstream,
+    Fstream,
+    File
+};
+
+static Input USE_INPUT = Stringstream;
 static int numberOfFailedAssertions = 0;
 
 #define ASSERT_EQUALS(expected, actual)  (assertEquals((expected), (actual), __LINE__))
@@ -38,11 +46,22 @@ static std::string pprint(const std::string &in)
     return ret;
 }
 
+static const char* inputString(Input input) {
+    switch (input) {
+        case Stringstream:
+            return "Stringstream";
+        case Fstream:
+            return "Fstream";
+        case File:
+            return "File";
+    }
+}
+
 static int assertEquals(const std::string &expected, const std::string &actual, int line)
 {
     if (expected != actual) {
         numberOfFailedAssertions++;
-        std::cerr << "------ assertion failed ---------" << std::endl;
+        std::cerr << "------ assertion failed (" << inputString(USE_INPUT) << ") ---------" << std::endl;
         std::cerr << "line " << line << std::endl;
         std::cerr << "expected:" << pprint(expected) << std::endl;
         std::cerr << "actual:" << pprint(actual) << std::endl;
@@ -77,10 +96,44 @@ static void testcase(const std::string &name, void (*f)(), int argc, char * cons
 
 #define TEST_CASE(F)    (testcase(#F, F, argc, argv))
 
+static std::string writeFile(const char code[], std::size_t size, const std::string &filename) {
+    std::string tmpfile = filename.empty() ? "code.tmp" : filename;
+    {
+        std::ofstream of(tmpfile, std::ios_base::out | std::ios_base::binary | std::ios_base::trunc);
+        of.write(code, size);
+    }
+    return tmpfile;
+}
+
+static simplecpp::TokenList makeTokenListFromFstream(const char code[], std::size_t size, std::vector<std::string> &filenames, const std::string &filename, simplecpp::OutputList *outputList)
+{
+    const std::string tmpfile = writeFile(code, size, filename);
+    std::ifstream fin(tmpfile);
+    simplecpp::TokenList tokenList(fin, filenames, tmpfile, outputList);
+    remove(tmpfile.c_str());
+    return tokenList;
+}
+
+static simplecpp::TokenList makeTokenListFromFile(const char code[], std::size_t size, std::vector<std::string> &filenames, const std::string &filename, simplecpp::OutputList *outputList)
+{
+    const std::string tmpfile = writeFile(code, size, filename);
+    simplecpp::TokenList tokenList(filenames, tmpfile, outputList);
+    remove(tmpfile.c_str());
+    return tokenList;
+}
+
 static simplecpp::TokenList makeTokenList(const char code[], std::size_t size, std::vector<std::string> &filenames, const std::string &filename=std::string(), simplecpp::OutputList *outputList=nullptr)
 {
-    std::istringstream istr(std::string(code, size));
-    return simplecpp::TokenList(istr,filenames,filename,outputList);
+    switch (USE_INPUT) {
+        case Stringstream: {
+            std::istringstream istr(std::string(code, size));
+            return simplecpp::TokenList(istr, filenames, filename, outputList);
+        }
+        case Fstream:
+            return makeTokenListFromFstream(code, size, filenames, filename, outputList);
+        case File:
+            return makeTokenListFromFile(code, size, filenames, filename, outputList);
+    }
 }
 
 static simplecpp::TokenList makeTokenList(const char code[], std::vector<std::string> &filenames, const std::string &filename=std::string(), simplecpp::OutputList *outputList=nullptr)
@@ -2362,8 +2415,10 @@ static void cpluscplusDefine()
     ASSERT_EQUALS("\n201103L", preprocess(code, dui));
 }
 
-int main(int argc, char **argv)
+static void runTests(int argc, char **argv, Input input)
 {
+    USE_INPUT = input;
+
     TEST_CASE(backslash);
 
     TEST_CASE(builtin);
@@ -2560,6 +2615,12 @@ int main(int argc, char **argv)
 
     TEST_CASE(stdcVersionDefine);
     TEST_CASE(cpluscplusDefine);
+}
 
+int main(int argc, char **argv)
+{
+    runTests(argc, argv, Stringstream);
+    runTests(argc, argv, Fstream);
+    runTests(argc, argv, File);
     return numberOfFailedAssertions > 0 ? EXIT_FAILURE : EXIT_SUCCESS;
 }