1 files changed, 17 insertions, 4 deletions
diff --git a/tests/test/simple/generated/driver.cxx b/tests/test/simple/generated/driver.cxx
index 18fd0ae..ca6dfcb 100644
--- a/tests/test/simple/generated/driver.cxx
+++ b/tests/test/simple/generated/driver.cxx
@@ -12,6 +12,9 @@
 #include <fstream>
 #include <iostream>
 
+#undef NDEBUG
+#include <cassert>
+
 using namespace std;
 
 // If the -s option is specified, then also sleep for 5 seconds.
@@ -46,13 +49,23 @@ main (int argc, char* argv[])
   }
   else
   {
-    ifstream ifs (argv[i]);
+    istream* is;
+    ifstream ifs;
+
+    if (argv[i] != string ("-"))
+    {
+      ifs.open (argv[i]);
 
-    if (!ifs.is_open ())
-      cerr << "unable to open " << argv[1] << endl;
+      if (!ifs.is_open ())
+        cerr << "unable to open " << argv[1] << endl;
+
+      is = &ifs;
+    }
+    else
+      is = &cin;
 
     string s;
-    r = getline (ifs, s) && s == "1.2.3" ? 0 : 1;
+    r = getline (*is, s) && s == "1.2.3" ? 0 : 1;
   }
 
   return r;