#include "testutil.hh"
#include <cxxtest/TestSuite.h>

#include <fstream>
#include <sstream>
#include <sys/stat.h>

using namespace std ;
using std::tr1::unordered_map ;

vector<AccessPoint> TestUtil::aps ;
vector<Mobile> TestUtil::mobiles ;
vector<Request> TestUtil::requests ;


void TestUtil::tear_down()
{
  aps.clear() ;
  mobiles.clear() ;
  requests.clear() ;
}


void TestUtil::set_up()
{
  tear_down() ;

  // Create mobile list
  Mobile mobile1 ;
  mobile1.set_mac_addr("aa:bb:cc:dd:ee:ff") ;
  mobiles.push_back(mobile1) ;
  mobile1.set_mac_addr("aa:bb:cc:dd:ee:77") ;
  mobiles.push_back(mobile1) ;

  // Create AP list
  AccessPoint ap1 ;
  ap1.set_mac_addr("11:22:33:44:55:01") ;
  aps.push_back(ap1) ;
  ap1.set_mac_addr("11:22:33:44:55:02") ;
  aps.push_back(ap1) ;
  ap1.set_mac_addr("11:22:33:44:55:03") ;
  aps.push_back(ap1) ;

  // Create request list
  vector < unordered_map<string,Measurement> > measurements(3) ;
  Measurement measurement1 ;

  measurement1.add_ss(-58) ;
  measurement1.add_ss(-55) ;
  measurement1.set_ap(&aps[0]) ;
  measurements[0][aps[0].get_mac_addr()] = measurement1 ;
  measurement1.clear() ;
  measurement1.add_ss(-50) ;
  measurement1.set_ap(&aps[2]) ;
  measurements[0][aps[2].get_mac_addr()] = measurement1 ;
  measurement1.clear() ;
  measurement1.add_ss(-42) ;
  measurement1.add_ss(-37) ;
  measurement1.set_ap(&aps[1]) ;
  measurements[0][aps[1].get_mac_addr()] = measurement1 ;

  measurement1.clear() ;
  measurement1.add_ss(-51) ;
  measurement1.add_ss(-19) ;
  measurement1.set_ap(&aps[0]) ;
  measurements[1][aps[0].get_mac_addr()] = measurement1 ;
  measurement1.clear() ;
  measurement1.add_ss(-70) ;
  measurement1.add_ss(-21) ;
  measurement1.set_ap(&aps[1]) ;
  measurements[1][aps[1].get_mac_addr()] = measurement1 ;
  measurement1.clear() ;
  measurement1.add_ss(-59) ;
  measurement1.set_ap(&aps[2]) ;
  measurements[1][aps[2].get_mac_addr()] = measurement1 ;

  measurement1.clear() ;
  measurement1.add_ss(-54) ;
  measurement1.add_ss(-32) ;
  measurement1.set_ap(&aps[0]) ;
  measurements[2][aps[0].get_mac_addr()] = measurement1 ;
  measurement1.clear() ;
  measurement1.add_ss(-45) ;
  measurement1.add_ss(-23) ;
  measurement1.set_ap(&aps[1]) ;
  measurements[2][aps[1].get_mac_addr()] = measurement1 ;
  measurement1.clear() ;
  measurement1.add_ss(-56) ;
  measurement1.set_ap(&aps[2]) ;
  measurements[2][aps[2].get_mac_addr()] = measurement1 ;

  // Create timestamp list
  vector<Timestamp> timestamps ;
  timestamps.push_back(Timestamp(1265120910725)) ;
  timestamps.push_back(Timestamp(1265120911234)) ;
  timestamps.push_back(Timestamp(1265120912345)) ;

  // Create request list
  requests.push_back(Request(&mobiles[0], timestamps[0],
                             measurements[0])) ;
  requests.push_back(Request(&mobiles[1], timestamps[1],
                             measurements[1])) ;
  requests.push_back(Request(&mobiles[0], timestamps[2],
                             measurements[2])) ;
}


