update test cases

1 files changed, 6 insertions, 3 deletions

diff --git a/tests/storage/test_chewing_table.cpp b/tests/storage/test_chewing_table.cpp
index b65b20f..cae766d 100644
--- a/tests/storage/test_chewing_table.cpp
+++ b/tests/storage/test_chewing_table.cpp
@@ -38,9 +38,12 @@ int main(int argc, char * argv[]) {
     largetable.store(new_chunk);
     largetable.load(new_chunk);
 
-    char* linebuf = NULL; size_t size = 0;
-    while( getline(&linebuf, &size, stdin) ){
-        linebuf[strlen(linebuf)-1] = '\0';
+    char* linebuf = NULL; size_t size = 0; ssize_t read;
+    while ((read = getline(&linebuf, &size, stdin)) != -1) {
+        if ( '\n' == linebuf[strlen(linebuf) - 1] ) {
+            linebuf[strlen(linebuf) - 1] = '\0';
+        }
+
 	if ( strcmp ( linebuf, "quit" ) == 0)
 	    break;