com.klarna.hiverunner.config.HiveRunnerConfig Java Examples

private TestRule getHiveRunnerConfigRule(Object target) {
    return new TestRule() {
        @Override
        public Statement apply(Statement base, Description description) {
            Set<Field> fields = ReflectionUtils.getAllFields(target.getClass(),
                    Predicates.and(
                            withAnnotation(HiveRunnerSetup.class),
                            withType(HiveRunnerConfig.class)));

            Preconditions.checkState(fields.size() <= 1,
                    "Exact one field of type HiveRunnerConfig should to be annotated with @HiveRunnerSetup");

            /*
             Override the config with test case config. Taking care to not replace the config instance since it
              has been passes around and referenced by some of the other test rules.
              */
            if (!fields.isEmpty()) {
                config.override(ReflectionUtils
                        .getFieldValue(target, fields.iterator().next().getName(), HiveRunnerConfig.class));
            }

            return base;
        }
    };
}
 

/**
 * Traverses the test case annotations. Will inject a HiveShell in the test case that envelopes the HiveServer.
 */
HiveShellContainer createHiveServerContainer(List<? extends Script> scripts, Object testCase,
    Path baseDir, HiveRunnerConfig config)
    throws IOException {

  HiveServerContext context = new StandaloneHiveServerContext(baseDir, config);

  return buildShell(scripts, testCase, config, context);
}
 

@Before
public void setup() throws IOException {
    basedir = Files.createTempDirectory("HiveServerContainerTest");
    StandaloneHiveServerContext context = new StandaloneHiveServerContext(basedir, new HiveRunnerConfig());
    container = new HiveServerContainer(context);
    container.init(new HashMap<>(), new HashMap<>());
}
 

public static TestRule create(final HiveRunnerConfig config, final Object target) {
    return new TestRule() {
        @Override
        public Statement apply(Statement base, Description description) {
            return new ThrowOnTimeout(base, config, target);
        }
    };
}
 

private void setupConfig(Object target) {
  Set<Field> fields = ReflectionUtils.getAllFields(target.getClass(),
      Predicates.and(
          withAnnotation(HiveRunnerSetup.class),
          withType(HiveRunnerConfig.class)));

  Preconditions.checkState(fields.size() <= 1,
      "Only one field of type HiveRunnerConfig should be annotated with @HiveRunnerSetup");

  if (!fields.isEmpty()) {
    config.override(ReflectionUtils
        .getFieldValue(target, fields.iterator().next().getName(), HiveRunnerConfig.class));
  }
}
 

private HiveShellContainer buildShell(List<? extends Script> scripts, Object testCase, HiveRunnerConfig config,
    HiveServerContext context) throws IOException {
  HiveServerContainer hiveTestHarness = new HiveServerContainer(context);

  HiveShellBuilder hiveShellBuilder = new HiveShellBuilder();
  hiveShellBuilder.setCommandShellEmulation(config.getCommandShellEmulator());

  HiveShellField shellSetter = loadScriptUnderTest(testCase, hiveShellBuilder);
  if (scripts != null) {
    hiveShellBuilder.overrideScriptsUnderTest(scripts);
  }

  hiveShellBuilder.setHiveServerContainer(hiveTestHarness);

  loadAnnotatedResources(testCase, hiveShellBuilder);

  loadAnnotatedProperties(testCase, hiveShellBuilder);

  loadAnnotatedSetupScripts(testCase, hiveShellBuilder);

  // Build shell
  HiveShellContainer shell = hiveShellBuilder.buildShell();

  // Set shell
  shellSetter.setShell(shell);

  if (shellSetter.isAutoStart()) {
    shell.start();
  }
  return shell;
}
 

@Override
public void setCommandShellEmulation(HiveShellBuilder builder, HiveRunnerConfig config) throws Exception {
	Method method = HiveRunnerConfig.class.getDeclaredMethod("getCommandShellEmulation");
	Object emulation = method.invoke(config);
	Class emulationClz = Class.forName("com.klarna.hiverunner.CommandShellEmulation");
	method = HiveShellBuilder.class.getDeclaredMethod("setCommandShellEmulation", emulationClz);
	method.invoke(builder, emulation);
}
 

@Override
public void setCommandShellEmulation(HiveShellBuilder builder, HiveRunnerConfig config) throws Exception {
	Method method = config.getClass().getDeclaredMethod("getCommandShellEmulator");
	Object emulator = method.invoke(config);
	Class emulatorClz = Class.forName("com.klarna.hiverunner.sql.cli.CommandShellEmulator");
	method = builder.getClass().getDeclaredMethod("setCommandShellEmulation", emulatorClz);
	method.invoke(builder, emulator);
}
 

@Override
public void setCommandShellEmulation(HiveShellBuilder builder, HiveRunnerConfig config) throws Exception {
	Method method = HiveRunnerConfig.class.getDeclaredMethod("getCommandShellEmulation");
	Object emulation = method.invoke(config);
	Class emulationClz = Class.forName("com.klarna.hiverunner.CommandShellEmulation");
	method = HiveShellBuilder.class.getDeclaredMethod("setCommandShellEmulation", emulationClz);
	method.invoke(builder, emulation);
}
 

@Override
public void setCommandShellEmulation(HiveShellBuilder builder, HiveRunnerConfig config) throws Exception {
	Method method = config.getClass().getDeclaredMethod("getCommandShellEmulator");
	Object emulator = method.invoke(config);
	Class emulatorClz = Class.forName("com.klarna.hiverunner.sql.cli.CommandShellEmulator");
	method = builder.getClass().getDeclaredMethod("setCommandShellEmulation", emulatorClz);
	method.invoke(builder, emulator);
}
 

FlinkStandaloneHiveServerContext(TemporaryFolder basedir, HiveRunnerConfig hiveRunnerConfig, int hmsPort) {
	this.basedir = basedir;
	this.hiveRunnerConfig = hiveRunnerConfig;
	this.hmsPort = hmsPort;
}
 

protected HiveRunnerConfig getHiveRunnerConfig() {
  return config;
}
 

public ThrowOnTimeout(Statement originalStatement, HiveRunnerConfig config, Object target) {
    this.originalStatement = originalStatement;
    this.config = config;
    this.target = target;
}
 

public StandaloneHiveServerContext(Path basedir, HiveRunnerConfig hiveRunnerConfig) {
    this.basedir = basedir;
    this.hiveRunnerConfig = hiveRunnerConfig;
}
 

FlinkStandaloneHiveServerContext(TemporaryFolder basedir, HiveRunnerConfig hiveRunnerConfig, int hmsPort) {
	this.basedir = basedir;
	this.hiveRunnerConfig = hiveRunnerConfig;
	this.hmsPort = hmsPort;
}
 

/**
 * Sets CommandShellEmulation for HiveShellBuilder.
 */
void setCommandShellEmulation(HiveShellBuilder builder, HiveRunnerConfig config) throws Exception;
 

/**
 * Sets CommandShellEmulation for HiveShellBuilder.
 */
void setCommandShellEmulation(HiveShellBuilder builder, HiveRunnerConfig config) throws Exception;