source: trunk/test/copyright_test.cc @ 687

// $Id: copyright_test.cc 687 2008-08-04 19:37:10Z peter $

/*
  Copyright (C) 2008 Peter Johansson

  This file is part of svndigest, http://dev.thep.lu.se/svndigest

  svndigest is free software; you can redistribute it and/or modify it
  under the terms of the GNU General Public License as published by
  the Free Software Foundation; either version 2 of the License, or
  (at your option) any later version.

  svndigest is distributed in the hope that it will be useful, but
  WITHOUT ANY WARRANTY; without even the implied warranty of
  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  General Public License for more details.

  You should have received a copy of the GNU General Public License
  along with this program; if not, write to the Free Software
  Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA
  02111-1307, USA.
*/

#include "Suite.h"

#include "Configuration.h"
#include "File.h"
#include "SVN.h"
#include "SVNinfo.h"

#include <cassert>
#include <iostream>
#include <fstream>
#include <sstream>
#include <string>
#include <vector>

std::vector<std::string> copyright_lines(std::istream&);

int main( int argc, char* argv[])
{
  using namespace theplu::svndigest;
  bool verbose=false;
  bool ok=true;
  if (argc>1 && argv[1]==std::string("-v") )
    verbose=true;

  std::string root=test::filename("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_old=copyright_lines(is);
  is.close();
  is.clear(std::ios::goodbit);

  if (copyrights_old.size()!=1) {
    if (verbose){
      std::cout << copyrights_old.size() << " Copyright lines\n";
      for (size_t i=0; i<copyrights_old.size(); ++i)
        std::cout << copyrights_old[i] << "\n";
    }
    ok = false;
  }
  else if (verbose)
    std::cout << "File contains 1 copyright line.\n"; 

  // warn about missing Copyright statement only in verbose mode
  if (verbose){
    std::string config_str("[copyright]\nmissing-copyright-warning=yes");
    std::stringstream ss(config_str);
    Configuration& config = Configuration::instance();
    config.load(ss);
  }

  if (verbose)
    std::cout << "Create SVN instance" << std::endl;
  SVN* svn=SVN::instance(root);
  if (!svn)
    return 1;

  // Extract repository location
  if (verbose)
    std::cout << "Extract repository location" << std::endl;
  std::string repo=SVNinfo(root).repos_root_url();
  if (verbose)
    std::cout << "Create File object" << std::endl;
  File file(0,filename,"");
  
  if (verbose)
    std::cout << "Get stats for file" << std::endl;
  file.parse(verbose, true);
  
  if (verbose)
    std::cout << "Updating copyright statements" << std::endl;
  std::map<std::string, Alias> alias;
  alias["jari"]=Alias("jh", 1);
  alias["peter"]=Alias("pj", 2);
  
  file.print_copyright(alias, verbose);

  is.open(filename.c_str());
  std::vector<std::string> copyrights=copyright_lines(is);
  is.close();
  is.clear(std::ios::goodbit);

  std::vector<std::string> copyright_correct;
  copyright_correct.push_back("Copyright (C) 2006 jh");
  copyright_correct.push_back("Copyright (C) 2007, 2008 pj");
  if (copyrights.size()!=copyright_correct.size()) {
    ok = false;
    if (verbose)
      std::cout << "ERROR: expected " << copyright_correct.size() 
                << " lines of Copyright (C)\n"
                << "But found " << copyrights.size() << " lines.\n";
  }
  else {
    for (size_t i=0; i<copyrights.size(); ++i)
      if (copyrights[i]!=copyright_correct[i]){
        ok=false;
        if (verbose)
          std::cerr << "ERROR: found '" << copyrights[i] << "'\n"
                    << "expected: '" << copyright_correct[i] << "'\n";
      }

  }


  // Restoring file
  std::ofstream os(filename.c_str());
  assert(os.good());
  os << original_file;
  os.close();

  if (ok) {
    if (verbose)
      std::cout << "Test is Ok!" << std::endl;
    return 0;
  }
  if (verbose)
    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;
}