Bug fix on creating TestCase inner class

Test: go/vts-web-staging
Bug: 117130291
Change-Id: I688914ef0a41b1b1118999e3438dd08d8bb3b910

3 files changed, 5 insertions, 16 deletions

diff --git a/src/main/java/com/android/vts/api/TestRunRestServlet.java b/src/main/java/com/android/vts/api/TestRunRestServlet.java
index 2c724ac..90fd3f9 100644
--- a/src/main/java/com/android/vts/api/TestRunRestServlet.java
+++ b/src/main/java/com/android/vts/api/TestRunRestServlet.java
@@ -16,20 +16,10 @@
 
 package com.android.vts.api;
 
-import com.android.vts.entity.CodeCoverageEntity;
 import com.android.vts.entity.TestCaseRunEntity;
 import com.android.vts.entity.TestEntity;
 import com.android.vts.entity.TestRunEntity;
-import com.android.vts.util.FilterUtil;
 import com.android.vts.util.TestRunDetails;
-import com.google.appengine.api.datastore.DatastoreService;
-import com.google.appengine.api.datastore.DatastoreServiceFactory;
-import com.google.appengine.api.datastore.Entity;
-import com.google.appengine.api.datastore.EntityNotFoundException;
-import com.google.appengine.api.datastore.FetchOptions;
-import com.google.appengine.api.datastore.Key;
-import com.google.appengine.api.datastore.KeyFactory;
-import com.google.appengine.api.datastore.Query;
 import com.google.gson.Gson;
 import java.io.IOException;
 import java.io.PrintWriter;
@@ -38,7 +28,6 @@ import java.util.List;
 import java.util.Map;
 import java.util.Objects;
 import java.util.logging.Logger;
-import javax.servlet.http.HttpServlet;
 import javax.servlet.http.HttpServletRequest;
 import javax.servlet.http.HttpServletResponse;
 
@@ -108,10 +97,7 @@ public class TestRunRestServlet extends BaseApiServlet {
             }
             Map<com.googlecode.objectify.Key<TestCaseRunEntity>, TestCaseRunEntity>
                     testCaseRunEntityKeyMap = ofy().load().keys(() -> testCaseKeyList.iterator());
-            for (Map.Entry<com.googlecode.objectify.Key<TestCaseRunEntity>, TestCaseRunEntity> entry :
-                    testCaseRunEntityKeyMap.entrySet()) {
-                details.addTestCase(entry.getValue());
-            }
+            testCaseRunEntityKeyMap.forEach((key, value) -> details.addTestCase(value));
         }
         return details;
     }
diff --git a/src/main/java/com/android/vts/config/ObjectifyListener.java b/src/main/java/com/android/vts/config/ObjectifyListener.java
index 5f35abd..02e0215 100644
--- a/src/main/java/com/android/vts/config/ObjectifyListener.java
+++ b/src/main/java/com/android/vts/config/ObjectifyListener.java
@@ -28,6 +28,7 @@ import com.android.vts.entity.ProfilingPointEntity;
 import com.android.vts.entity.ProfilingPointRunEntity;
 import com.android.vts.entity.ProfilingPointSummaryEntity;
 import com.android.vts.entity.RoleEntity;
+import com.android.vts.entity.TestAcknowledgmentEntity;
 import com.android.vts.entity.TestCaseRunEntity;
 import com.android.vts.entity.TestCoverageStatusEntity;
 import com.android.vts.entity.TestEntity;
diff --git a/src/main/java/com/android/vts/entity/TestCaseRunEntity.java b/src/main/java/com/android/vts/entity/TestCaseRunEntity.java
index 30776d4..85a6c43 100644
--- a/src/main/java/com/android/vts/entity/TestCaseRunEntity.java
+++ b/src/main/java/com/android/vts/entity/TestCaseRunEntity.java
@@ -138,7 +138,9 @@ public class TestCaseRunEntity implements DashboardEntity {
     private void onLoad() {
         if (testCaseNames.size() == results.size()) {
             for (int index = 0; index < testCaseNames.size(); index++) {
-                this.addTestCase(testCaseNames.get(index), results.get(index).intValue());
+                String name = testCaseNames.get(index);
+                int result = results.get(index).intValue();
+                this.testCases.add(new TestCase(this.id, this.testCases.size(), name, result));
             }
         }
     }