Return "missing property" when getting results as list

IntelLabs · Apr 26, 2018 · 9f385f8 · 9f385f8
1 parent 9e37069
commit 9f385f8
Show file tree

Hide file tree

Showing 4 changed files with 22 additions and 4 deletions.
diff --git a/src/PMGDQueryHandler.cc b/src/PMGDQueryHandler.cc
@@ -484,10 +484,12 @@ void PMGDQueryHandler::build_results(Iterator &ni,
         for (; ni; ni.next()) {
             for (int i = 0; i < keyids.size(); ++i) {
                 Property j_p;
-                if (!ni->check_property(keyids[i], j_p))
-                    continue;
                 PMGDPropList &list = rmap[qn.response_keys(i)];
                 PMGDProp *p_p = list.add_values();
+                if (!ni->check_property(keyids[i], j_p)) {
+                    construct_missing_property(p_p);
+                    continue;
+                }
                 construct_protobuf_property(j_p, p_p);
             }
             count++;
@@ -584,3 +586,10 @@ void PMGDQueryHandler::construct_protobuf_property(const Property &j_p, PMGDProp
         p_p->set_blob_value(j_p.blob_value().value, j_p.blob_value().size);
     }
 }
+
+void PMGDQueryHandler::construct_missing_property(PMGDProp *p_p)
+{
+    // Assumes matching enum values!
+    p_p->set_type(PMGDProp::StringType);
+    p_p->set_string_value("Missing property");
+}
diff --git a/src/PMGDQueryHandler.h b/src/PMGDQueryHandler.h
@@ -92,6 +92,7 @@ namespace VDMS {
                                                     const PMGDQueryNode &qn,
                                                     PMGDCmdResponse *response);
         void construct_protobuf_property(const PMGD::Property &j_p, PMGDProp*p_p);
+        void construct_missing_property(PMGDProp *p_p);
 
         void set_response(PMGDCmdResponse *response, PMGDCmdErrorCode error_code,
                             std::string error_msg)

diff --git a/tests/AddAndFind_query.json b/tests/AddAndFind_query.json
@@ -190,7 +190,7 @@
          "Age":  [">", 0, "<=", 100 ]
       },
         "results": {
-            "list":["Name","Age","Email"],
+            "list":["Name","Age","Email", "Study"],
             "sort" :"Age"
           }
     }

diff --git a/tests/json_queries.cc b/tests/json_queries.cc
@@ -244,7 +244,15 @@ TEST(QueryHandler, AddAndFind){
         if (cmd =="FindEntity")
             out_query_num++;
 
-        if (j == 12) { // Last FindEntiyu
+        if (j == 11) { // Second Last FindEntity
+            EXPECT_EQ(query["FindEntity"]["entities"][2]["Study"].asString(),
+              "Missing property");
+
+            EXPECT_EQ(query["FindEntity"]["entities"][3]["Study"].asString(),
+              "Missing property");
+        }
+
+        if (j == 12) { // Last FindEntiy
             EXPECT_EQ(query["FindEntity"]["entities"][0]["Birthday"].asString(),
               "1946-10-07T17:59:24-07:00");