com.gopivotal.cloudfoundry.test.support.runner.MethodInvoker.java Source code

Introduction

Here is the source code for com.gopivotal.cloudfoundry.test.support.runner.MethodInvoker.java

Source

/*
 * Copyright 2013 the original author or authors.
 *
 * 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 com.gopivotal.cloudfoundry.test.support.runner;

import com.gopivotal.cloudfoundry.test.support.application.Application;
import com.gopivotal.cloudfoundry.test.support.application.ApplicationFactory;
import com.gopivotal.cloudfoundry.test.support.service.ServicesHolder;
import com.gopivotal.cloudfoundry.test.support.util.TcfUtils;
import org.junit.runners.model.FrameworkMethod;
import org.junit.runners.model.Statement;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.context.ApplicationContext;
import org.springframework.test.context.TestContext;
import org.springframework.test.context.TestExecutionListener;

import java.util.Arrays;
import java.util.Map;
import java.util.TreeMap;

final class MethodInvoker extends Statement implements TestExecutionListener {

    private final Logger logger = LoggerFactory.getLogger(this.getClass());

    private final FrameworkMethod frameworkMethod;

    private final Object instance;

    private final Object monitor = new Object();

    private final String name;

    private volatile TestContext testContext;

    MethodInvoker(FrameworkMethod frameworkMethod, Object instance, String name) {
        this.frameworkMethod = frameworkMethod;
        this.instance = instance;
        this.name = name;
    }

    @Override
    public void evaluate() throws Throwable {
        ApplicationContext applicationContext = getApplicationContext();

        Application application = null;
        try {
            application = applicationContext.getBean(ApplicationFactory.class).create(this.name);
            application.push().bind(applicationContext.getBean(ServicesHolder.class).get());
            application.start().getTestOperations().waitForStart();

            this.frameworkMethod.invokeExplosively(this.instance, application);
        } catch (Exception e) {
            if (application != null) {
                this.logger.error(getCrashLogs(application));
            }

            throw e;
        } finally {
            TcfUtils.deleteQuietly(application);
        }
    }

    private ApplicationContext getApplicationContext() {
        synchronized (this.monitor) {
            return this.testContext.getApplicationContext();
        }
    }

    @Override
    public void beforeTestClass(TestContext testContext) {
    }

    @Override
    public void prepareTestInstance(TestContext testContext) {
    }

    @Override
    public void beforeTestMethod(TestContext testContext) {
        synchronized (this.monitor) {
            this.testContext = testContext;
        }
    }

    @Override
    public void afterTestMethod(TestContext testContext) {
    }

    @Override
    public void afterTestClass(TestContext testContext) {
    }

    private String getCrashLogs(Application application) {
        StringBuilder sb = new StringBuilder("Crash Logs:\n\n");

        Map<String, String> crashLogs = new TreeMap<>(application.getCrashLogs());
        for (Map.Entry<String, String> entry : crashLogs.entrySet()) {
            String key = entry.getKey();
            sb.append(key).append('\n');
            sb.append(separator(key.length())).append('\n');

            String value = entry.getValue();
            if (value != null) {
                sb.append(entry.getValue());
            }
            sb.append('\n');
        }

        return sb.toString();
    }

    private String separator(int length) {
        char[] separator = new char[length];
        Arrays.fill(separator, '=');
        return new String(separator);
    }

}