1 /*
2  * Copyright (C) 2016 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 
17 package com.android.compatibility.common.tradefed.targetprep;
18 
19 import com.android.compatibility.common.util.MonitoringUtils;
20 import com.android.tradefed.device.DeviceNotAvailableException;
21 import com.android.tradefed.device.ITestDevice;
22 import com.android.tradefed.log.LogUtil.CLog;
23 import com.android.tradefed.suite.checker.ISystemStatusChecker;
24 import com.android.tradefed.suite.checker.StatusCheckerResult;
25 import com.android.tradefed.suite.checker.StatusCheckerResult.CheckStatus;
26 
27 /**
28  * Checks network connectivity status on device after module execution.
29  */
30 public class NetworkConnectivityChecker implements ISystemStatusChecker {
31 
32     // Only report is as failed (capture bugreport) when status goes from pass-> fail
33     private boolean mIsFailed = false;
34 
35     /**
36      * {@inheritDoc}
37      */
38     @Override
postExecutionCheck(ITestDevice device)39     public StatusCheckerResult postExecutionCheck(ITestDevice device) throws DeviceNotAvailableException {
40         if (!MonitoringUtils.checkDeviceConnectivity(device)) {
41             if (mIsFailed) {
42                 CLog.w("NetworkConnectivityChecker is still failing on %s.",
43                         device.getSerialNumber());
44                 return new StatusCheckerResult(CheckStatus.SUCCESS);
45             }
46             mIsFailed = true;
47             return new StatusCheckerResult(CheckStatus.FAILED);
48         }
49         mIsFailed = false;
50         return new StatusCheckerResult(CheckStatus.SUCCESS);
51     }
52 }
53