Changeset 594

Timestamp:

Apr 14, 2008, 5:24:02 AM (14 years ago)

Author:

Peter Johansson

Message:

working on copyright test

File:

• trunk/test/copyright.cc (modified) (4 diffs)

trunk/test/copyright.cc

@@ -31 +31 @@
 #include <fstream>
 #include <string>
+#include <vector>
+
+std::vector<std::string> copyright_lines(std::istream&);
 
 int main( int argc, char* argv[])
@@ -36 +39 @@
   using namespace theplu::svndigest;
   bool verbose=false;
+  bool ok=true;
   if (argc>1 && argv[1]==std::string("-v") )
     verbose=true;
 
   std::string root("toy_project");
+  std::string filename = root + "/README";
+
+  // Saving original file
+  std::ifstream is(filename.c_str());
+  assert(is.good());
+  std::string original_file;
+  std::getline(is, original_file, '\0');
+  is.close();
+  is.clear(std::ios::goodbit);
+
+  is.open(filename.c_str());
+  std::vector<std::string> copyrights=copyright_lines(is);
+  is.close();
+  is.clear(std::ios::goodbit);
+
+  if (copyrights.size()!=1) {
+    if (verbose){
+      std::cout << copyrights.size() << " Copyright lines\n";
+      for (size_t i=0; i<copyrights.size(); ++i)
+        std::cout << copyrights[i] << "\n";
+    }
+    ok = false;
+  }
+  else if (verbose)
+    std::cout << "File containes 1 copyright line.\n"; 
+
   if (verbose)
     std::cout << "Create SVN instance" << std::endl;
@@ -52 +82 @@
   if (verbose)
     std::cout << "Create File object" << std::endl;
-  File file(0,root+"/README","");
+  File file(0,filename,"");
   
   if (verbose)
@@ -63 +93 @@
   file.print_copyright(alias, verbose);
 
+  // Restoring file
+  std::ofstream os(filename.c_str());
+  assert(os.good());
+  os << original_file;
+
+  if (ok) {
+    if (verbose)
+      std::cout << "Test is Ok!" << std::endl;
+    return 0;
+  }
   if (verbose)
-    std::cout << "Done!" << std::endl;
-  return 0;
+    std::cout << "Test failed." << std::endl;
+  return 1;
 }
 
+std::vector<std::string> copyright_lines(std::istream& is)
+{
+  using namespace theplu::svndigest;
+  std::vector<std::string> res;
+  std::string line;
+  while (std::getline(is, line)){
+    if (match_begin(line.begin(), line.end(), "Copyright (C)"))
+      res.push_back(line);
+  }
+  
+  return res;
+}
+