owlps/owlps-positioning/tests/inputcsv_test.hh

#include <cxxtest/TestSuite.h>

#include "inputcsv.hh"
#include "stock.hh"

#include <string>
#include <fstream>
#include <cstdio>

class InputCSV_test: public CxxTest::TestSuite
{
private:
  std::string csv_file_name ; // Test CSV file name
  std::vector<AccessPoint> aps ; // List of test AccessPoint


public:
  InputCSV_test(void)
  {
    // If we cannot open the file, we want to stop the test
    CxxTest::setAbortTestOnFail(true) ;

    // Clear the stock
    Stock::clear() ;

    csv_file_name = "/tmp/InputCSV_test_csv_file.csv" ;
    aps = TestUtil::create_test_csv_file(csv_file_name) ;

    // Back to the normal behaviour (i.e. do not abort on fail)
    CxxTest::setAbortTestOnFail(false) ;
  }


  ~InputCSV_test(void)
  {
    // Delete the test CSV file
    TestUtil::remove_file(csv_file_name) ;

    // Clear the stock
    Stock::clear() ;
  }


  static InputCSV_test* createSuite(void)
  {
    return new InputCSV_test() ;
  }


  static void destroySuite(InputCSV_test *suite)
  {
    delete suite ;
  }


  void test_inputcsv(void)
  {
    // Opening file
    InputCSV inputcsv1(csv_file_name) ;
    TS_ASSERT(inputcsv1) ;
    TS_ASSERT(! inputcsv1.eof()) ;

    Request request1 ;
    std::tr1::unordered_map<std::string, Measurement> measurements ;

    // First Request //

    request1 = inputcsv1.get_next_request() ;
    TS_ASSERT_EQUALS(request1.get_mobile()->get_mac_addr(),
                     "aa:bb:cc:dd:ee:ff") ;
    TS_ASSERT_EQUALS(request1.get_timestamp().get_timestamp_ms(),
                     1265120910725llu) ;

    measurements[aps[0].get_mac_addr()].add_ss(-58) ;
    measurements[aps[2].get_mac_addr()].add_ss(-50) ;
    measurements[aps[1].get_mac_addr()].add_ss(-42) ;
    measurements[aps[0].get_mac_addr()].add_ss(-55) ;
    measurements[aps[1].get_mac_addr()].add_ss(-37) ;

    for (std::vector<AccessPoint>::const_iterator i = aps.begin() ;
         i != aps.end() ; ++i)
      {
        std::tr1::unordered_map<std::string, Measurement>::const_iterator
        measurement_it1 =
          request1.get_measurements().find(i->get_mac_addr()) ;
        TS_ASSERT_DIFFERS(request1.get_measurements().end(), measurement_it1) ;
        TS_ASSERT_EQUALS(measurements[i->get_mac_addr()].get_ss_list(),
                         measurement_it1->second.get_ss_list()) ;
      }

    TS_ASSERT(inputcsv1) ;
    TS_ASSERT(! inputcsv1.eof()) ;

    // Second Request //

    request1 = inputcsv1.get_next_request() ;
    TS_ASSERT_EQUALS(request1.get_mobile()->get_mac_addr(),
                     "aa:bb:cc:dd:ee:77") ;
    TS_ASSERT_EQUALS(request1.get_timestamp().get_timestamp_ms(),
                     1265120911234llu) ;

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

    for (std::vector<AccessPoint>::const_iterator i = aps.begin() ;
         i != aps.end() ; ++i)
      {
        std::tr1::unordered_map<std::string, Measurement>::const_iterator
        measurement_it1 =
          request1.get_measurements().find(i->get_mac_addr()) ;
        TS_ASSERT_DIFFERS(request1.get_measurements().end(), measurement_it1) ;
        TS_ASSERT_EQUALS(measurements[i->get_mac_addr()].get_ss_list(),
                         measurement_it1->second.get_ss_list()) ;
      }

    // Third Request //

    TS_ASSERT(inputcsv1) ;
    TS_ASSERT(! inputcsv1.eof()) ;

    request1 = inputcsv1.get_next_request() ;
    TS_ASSERT_EQUALS(request1.get_mobile()->get_mac_addr(),
                     "aa:bb:cc:dd:ee:ff") ;
    TS_ASSERT_EQUALS(request1.get_timestamp().get_timestamp_ms(),
                     1265120912345llu) ;

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

    for (std::vector<AccessPoint>::const_iterator i = aps.begin() ;
         i != aps.end() ; ++i)
      {
        std::tr1::unordered_map<std::string, Measurement>::const_iterator
        measurement_it1 =
          request1.get_measurements().find(i->get_mac_addr()) ;
        TS_ASSERT_DIFFERS(request1.get_measurements().end(), measurement_it1) ;
        TS_ASSERT_EQUALS(measurements[i->get_mac_addr()].get_ss_list(),
                         measurement_it1->second.get_ss_list()) ;
      }

    // End of file

    TS_ASSERT(! inputcsv1) ;
    TS_ASSERT(inputcsv1.eof()) ;

    request1 = inputcsv1.get_next_request() ;
    TS_ASSERT(! request1) ;

    TS_ASSERT(! inputcsv1) ;
    TS_ASSERT(inputcsv1.eof()) ;
  }

} ;