/*
 * Copyright (C) 2017 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.tradefed.device;

import static org.junit.Assert.*;
import static org.mockito.ArgumentMatchers.anyLong;
import static org.mockito.ArgumentMatchers.eq;
import static org.mockito.Mockito.when;

import com.android.tradefed.util.CommandResult;
import com.android.tradefed.util.CommandStatus;
import com.android.tradefed.util.IRunUtil;

import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.junit.runners.JUnit4;
import org.mockito.Mock;
import org.mockito.MockitoAnnotations;

/** Unit tests for {@link AndroidDebugBridgeWrapper}. */
@RunWith(JUnit4.class)
public class AndroidDebugBridgeWrapperTest {

    private AndroidDebugBridgeWrapper mBridge;
    @Mock IRunUtil mMockRunUtil;

    @Before
    public void setUp() {
        MockitoAnnotations.initMocks(this);

        mBridge =
                new AndroidDebugBridgeWrapper() {
                    @Override
                    IRunUtil getRunUtil() {
                        return mMockRunUtil;
                    }
                };
    }

    /** Test parsing the adb version when available. */
    @Test
    public void testAdbVersionParsing() {
        CommandResult res = new CommandResult();
        res.setStatus(CommandStatus.SUCCESS);
        res.setStdout("Android Debug Bridge version 1.0.36\nRevision 0e7324e9095a-android\n");
        when(mMockRunUtil.runTimedCmd(anyLong(), eq("fakeadb"), eq("version"))).thenReturn(res);

        String version = mBridge.getAdbVersion("fakeadb");
        assertEquals("1.0.36-0e7324e9095a-android", version);
    }

    /** Test parsing the adb version when available with alternative format. */
    @Test
    public void testAdbVersionParsing_altFormat() {
        CommandResult res = new CommandResult();
        res.setStatus(CommandStatus.SUCCESS);
        res.setStdout(
                "Android Debug Bridge version 1.0.36\n"
                        + "Version 0.0.0-4407735\n"
                        + "Installed as /usr/local/adb\n");
        when(mMockRunUtil.runTimedCmd(anyLong(), eq("fakeadb"), eq("version"))).thenReturn(res);

        String version = mBridge.getAdbVersion("fakeadb");
        assertEquals("1.0.36 subVersion: 0.0.0-4407735 install path: /usr/local/adb", version);
    }

    /** Test when the version process fails. */
    @Test
    public void testAdbVersionParse_runFail() {
        CommandResult res = new CommandResult();
        res.setStatus(CommandStatus.FAILED);
        res.setStdout("");
        res.setStderr("");
        when(mMockRunUtil.runTimedCmd(anyLong(), eq("fakeadb"), eq("version"))).thenReturn(res);

        assertNull(mBridge.getAdbVersion("fakeadb"));
    }

    /** Test when the revision is not present, in that case we output a partial version. */
    @Test
    public void testAdbVersionParsing_partial() {
        CommandResult res = new CommandResult();
        res.setStatus(CommandStatus.SUCCESS);
        res.setStdout("Android Debug Bridge version 1.0.36\n");
        when(mMockRunUtil.runTimedCmd(anyLong(), eq("fakeadb"), eq("version"))).thenReturn(res);

        String version = mBridge.getAdbVersion("fakeadb");
        assertEquals("1.0.36", version);
    }

    /** Test when the output from 'adb version' is not as expected at all. */
    @Test
    public void testAdbVersionParsing_parseFail() {
        CommandResult res = new CommandResult();
        res.setStatus(CommandStatus.SUCCESS);
        res.setStdout("This is probably not the right output\n");
        when(mMockRunUtil.runTimedCmd(anyLong(), eq("fakeadb"), eq("version"))).thenReturn(res);

        assertNull(mBridge.getAdbVersion("fakeadb"));
    }
}