StorageConfigurationTest.java

/*
 * Copyright 2015-2016 OpenCB
 *
 * 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.opencb.opencga.storage.core.config;

import org.junit.Test;
import org.opencb.commons.datastore.core.ObjectMap;

import java.io.*;
import java.nio.file.Paths;
import java.util.Arrays;
import java.util.Collections;

/**
 * Created by imedina on 01/05/15.
 */
public class StorageConfigurationTest {

    @Test
    public void testDefault() throws Exception {
        StorageConfiguration storageConfiguration = new StorageConfiguration();

//        Map<String, String> options = new HashMap<>();
        ObjectMap options = new ObjectMap();
        options.put("key", "defaultValue");

        StorageEngineConfiguration storageEngineConfiguration1 = new StorageEngineConfiguration(
                "mongodb",
                new StorageEtlConfiguration("org.opencb.opencga.storage.mongodb.alignment.MongoDBAlignmentStorageManager", new ObjectMap
                        (), new DatabaseCredentials(Arrays.asList("mongodb-dev:27017"), "user", "password")),
                new StorageEtlConfiguration("org.opencb.opencga.storage.mongodb.alignment.MongoDBVariantStorageManager", new ObjectMap(),
                        new DatabaseCredentials(Arrays.asList("mongodb-dev:27017"), "user", "password")),
                options);

        StorageEngineConfiguration storageEngineConfiguration2 = new StorageEngineConfiguration(
                "hadoop",
                new StorageEtlConfiguration("org.opencb.opencga.storage.hadoop.alignment.HadoopAlignmentStorageManager", new ObjectMap(),
                        new DatabaseCredentials(Arrays.asList("who-master:60000"), "user", "password")),
                new StorageEtlConfiguration("org.opencb.opencga.storage.hadoop.alignment.HadoopVariantStorageManager", new ObjectMap(),
                        new DatabaseCredentials(Arrays.asList("who-master:60000"), "user", "password")),
                options);


        CellBaseConfiguration cellBaseConfiguration = new CellBaseConfiguration(Arrays.asList("localhost"), "v3", new DatabaseCredentials
                (Arrays.asList("localhost"), "user", "password"));
        ServerConfiguration serverConfiguration =
                new ServerConfiguration(9090, 9091, "mongodb", Arrays.asList("localhost"), Collections.emptyMap());

        storageConfiguration.setDefaultStorageEngineId("mongodb");

        storageConfiguration.setCellbase(cellBaseConfiguration);
        storageConfiguration.setServer(serverConfiguration);

        storageConfiguration.getStorageEngines().add(storageEngineConfiguration1);
        storageConfiguration.getStorageEngines().add(storageEngineConfiguration2);

        File file = Paths.get("/tmp/storage-configuration-test.yml").toFile();
        try (FileOutputStream os = new FileOutputStream(file)) {
            storageConfiguration.serialize(os);
        }
        try (FileInputStream is = new FileInputStream(file)) {
            StorageConfiguration.load(is, "yml");
        }
    }

    @Test
    public void testLoad() throws Exception {
        StorageConfiguration storageConfiguration = StorageConfiguration.load(getClass().getResource("/storage-configuration-test.yml")
                .openStream());
        System.out.println("storageConfiguration = " + storageConfiguration);
    }


}