Function TEST_F

serving/processor/serving/model_session_test.cc:229–240

};

TEST_F(ModelSessionMgrTest, CreateModelSessionReturnStatusOK) {
  MetaGraphDef test_graph_def;
  SessionOptions sess_options;
  RunOptions run_options;
  TestableModelSessionMgr mgr(test_graph_def, &sess_options, &run_options);

  Version version;
  ModelConfig config = CreateValidModelConfig();
  FakeFeatureStoreMgr feature_store(&config);
  EXPECT_TRUE(mgr.CreateModelSession(version, config.checkpoint_dir.c_str(),
        &feature_store, false, false, &config, "graph_signature").ok());
}

TEST_F(ModelSessionMgrTest, CreateModelSessionWhenPrevServingDone) {
  MetaGraphDef test_graph_def;