@@ -24,7 +24,7 @@ TEST(BackendIntegTest, TestOTFProtocolAndHandler) {
24
24
// model_name length
25
25
.WillOnce (::testing::Return (5 ))
26
26
// model_path length
27
- .WillOnce (::testing::Return (45 ))
27
+ .WillOnce (::testing::Return (37 ))
28
28
// batch_size
29
29
.WillOnce (::testing::Return (1 ))
30
30
// handler length
@@ -44,8 +44,8 @@ TEST(BackendIntegTest, TestOTFProtocolAndHandler) {
44
44
strncpy (data, " mnist" , length);
45
45
}))
46
46
.WillOnce (testing::Invoke ([=](size_t length, char * data) {
47
- ASSERT_EQ (length, 45 );
48
- strncpy (data, " ../test/ resources/examples/mnist/base_handler" , length);
47
+ ASSERT_EQ (length, 37 );
48
+ strncpy (data, " resources/examples/mnist/base_handler" , length);
49
49
}))
50
50
.WillOnce (testing::Invoke ([=](size_t length, char * data) {
51
51
ASSERT_EQ (length, 11 );
@@ -59,7 +59,7 @@ TEST(BackendIntegTest, TestOTFProtocolAndHandler) {
59
59
EXPECT_CALL (*client_socket, SendAll (testing::_, testing::_)).Times (1 );
60
60
auto load_model_request = OTFMessage::RetrieveLoadMsg (*client_socket);
61
61
ASSERT_EQ (load_model_request->model_dir ,
62
- " ../test/ resources/examples/mnist/base_handler" );
62
+ " resources/examples/mnist/base_handler" );
63
63
ASSERT_EQ (load_model_request->model_name , " mnist" );
64
64
ASSERT_EQ (load_model_request->envelope , " " );
65
65
ASSERT_EQ (load_model_request->model_name , " mnist" );
@@ -68,9 +68,9 @@ TEST(BackendIntegTest, TestOTFProtocolAndHandler) {
68
68
69
69
// initialize backend
70
70
auto backend = std::make_shared<torchserve::Backend>();
71
- MetricsRegistry::Initialize (" ../test/ resources/metrics/default_config.yaml" ,
71
+ MetricsRegistry::Initialize (" resources/metrics/default_config.yaml" ,
72
72
MetricsContext::BACKEND);
73
- backend->Initialize (" ../test/ resources/examples/mnist/base_handler" );
73
+ backend->Initialize (" resources/examples/mnist/base_handler" );
74
74
75
75
// load the model
76
76
auto load_model_response = backend->LoadModel (load_model_request);
@@ -125,7 +125,7 @@ TEST(BackendIntegTest, TestOTFProtocolAndHandler) {
125
125
.WillOnce (testing::Invoke ([=](size_t length, char * data) {
126
126
ASSERT_EQ (length, 3883 );
127
127
// strncpy(data, "valu", length);
128
- std::ifstream input (" ../test/ resources/examples/mnist/0_png.pt" ,
128
+ std::ifstream input (" resources/examples/mnist/0_png.pt" ,
129
129
std::ios::in | std::ios::binary);
130
130
std::vector<char > image ((std::istreambuf_iterator<char >(input)),
131
131
(std::istreambuf_iterator<char >()));
0 commit comments