GetServerSSHKeyFingerprintQueryTest.java

package org.ovirt.engine.core.bll;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertNull;
import static org.mockito.Mockito.doReturn;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;

import org.junit.Before;
import org.junit.Test;
import org.ovirt.engine.core.bll.utils.EngineSSHClient;
import org.ovirt.engine.core.common.queries.ServerParameters;

public class GetServerSSHKeyFingerprintQueryTest extends AbstractQueryTest<ServerParameters, GetServerSSHKeyFingerprintQuery<ServerParameters>> {

    String serverName = "localhost";
    String fingerPrint = "b5:ad:16:19:06:9f:b3:41:69:eb:1c:42:1d:12:b5:31";

    EngineSSHClient mockEngineSSHClient;

    private void setupMock() throws Exception {
        mockEngineSSHClient = mock(EngineSSHClient.class);
        doReturn(mockEngineSSHClient).when(getQuery()).getEngineSSHClient();
    }

    @Before
    @Override
    public void setUp() throws Exception {
        super.setUp();
        setupMock();
    }

    @Test
    public void testExecuteQueryCommnad() throws Exception {
        when(getQueryParameters().getServer()).thenReturn(serverName);
        doReturn(fingerPrint).when(mockEngineSSHClient).getHostFingerprint();
        getQuery().executeQueryCommand();
        String serverFingerprint = getQuery().getQueryReturnValue().getReturnValue();

        assertNotNull(serverFingerprint);
        assertEquals(fingerPrint, serverFingerprint);
    }

    @Test
    public void testExecuteQueryCommnadFails() {
        getQuery().executeQueryCommand();
        String serverFingerprint = getQuery().getQueryReturnValue().getReturnValue();

        assertNull(serverFingerprint);
    }
}