1 /* 2 * Copyright (C) 2017 The Android Open Source Project 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 package com.android.tradefed.targetprep; 17 18 import com.android.tradefed.build.IBuildInfo; 19 import com.android.tradefed.device.ITestDevice; 20 21 import org.easymock.EasyMock; 22 import org.junit.Before; 23 import org.junit.Test; 24 import org.junit.runner.RunWith; 25 import org.junit.runners.JUnit4; 26 27 /** Unit Tests for {@link RebootTargetPreparer}. */ 28 @RunWith(JUnit4.class) 29 public class RebootTargetPreparerTest { 30 31 private RebootTargetPreparer mRebootTargetPreparer; 32 private ITestDevice mMockDevice; 33 private IBuildInfo mMockBuildInfo; 34 35 @Before 36 public void setUp() { 37 mRebootTargetPreparer = new RebootTargetPreparer(); 38 mMockDevice = EasyMock.createMock(ITestDevice.class); 39 mMockBuildInfo = EasyMock.createMock(IBuildInfo.class); 40 } 41 42 @Test 43 public void testSetUp() throws Exception { 44 mMockDevice.reboot(); 45 EasyMock.expectLastCall().once(); 46 EasyMock.replay(mMockDevice, mMockBuildInfo); 47 48 mRebootTargetPreparer.setUp(mMockDevice, mMockBuildInfo); 49 EasyMock.verify(mMockDevice, mMockBuildInfo); 50 } 51 } 52