#include "catch2/catch.hpp"
#include "../src/pPowerManger/SQLiteDB.h"
#include "../src/pPowerManger/driver.h"

TEST_CASE("SQLiteDB Test", "[SQLiteDB]") {
    SQLiteDB db("test.db");

    SECTION("Create Tables") {
        REQUIRE(db.createPowerDataTables() == true);
        // REQUIRE(db.createStateTables() == true);
        REQUIRE(db.createDisCcuSysTables() == true);
        REQUIRE(db.createDriversTables("test") == true);

    }
    
    SECTION("Test isDatabaseConnected") {
        REQUIRE(db.isDatabaseConnected("/home/zjk/project/H100/H100PowerManger/bin","test.db") == true);
    }

    SECTION("Test insertPowerData") {
        PowerData powerData;
        StateData stateData;
        stateData.state_id = 1;
        stateData.state_name = "test";
        stateData.event = "test";
        stateData.event_id = 1;
        stateData.fault_code = 1;
        stateData.note = "test";
        REQUIRE(db.insertStateData(stateData) == true);
        // db.insertPowerData(powerData);
        
        StateData stateDataForGet;
        REQUIRE(db.getStateData(stateDataForGet) == true);
        REQUIRE(stateDataForGet.state_id == 1);
        REQUIRE(stateDataForGet.state_name == "test");
        REQUIRE(stateDataForGet.event == "test");
        REQUIRE(stateDataForGet.event_id == 1);
        REQUIRE(stateDataForGet.fault_code == 1);
        REQUIRE(stateDataForGet.note == "test");
        
        //REQUIRE(db.clearPowerData(stateDataForGet) == true);
        }
    
    SECTION("Test insertStateData") {
        
        for(int i = 0; i < 100; i++)
        {   DriverTable driver;
            REQUIRE(db.insertDriverTables(driver,"test") == true);
        }
        
    }

    SECTION("Test GETDATA") {
        StateData stateData;
        std::string time;
        REQUIRE(db.getStateData(stateData,time) == true);
        REQUIRE(db.getStateData(stateData) == true);
        REQUIRE(stateData.state_id == 1);
        REQUIRE(stateData.state_name == "test");
        cout << "-------------------------------------------------" << time << endl;
        
    }
       
    
}