summaryrefslogtreecommitdiff
path: root/src/test/java/com/android/vts/api/DataRestServletTest.java
diff options
context:
space:
mode:
Diffstat (limited to 'src/test/java/com/android/vts/api/DataRestServletTest.java')
-rw-r--r--src/test/java/com/android/vts/api/DataRestServletTest.java143
1 files changed, 0 insertions, 143 deletions
diff --git a/src/test/java/com/android/vts/api/DataRestServletTest.java b/src/test/java/com/android/vts/api/DataRestServletTest.java
deleted file mode 100644
index b01c761..0000000
--- a/src/test/java/com/android/vts/api/DataRestServletTest.java
+++ /dev/null
@@ -1,143 +0,0 @@
-/*
- * Copyright (C) 2018 The Android Open Source Project
- *
- * Licensed under the Apache License, Version 2.0 (the "License"); you
- * may not use this file except in compliance with the License. You may
- * obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
- * implied. See the License for the specific language governing
- * permissions and limitations under the License.
- */
-
-package com.android.vts.api;
-
-import com.android.vts.entity.BranchEntity;
-import com.android.vts.entity.BuildTargetEntity;
-import com.android.vts.entity.DeviceInfoEntity;
-import com.android.vts.entity.TestEntity;
-import com.android.vts.entity.TestRunEntity;
-import com.android.vts.util.ObjectifyTestBase;
-import com.google.gson.Gson;
-import com.googlecode.objectify.Key;
-import org.junit.jupiter.api.BeforeEach;
-import org.junit.jupiter.api.Test;
-import org.mockito.Mock;
-import org.mockito.Spy;
-
-import javax.servlet.ServletException;
-import javax.servlet.http.HttpServletRequest;
-import javax.servlet.http.HttpServletResponse;
-import java.io.IOException;
-import java.io.PrintWriter;
-import java.io.StringWriter;
-import java.util.LinkedList;
-
-import static com.googlecode.objectify.ObjectifyService.factory;
-import static org.junit.Assert.assertEquals;
-import static org.mockito.Mockito.when;
-
-public class DataRestServletTest extends ObjectifyTestBase {
-
- private Gson gson;
-
- @Spy private DataRestServlet servlet;
-
- @Mock private HttpServletRequest request;
-
- @Mock private HttpServletResponse response;
-
- /** It be executed before each @Test method */
- @BeforeEach
- void setUpExtra() {
- gson = new Gson();
-
- factory().register(TestEntity.class);
- factory().register(TestRunEntity.class);
- factory().register(BranchEntity.class);
- factory().register(BuildTargetEntity.class);
- factory().register(DeviceInfoEntity.class);
-
- BranchEntity branchEntity1 = new BranchEntity("master");
- branchEntity1.save();
- BranchEntity branchEntity2 = new BranchEntity("pi");
- branchEntity2.save();
-
- BuildTargetEntity buildTargetEntity1 = new BuildTargetEntity("aosp_arm64_ab-userdebug");
- buildTargetEntity1.save();
- BuildTargetEntity buildTargetEntity2 = new BuildTargetEntity("sailfish-userdebug");
- buildTargetEntity2.save();
-
- Key testParentKey = Key.create(TestEntity.class, "test1");
- Key testRunParentKey = Key.create(testParentKey, TestRunEntity.class, 1);
- DeviceInfoEntity deviceInfoEntity1 =
- new DeviceInfoEntity(
- testRunParentKey,
- "pi",
- "sailfish",
- "sailfish-userdebug",
- "4585723",
- "64",
- "arm64-v8a");
- deviceInfoEntity1.setId(2384723984L);
- deviceInfoEntity1.save();
-
- DeviceInfoEntity deviceInfoEntity2 =
- new DeviceInfoEntity(
- testRunParentKey,
- "master",
- "walleye",
- "aosp_arm64_ab-userdebug",
- "4585723",
- "64",
- "arm64-v8a");
- deviceInfoEntity2.setId(2384723422L);
- deviceInfoEntity2.save();
- }
-
- @Test
- public void testBranchData() throws IOException, ServletException {
-
- when(request.getPathInfo()).thenReturn("/branch");
- when(request.getParameter("schKey")).thenReturn("*");
-
- StringWriter sw = new StringWriter();
- PrintWriter pw = new PrintWriter(sw);
-
- when(response.getWriter()).thenReturn(pw);
-
- servlet.doGet(request, response);
- String result = sw.getBuffer().toString().trim();
-
- LinkedList resultList = gson.fromJson(result, LinkedList.class);
-
- assertEquals(resultList.size(), 2);
- assertEquals(resultList.get(0), "master");
- assertEquals(resultList.get(1), "pi");
- }
-
- @Test
- public void testDeviceData() throws IOException, ServletException {
-
- when(request.getPathInfo()).thenReturn("/device");
- when(request.getParameter("schKey")).thenReturn("*");
-
- StringWriter sw = new StringWriter();
- PrintWriter pw = new PrintWriter(sw);
-
- when(response.getWriter()).thenReturn(pw);
-
- servlet.doGet(request, response);
- String result = sw.getBuffer().toString().trim();
-
- LinkedList resultList = gson.fromJson(result, LinkedList.class);
-
- assertEquals(resultList.size(), 2);
- assertEquals(resultList.get(0), "aosp_arm64_ab-userdebug");
- assertEquals(resultList.get(1), "sailfish-userdebug");
- }
-}