Commit 9c835ef5 authored by groot's avatar groot
Browse files

fix build error 


Former-commit-id: 1f517fb573f9319e2c9ce41bf0ba2725bd764898
parent 294ebb18

cpp/unittest/server/rpc_test.cpp

+11 −11
Original line number Diff line number Diff line
@@ -50,19 +50,19 @@ class RpcHandlerTest : public testing::Test { 
    void

    SetUp() override {



        auto res_mgr = engine::ResMgrInst::GetInstance();

        auto res_mgr = scheduler::ResMgrInst::GetInstance();

        res_mgr->Clear();

        res_mgr->Add(engine::ResourceFactory::Create("disk", "DISK", 0, true, false));

        res_mgr->Add(engine::ResourceFactory::Create("cpu", "CPU", 0, true, true));

        res_mgr->Add(engine::ResourceFactory::Create("gtx1660", "GPU", 0, true, true));

        res_mgr->Add(scheduler::ResourceFactory::Create("disk", "DISK", 0, true, false));

        res_mgr->Add(scheduler::ResourceFactory::Create("cpu", "CPU", 0, true, true));

        res_mgr->Add(scheduler::ResourceFactory::Create("gtx1660", "GPU", 0, true, true));



        auto default_conn = engine::Connection("IO", 500.0);

        auto PCIE = engine::Connection("IO", 11000.0);

        auto default_conn = scheduler::Connection("IO", 500.0);

        auto PCIE = scheduler::Connection("IO", 11000.0);

        res_mgr->Connect("disk", "cpu", default_conn);

        res_mgr->Connect("cpu", "gtx1660", PCIE);

        res_mgr->Start();

        engine::SchedInst::GetInstance()->Start();

        engine::JobMgrInst::GetInstance()->Start();

        scheduler::SchedInst::GetInstance()->Start();

        scheduler::JobMgrInst::GetInstance()->Start();



        engine::DBOptions opt;
@@ -106,9 +106,9 @@ class RpcHandlerTest : public testing::Test { 
    void

    TearDown() override {

        server::DBWrapper::GetInstance().StopService();

        engine::JobMgrInst::GetInstance()->Stop();

        engine::ResMgrInst::GetInstance()->Stop();

        engine::SchedInst::GetInstance()->Stop();

        scheduler::JobMgrInst::GetInstance()->Stop();

        scheduler::ResMgrInst::GetInstance()->Stop();

        scheduler::SchedInst::GetInstance()->Stop();

        boost::filesystem::remove_all("/tmp/milvus_test");

    }

 protected: