mabain/src/unittest/mabain_unittest.cpp

   1 /**
   2  * Copyright (C) 2017 Cisco Inc.
   3  *
   4  * This program is free software: you can redistribute it and/or  modify
   5  * it under the terms of the GNU General Public License, version 2,
   6  * as published by the Free Software Foundation.
   7  *
   8  * This program is distributed in the hope that it will be useful,
   9  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  10  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  11  * GNU General Public License for more details.
  12  *
  13  * You should have received a copy of the GNU General Public License
  14  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  15  */
  16
  17 // @author Changxue Deng <chadeng@cisco.com>
  18
  19 #include <gtest/gtest.h>
  20 #include <gtest/tap.h>
  21
  22 #include "../db.h"
  23 #define MB_DIR "/var/tmp/mabain_test/"
  24
  25 GTEST_API_ int main(int argc, char **argv)
  26 {
  27     testing::InitGoogleTest(&argc, argv);
  28     testing::TestEventListeners& listeners = testing::UnitTest::GetInstance()->listeners();
  29     listeners.Append(new tap::TapListener());
  30
  31     mode_t mode = 0777;
  32     mkdir(MB_DIR, mode);
  33
  34     mabain::DB::SetLogFile("/var/tmp/mabain_test/mabain.log");
  35     int rval = RUN_ALL_TESTS();
  36
  37     mabain::DB::CloseLogFile();
  38     return rval;
  39 }
  40