// Create the file output_file_name and fill it with the contents of
// output_lines
void TestUtil::fill_file(const string &output_file_name,
                         const vector<string> output_lines)
{
  // Open the file
  std::ofstream output_file ;
  output_file.open(output_file_name.c_str()) ;
  if (! output_file)
    TS_FAIL("Cannot open test file `"+ output_file_name
            +"` for creation!") ;

  // Write contents to the file
  for (std::vector<std::string>::const_iterator i = output_lines.begin() ;
       i != output_lines.end() ; ++i)
    output_file << *i ;

  output_file.close() ;
}


void TestUtil::remove_file(const string &file_name)
{
  if (remove(file_name.c_str()) == -1)
    TS_WARN("Cannot remove test file `"+ file_name +"`!") ;
}


void TestUtil::
create_test_csv_file(const string &file_name, bool with_spaces)
{
  // Fill name and contents of the test CSV file
  vector<string> csv_lines ;
  ostringstream line ;

  if (with_spaces)
    line << "\n   \n     " ;
  line << mobiles[0].get_mac_addr() ;
  line << ';' ;
  line << requests[0].get_timestamp().get_timestamp_ms() ;
  line << ";0;0;0;0;" ;
  line << aps[0].get_mac_addr() ;
  line << ";" ;
  line << requests[0].get_measurements().at(aps[0].get_mac_addr())
       .get_ss_list()[0] ;
  line << ";" ;
  line << aps[2].get_mac_addr() ;
  line << ";" ;
  line << requests[0].get_measurements().at(aps[2].get_mac_addr())
       .get_ss_list()[0] ;
  line << ";" ;
  line << aps[1].get_mac_addr() ;
  line << ";" ;
  line << requests[0].get_measurements().at(aps[1].get_mac_addr())
       .get_ss_list()[0] ;
  line << ";" ;
  line << aps[0].get_mac_addr() ;
  line << ";" ;
  line << requests[0].get_measurements().at(aps[0].get_mac_addr())
       .get_ss_list()[1] ;
  line << ";" ;
  line << aps[1].get_mac_addr() ;
  line << ";" ;
  line << requests[0].get_measurements().at(aps[1].get_mac_addr())
       .get_ss_list()[1] ;
  line << '\n' ;
  csv_lines.push_back(line.str()) ;

  line.str("") ;
  line << mobiles[1].get_mac_addr() ;
  line << ';' ;
  line << requests[1].get_timestamp().get_timestamp_ms() ;
  line << ";0;0;0;0;" ;
  line << aps[2].get_mac_addr() ;
  line << ";" ;
  line << requests[1].get_measurements().at(aps[2].get_mac_addr())
       .get_ss_list()[0] ;
  line << ";" ;
  line << aps[0].get_mac_addr() ;
  line << ";" ;
  line << requests[1].get_measurements().at(aps[0].get_mac_addr())
       .get_ss_list()[0] ;
  line << ";" ;
  line << aps[1].get_mac_addr() ;
  line << ";" ;
  line << requests[1].get_measurements().at(aps[1].get_mac_addr())
       .get_ss_list()[0] ;
  line << ";" ;
  line << aps[1].get_mac_addr() ;
  line << ";" ;
  line << requests[1].get_measurements().at(aps[1].get_mac_addr())
       .get_ss_list()[1] ;
  line << ";" ;
  line << aps[0].get_mac_addr() ;
  line << ";" ;
  line << requests[1].get_measurements().at(aps[0].get_mac_addr())
       .get_ss_list()[1] ;
  line << '\n' ;
  if (with_spaces)
    line << "     \n\
\n\t\n\
\n   \t   \n\
\n" ;
  csv_lines.push_back(line.str()) ;

  line.str("") ;
  if (with_spaces)
    line << '\t' ;
  line << mobiles[0].get_mac_addr() ;
  line << ';' ;
  line << requests[2].get_timestamp().get_timestamp_ms() ;
  line << ";0;0;0;0;" ;
  line << aps[2].get_mac_addr() ;
  line << ";" ;
  line << requests[2].get_measurements().at(aps[2].get_mac_addr())
       .get_ss_list()[0] ;
  line << ";" ;
  line << aps[1].get_mac_addr() ;
  line << ";" ;
  line << requests[2].get_measurements().at(aps[1].get_mac_addr())
       .get_ss_list()[0] ;
  line << ";" ;
  line << aps[0].get_mac_addr() ;
  line << ";" ;
  line << requests[2].get_measurements().at(aps[0].get_mac_addr())
       .get_ss_list()[0] ;
  line << ";" ;
  line << aps[1].get_mac_addr() ;
  line << ";" ;
  line << requests[2].get_measurements().at(aps[1].get_mac_addr())
       .get_ss_list()[1] ;
  line << ";" ;
  line << aps[0].get_mac_addr() ;
  line << ";" ;
  line << requests[2].get_measurements().at(aps[0].get_mac_addr())
       .get_ss_list()[1] ;
  line << '\n' ;
  if (with_spaces)
    line << "\n\t\n" ;
  csv_lines.push_back(line.str()) ;

  // Create and fill the test CSV file
  fill_file(file_name, csv_lines) ;
}


