InsecureDefaultsServiceBeanTest.java

/**
 * Copyright (c) Codice Foundation
 * <p>
 * This is free software: you can redistribute it and/or modify it under the terms of the GNU Lesser
 * General Public License as published by the Free Software Foundation, either version 3 of the
 * License, or any later version.
 * <p>
 * This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without
 * even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
 * Lesser General Public License for more details. A copy of the GNU Lesser General Public License
 * is distributed along with this program and can be found at
 * <http://www.gnu.org/licenses/lgpl.html>.
 */
package org.codice.ddf.admin.insecure.defaults.service;

import static org.hamcrest.MatcherAssert.assertThat;
import static org.hamcrest.Matchers.is;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;

import java.lang.management.ManagementFactory;
import java.util.ArrayList;
import java.util.List;

import javax.management.MalformedObjectNameException;
import javax.management.ObjectName;

import org.codice.ddf.admin.insecure.defaults.service.Alert.Level;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;

public class InsecureDefaultsServiceBeanTest {

    private static final String BASE_ALERT_TEST_MESSAGE = "This is test alert ";

    private static final String KEYSTORE_SYSTEM_PROPERTY = "javax.net.ssl.keyStore";

    private static final String TRUSTSTORE_SYSTEM_PROPERTY = "javax.net.ssl.trustStore";

    private ObjectName mbeanName;

    @Before
    public void before() throws Exception {
        mbeanName = new ObjectName(InsecureDefaultsServiceBean.MBEAN_NAME);
    }

    @After
    public void after() throws Exception {
        if (isRegistered()) {
            ManagementFactory.getPlatformMBeanServer()
                    .unregisterMBean(mbeanName);
        }
    }

    @Test
    public void validate() {
        // Setup
        final int validatorCount = 1;
        InsecureDefaultsServiceBean bean = createInsecureDefaultsServiceBean(validatorCount);

        // Perform Test
        List<Alert> alerts = bean.validate();

        // Verify
        assertThat(alerts.size(), is(1));
        assertThat(alerts.get(0)
                .getLevel(), is(Level.WARN));
        assertThat(alerts.get(0)
                .getMessage(), is(BASE_ALERT_TEST_MESSAGE + 0));
    }

    // Verify alerts get cleared on each call to validate.
    @Test
    public void validateCalledMultipleTimes() {
        // Setup
        final int validatorCount = 2;
        InsecureDefaultsServiceBean bean = createInsecureDefaultsServiceBean(validatorCount);

        // Call validate twice
        for (int i = 0; i <= 1; i++) {
            // Perform Test
            List<Alert> alerts = bean.validate();

            // Verify
            assertThat(alerts.size(), is(2));
        }
    }

    /**
     * Tests the {@link InsecureDefaultsServiceBean#init()} method
     */
    @Test
    public void testInit() throws Exception {
        InsecureDefaultsServiceBean serviceBean = createInsecureDefaultsServiceBean(0);
        serviceBean.init();
        assertThat(isRegistered(), is(true));
    }

    /**
     * Tests the {@link InsecureDefaultsServiceBean#init()} method for the case where
     * the serviceBean has already been initialized
     */
    @Test
    public void testInitException() throws Exception {
        InsecureDefaultsServiceBean serviceBean = createInsecureDefaultsServiceBean(0);
        serviceBean.init();
        serviceBean.init();
        assertThat(isRegistered(), is(true));
    }

    /**
     * Tests the {@link InsecureDefaultsServiceBean#destroy()} method
     */
    @Test
    public void testDestroy() throws Exception {
        InsecureDefaultsServiceBean serviceBean = createInsecureDefaultsServiceBean(0);
        serviceBean.init();
        assertThat(isRegistered(), is(true));
        serviceBean.destroy();
        assertThat(isRegistered(), is(false));
    }

    /**
     * Tests the {@link InsecureDefaultsServiceBean#destroy()} method for the case where
     * the serviceBean has not yet been initialized.
     */
    @Test
    public void testDestroyException() throws Exception {
        InsecureDefaultsServiceBean serviceBean = createInsecureDefaultsServiceBean(0);
        serviceBean.destroy();
        assertThat(isRegistered(), is(false));
    }

    /**
     * Tests the {@link InsecureDefaultsServiceBean#addKeystoreValidator()} method
     */
    @Test
    public void testAddKeystoreValidator() {
        System.setProperty(KEYSTORE_SYSTEM_PROPERTY, "TestKeystorePath");
        System.setProperty(TRUSTSTORE_SYSTEM_PROPERTY, "TestTruststorePath");

        InsecureDefaultsServiceBean serviceBean = new InsecureDefaultsServiceBean();
        List<Validator> result = serviceBean.getValidators();
        assertThat("Should create nine validators.", result.size(), is(9));
    }

    private boolean isRegistered() throws MalformedObjectNameException {
        return ManagementFactory.getPlatformMBeanServer()
                .isRegistered(mbeanName);
    }

    private List<Validator> createMockValidators(int count) {
        List<Validator> validators = new ArrayList<>(count);
        for (int i = 0; i < count; i++) {
            Validator mockValidator = mock(Validator.class);
            List<Alert> mockAlerts = new ArrayList<>(1);
            mockAlerts.add(new Alert(Level.WARN, BASE_ALERT_TEST_MESSAGE + i));
            when(mockValidator.validate()).thenReturn(mockAlerts);
            validators.add(mockValidator);
        }

        return validators;
    }

    private InsecureDefaultsServiceBean createInsecureDefaultsServiceBean(int validatorCount) {
        InsecureDefaultsServiceBean bean = new InsecureDefaultsServiceBean() {
            @Override
            void addValidators() {
                return;
            }
        };
        bean.setValidators(createMockValidators(validatorCount));

        return bean;
    }
}