From ea5badc0d3a9d1785e40e76d5becf1198b7e19cc Mon Sep 17 00:00:00 2001
From: krunal <krunal1313@gmail.com>
Date: Mon, 7 Aug 2023 20:53:10 +0530
Subject: [PATCH] updating test

---
 test/core_api_utils_test.cpp | 49 ++++++++++++++++++++++++++++--------
 1 file changed, 39 insertions(+), 10 deletions(-)

diff --git a/test/core_api_utils_test.cpp b/test/core_api_utils_test.cpp
index 82236699..d2ae2599 100644
--- a/test/core_api_utils_test.cpp
+++ b/test/core_api_utils_test.cpp
@@ -933,7 +933,7 @@ TEST_F(CoreAPIUtilsTest, ExportIncludeExcludeFields) {
 
     // exclude fields
 
-    delete dynamic_cast<deletion_state_t*>(req->data);
+    delete dynamic_cast<export_state_t*>(req->data);
     req->data = nullptr;
     res->body.clear();
     req->params.erase("include_fields");
@@ -994,7 +994,7 @@ TEST_F(CoreAPIUtilsTest, ExportIncludeExcludeFieldsWithFilter) {
 
     // exclude fields
 
-    delete dynamic_cast<deletion_state_t*>(req->data);
+    delete dynamic_cast<export_state_t*>(req->data);
     req->data = nullptr;
     res->body.clear();
     req->params.erase("include_fields");
@@ -1173,16 +1173,45 @@ TEST_F(CoreAPIUtilsTest, SampleGzipIndexTest) {
     }
 
     auto req = std::make_shared<http_req>();
-    char buf[196605]; //chunk size limit
     std::ifstream infile(std::string(ROOT_DIR)+"test/resources/hnstories.jsonl.gz");
+    std::stringstream outbuffer;
 
-    while(infile >> buf) {
-        req->body = buf;
+    infile.seekg (0, infile.end);
+    int length = infile.tellg();
+    infile.seekg (0, infile.beg);
 
-        auto res = ReplicationState::handle_gzip(req);
-        if(!res.error().empty()) {
-            LOG(ERROR) << res.error();
-            FAIL();
-        }
+    req->body.resize(length);
+    infile.read(&req->body[0], length);
+
+    auto res = ReplicationState::handle_gzip(req);
+    if (!res.error().empty()) {
+        LOG(ERROR) << res.error();
+        FAIL();
+    } else {
+        outbuffer << req->body;
     }
+
+    std::vector<std::string> doc_lines;
+    std::string line;
+    while(std::getline(outbuffer, line)) {
+        doc_lines.push_back(line);
+    }
+
+    ASSERT_EQ(14, doc_lines.size());
+    ASSERT_EQ("{\"points\":1,\"title\":\"DuckDuckGo Settings\"}", doc_lines[0]);
+    ASSERT_EQ("{\"points\":1,\"title\":\"Making Twitter Easier to Use\"}", doc_lines[1]);
+    ASSERT_EQ("{\"points\":2,\"title\":\"London refers Uber app row to High Court\"}", doc_lines[2]);
+    ASSERT_EQ("{\"points\":1,\"title\":\"Young Global Leaders, who should be nominated? (World Economic Forum)\"}", doc_lines[3]);
+    ASSERT_EQ("{\"points\":1,\"title\":\"Blooki.st goes BETA in a few hours\"}", doc_lines[4]);
+    ASSERT_EQ("{\"points\":1,\"title\":\"Unicode Security Data: Beta Review\"}", doc_lines[5]);
+    ASSERT_EQ("{\"points\":2,\"title\":\"FileMap: MapReduce on the CLI\"}", doc_lines[6]);
+    ASSERT_EQ("{\"points\":1,\"title\":\"[Full Video] NBC News Interview with Edward Snowden\"}", doc_lines[7]);
+    ASSERT_EQ("{\"points\":1,\"title\":\"Hybrid App Monetization Example with Mobile Ads and In-App Purchases\"}", doc_lines[8]);
+    ASSERT_EQ("{\"points\":1,\"title\":\"We need oppinion from Android Developers\"}", doc_lines[9]);
+    ASSERT_EQ("{\"points\":1,\"title\":\"\\\\t Why Mobile Developers Should Care About Deep Linking\"}", doc_lines[10]);
+    ASSERT_EQ("{\"points\":2,\"title\":\"Are we getting too Sassy? Weighing up micro-optimisation vs. maintainability\"}", doc_lines[11]);
+    ASSERT_EQ("{\"points\":2,\"title\":\"Google's XSS game\"}", doc_lines[12]);
+    ASSERT_EQ("{\"points\":1,\"title\":\"Telemba Turns Your Old Roomba and Tablet Into a Telepresence Robot\"}", doc_lines[13]);
+
+    infile.close();
 }
\ No newline at end of file