diff options
Diffstat (limited to 'src/main/java/com/android/vts/api/DatastoreRestServlet.java')
| -rw-r--r-- | src/main/java/com/android/vts/api/DatastoreRestServlet.java | 10 |
1 files changed, 3 insertions, 7 deletions
diff --git a/src/main/java/com/android/vts/api/DatastoreRestServlet.java b/src/main/java/com/android/vts/api/DatastoreRestServlet.java index fb5ac65..fa0ae65 100644 --- a/src/main/java/com/android/vts/api/DatastoreRestServlet.java +++ b/src/main/java/com/android/vts/api/DatastoreRestServlet.java @@ -379,15 +379,13 @@ public class DatastoreRestServlet extends BaseApiServlet { List listEntity = (List) entity; if (listEntity.size() > 0 && listEntity.get(0) instanceof TestCaseRunEntity) { - List<TestCaseRunEntity> dashboardEntityList = - (List<TestCaseRunEntity>) entity; Map< com.googlecode.objectify.Key< TestCaseRunEntity>, TestCaseRunEntity> testCaseRunEntityMap = DashboardEntity.saveAll( - dashboardEntityList, + testCaseRunEntityList, this .MAX_ENTITY_SIZE_PER_TRANSACTION); @@ -401,11 +399,9 @@ public class DatastoreRestServlet extends BaseApiServlet { .collect(Collectors.toList()); } else if (listEntity.size() > 0 && listEntity.get(0) instanceof TestRunEntity) { - List<TestRunEntity> dashboardEntityList = - (List<TestRunEntity>) entity; - dashboardEntityList.get(0).setTestCaseIds(testCaseIds); + testRunEntityList.get(0).setTestCaseIds(testCaseIds); DashboardEntity.saveAll( - dashboardEntityList, + testRunEntityList, this.MAX_ENTITY_SIZE_PER_TRANSACTION); } else { List<DashboardEntity> dashboardEntityList = |