leader-common/src/test/kotlin/org/apache/amaterasu/leader/common/configuration/ConfigManagerTests.kt

/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You 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.apache.amaterasu.leader.common.configuration

import org.jetbrains.spek.api.Spek
import org.jetbrains.spek.api.dsl.given
import org.jetbrains.spek.api.dsl.it
import org.jetbrains.spek.api.dsl.on
import java.io.File
import kotlin.test.assertEquals

class ConfigManagerTests : Spek({

    val marker = this.javaClass.getResource("/maki.yml")!!.path

    given("a ConfigManager for a job ") {

        val repoPath = "${File(marker).parent}/test_repo"
        val cfg = ConfigManager("test", repoPath)

        it("loads the job level environment"){
            assertEquals(cfg.config[Job.master] , "yarn")
        }

        on("getting an env for an action with default path") {
            val startConf = cfg.getActionConfiguration("start")
            it("loads the specific configuration defined in the actions folder"){
                assertEquals(startConf[Job.master] , "mesos")
            }
        }

        on("getting an env for an action with a conf: property in the maki.yml"){
            val step2conf = cfg.getActionConfiguration("step2", "src/{action_name}/{env}/")

            it("loads the specific configuration defined in the actions folder"){
                assertEquals(step2conf[Job.name] , "test2")
            }
        }

        on("getting an env for an action with no action level config"){
            val step3conf = cfg.getActionConfiguration("step3")

            it("loads only the job level conf"){
                assertEquals(step3conf[Job.name] , "test")
            }
        }

        on("receiving a path to a specific file" ){
            val step4conf = cfg.getActionConfiguration("step4", "src/start/env/{env}/job.yml")

            it("loads the specific configuration from the file"){
                assertEquals(step4conf[Job.master] , "mesos")
            }
        }


    }

    given("a ConfigManager for a job with spark framework") {

        val repoPath = "${File(marker).parent}/spark_repo"
        val cfg = ConfigManager("test", repoPath, listOf("sparkConfiguration"))

        it("load the framework configuration for spark"){
            val spark: Map<String, String> = cfg.config["sparkConfiguration"]
            assertEquals(spark["spark.executor.memory"], "1g")
        }
    }
})