org.midonet.config.TestHostIdGenerator.java Source code

Introduction

Here is the source code for org.midonet.config.TestHostIdGenerator.java

Source

/*
 * Copyright 2014 Midokura SARL
 *
 * 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 org.midonet.config;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.util.Arrays;
import java.util.Properties;
import java.util.UUID;

import org.apache.commons.configuration.HierarchicalConfiguration;
import org.junit.After;
import org.junit.Assert;
import org.junit.Before;
import org.junit.Test;

public class TestHostIdGenerator {

    static final String localPropertiesFile = "host_uuid.properties";
    static final String uuidPropertyName = "host_uuid";
    static final String hostId1 = "e3f9adc0-5175-11e1-b86c-0800200c9a66";
    static final String hostId2 = "e3f9adc0-5175-11e1-b86c-0800200c9a67";
    File propFile;
    HostIdConfig config;
    HostIdConfig configFake;

    @ConfigGroup(HostConfig.GROUP_NAME)
    public interface HostConfig extends HostIdConfig {

        public static final String GROUP_NAME = "host";

        /**
         * Gets a unique identifier for this host.
         * @return The identifier.
         */
        @Override
        @ConfigString(key = "host_uuid", defaultValue = "")
        String getHostId();

        /**
         * Gets the path of the host properties file.
         * @return The file path.
         */
        @Override
        @ConfigString(key = "properties_file", defaultValue = localPropertiesFile)
        String getHostPropertiesFilePath();
    }

    @After
    public void tearDown() throws Exception {
        if (propFile.exists())
            propFile.delete();
    }

    @Before
    public void setUp() throws Exception {
        final HierarchicalConfiguration fakeConfiguration = new HierarchicalConfiguration();
        fakeConfiguration.addNodes(HostConfig.GROUP_NAME,
                Arrays.asList(new HierarchicalConfiguration.Node("properties_file", localPropertiesFile)));
        final HierarchicalConfiguration configuration = new HierarchicalConfiguration();

        // this configuration will use the default hostid properties location.
        configuration.addNodes(HostConfig.GROUP_NAME,
                Arrays.asList(new HierarchicalConfiguration.Node(uuidPropertyName, hostId1)));
        propFile = new File(localPropertiesFile);
        config = ConfigProvider.providerForIniConfig(configuration).getConfig(HostConfig.class);
        configFake = ConfigProvider.providerForIniConfig(fakeConfiguration).getConfig(HostConfig.class);

        Properties properties = new Properties();
        properties.setProperty(uuidPropertyName, hostId2);
        properties.store(new FileOutputStream(localPropertiesFile), null);
    }

    @Test
    public void getIdFromConfFile() throws Exception {
        UUID id = HostIdGenerator.getHostId(config);
        Assert.assertEquals(id.toString(), hostId1);
    }

    @Test
    public void getIdFromPropertyFile() throws Exception {
        // pass a conf file with no ID specified so we will try to read it
        // from the properties file
        UUID id = HostIdGenerator.getHostId(configFake);
        Assert.assertTrue(id.toString().equals(hostId2));
    }

    @Test
    public void generateRandomId() throws Exception {
        // delete properties file
        boolean res = propFile.delete();
        Assert.assertTrue(res);
        UUID id = HostIdGenerator.getHostId(configFake);
        // check that the id has been written in the property file
        boolean exists = propFile.exists();
        Assert.assertTrue(exists);
        Properties properties = new Properties();
        properties.load(new FileInputStream(localPropertiesFile));
        UUID idFromProperty = UUID.fromString(properties.getProperty(uuidPropertyName));
        Assert.assertTrue(id.equals(idFromProperty));
    }

    @Test(expected = HostIdGenerator.PropertiesFileNotWritableException.class)
    public void propertyFileCorrupted() throws Exception {
        // delete properties file so no ID will be loaded from there
        boolean res = propFile.delete();
        propFile.createNewFile();
        propFile.setReadOnly();
        UUID id = HostIdGenerator.getHostId(configFake);
    }
}