diff options
Diffstat (limited to 'src/test/java/com/android/vts/api/DataRestServletTest.java')
-rw-r--r-- | src/test/java/com/android/vts/api/DataRestServletTest.java | 143 |
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"); - } -} |