// Test equality of two Request comparing pointed values instead of
// pointers and taking care of the unordered_map
bool TestUtil::request_equals(const Request &first,
                              const Request &second)
{
  // Try a classical comparison
  if (first == second)
    return true ;

  // Compare timestamp
  if (first.get_timestamp() != second.get_timestamp())
    return false ;

  // Compare mobile values
  if (first.get_mobile() == NULL || second.get_mobile() == NULL)
    {
      if (first.get_mobile() != NULL || second.get_mobile() != NULL)
        return false ;
    }
  else if (*first.get_mobile() != *second.get_mobile())
    return false ;

  // Compare measurements
  if (first.get_measurements() != second.get_measurements())
    if (! measurements_unordered_map_equals(first.get_measurements(),
                                            second.get_measurements()))
      return false ;

  return true ;
}


// Test equality of two unordered_map<string,Measurement>
bool TestUtil::
measurements_unordered_map_equals(
  unordered_map<string,Measurement> first,
  unordered_map<string,Measurement> second)
{
  if (first.size() != second.size())
    return false ;

  // For each element in 'first', we look for the same element
  // in 'second'
  for (unordered_map<string,Measurement>::const_iterator i =
         first.begin() ; i != first.end() ; ++i)
    {
      unordered_map<string,Measurement>::const_iterator
      second_measurement = second.find(i->first) ;
      if (second_measurement == second.end())
        return false ;
      if (! measurement_equals(i->second, second_measurement->second))
        return false ;
    }

  return true ;
}


// Test equality of two Measurement comparing pointed values instead of
// pointers
bool TestUtil::measurement_equals(const Measurement &first,
                                  const Measurement &second)
{
  // Try a classical comparison
  if (first == second)
    return true ;

  // Compare average_ss
  if (first.get_average_ss() != second.get_average_ss())
    return false ;

  // Compare ap values
  if (first.get_ap() == NULL || second.get_ap() == NULL)
    {
      if (first.get_ap() != NULL || second.get_ap() != NULL)
        return false ;
    }
  else if (*first.get_ap() != *second.get_ap())
    return false ;

  // Compare ss_list
  if (first.get_ss_list() != second.get_ss_list())
    return false ;

  return true ;
}


bool TestUtil::file_size_equals(const string &file1_name,
                                const string &file2_name)
{
  try
    {
      return (file_size(file1_name) == file_size(file2_name)) ;
    }
  catch (...)
    {
      return false ;
    }
}


off_t TestUtil::file_size(const string &file_name)
{
  struct stat file_info ;
  if (stat(file_name.c_str(), &file_info) != 0)
    throw "Cannot get information for file `" + file_name + "`!" ;
  return file_info.st_size ;
}