/*
 * 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.slider.common.tools

import groovy.transform.CompileStatic
import groovy.util.logging.Slf4j
import org.apache.slider.test.SliderTestBase
import org.junit.Test

@CompileStatic
@Slf4j
class TestExecutionEnvironment extends SliderTestBase {
  
  @Test
  public void testClientEnv() throws Throwable {
    SliderUtils.validateSliderClientEnvironment(log)
  }

  @Test
  public void testWinutils() throws Throwable {
    SliderUtils.maybeVerifyWinUtilsValid();
  }
  
  @Test
  public void testServerEnv() throws Throwable {
    SliderUtils.validateSliderServerEnvironment(log, true)
  }
  
  @Test
  public void testServerEnvNoDependencies() throws Throwable {
    SliderUtils.validateSliderServerEnvironment(log, false)
  }
  
  @Test
  public void testopenSSLEnv() throws Throwable {
    SliderUtils.validateOpenSSLEnv(log)
  }
  
  @Test
  public void testValidatePythonEnv() throws Throwable {
    SliderUtils.validatePythonEnv(log)
  }

  @Test
  public void testNativeLibs() throws Throwable {
    assertNativeLibrariesPresent()
  }
}
