IntentAddShellCommandTest.java

/*
 * Copyright (c) 2015 NEC Corporation. All rights reserved.
 *
 * This program and the accompanying materials are made available under the
 * terms of the Eclipse Public License v1.0 which accompanies this distribution,
 * and is available at http://www.eclipse.org/legal/epl-v10.html
 */
package org.opendaylight.nic.cli;

import com.google.gson.Gson;
import org.apache.felix.service.command.CommandSession;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.Mock;
import org.mockito.runners.MockitoJUnitRunner;
import org.opendaylight.nic.api.NicConsoleProvider;
import org.opendaylight.nic.constraints.QualityOfServiceConstraint;
import org.opendaylight.nic.mapped.MappedObject;
import org.opendaylight.nic.mapping.api.IntentMappingService;
import org.opendaylight.yang.gen.v1.urn.opendaylight.intent.rev150122.intent.Actions;
import org.opendaylight.yang.gen.v1.urn.opendaylight.intent.rev150122.intent.Constraints;
import org.opendaylight.yang.gen.v1.urn.opendaylight.intent.rev150122.intent.Subjects;

import java.io.InvalidObjectException;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

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

@RunWith(MockitoJUnitRunner.class)
public class IntentAddShellCommandTest {

    @Mock private NicConsoleProvider nicConsole;
    @Mock private IntentMappingService mappingService;
    @Mock private CommandSession cmdSession;

    private IntentAddShellCommand nicAddShellCmd;

    @Before
    public void setUp() throws Exception {
        nicAddShellCmd = new IntentAddShellCommand(nicConsole, mappingService);
    }

    @Test
    public final void testDoExecute() throws Exception {
        Object feedBack = nicAddShellCmd.doExecute();
            assertNotNull(feedBack);
            assertEquals("As we test a mocked object the feedback should be error message","No subject found with value: any in the mapping service. Try adding it using: intent:map' command", feedBack);
    }

    @Test
    public final void testCreateActions() throws InvalidObjectException {
        List<Actions> actions = nicAddShellCmd.createActions();
        assertNotNull(actions);
        assertEquals("by default at least 1 action should be created", true, actions.size() > 0);
    }

    @Test
    public final void testCreateSubjects() throws InvalidObjectException {

        Map<String, String> resultMap = new HashMap<>();
        resultMap.put("any","0.0.0.0");

        when(mappingService.get("any")).thenReturn(resultMap);
        when(mappingService.get("any")).thenReturn(resultMap);

        List<Subjects> subjects = nicAddShellCmd.createSubjects();
        assertNotNull(subjects);
        assertEquals("there are should be at least 2 subjects", true , subjects.size() > 1);
    }

    @Test
    public final void testExecute() throws Exception {
        assertNotNull(nicAddShellCmd.execute(cmdSession));
    }

    /**
     * Test case for {@link IntentAddShellCommand#createConstraints()}.
     */
    @Test
    public final void testCreateConstraints() throws InvalidObjectException {
        Gson gson = new Gson();

        MappedObject qos = new QualityOfServiceConstraint("HTTP");
        qos.setKey("HTTP");
        String object = gson.toJson(qos, MappedObject.class);

        Map<String, String> resultMap = new HashMap<>();
        resultMap.put("QOS", object);
        resultMap.put("type", "QoSConstraint");
        mappingService.add("QOS", resultMap);

        when(mappingService.get("QOS")).thenReturn(resultMap);
        when(mappingService.get("QOS")).thenReturn(resultMap);

        List<Constraints> constraints = nicAddShellCmd.createConstraints();
        assertNotNull(constraints);
        assertEquals("there are should be at least 1 constraints", true , constraints.size() > 0);
    }

}