cucumber.runtime.java.spring.GlueCodeContext Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of cucumber-spring Show documentation

There is a newer version: 1.2.6

package cucumber.runtime.java.spring;

import java.util.HashMap;
import java.util.Map;

public class GlueCodeContext {
    public static final GlueCodeContext INSTANCE = new GlueCodeContext();
    private final Map objects = new HashMap();
    private final Map callbacks = new HashMap();
    private int counter;

    private GlueCodeContext() {
    }

    public void start() {
        cleanUp();
        counter++;
    }

    public String getId() {
        return "cucumber_glue_" + counter;
    }

    public void stop() {
        for (Runnable callback : callbacks.values()) {
            callback.run();
        }
        cleanUp();
    }

    public Object get(String name) {
        return objects.get(name);
    }

    public void put(String name, Object object) {
        objects.put(name, object);
    }

    public Object remove(String name) {
        callbacks.remove(name);
        return objects.remove(name);
    }


    private void cleanUp() {
        objects.clear();
        callbacks.clear();
    }


    public void registerDestructionCallback(String name, Runnable callback) {
        callbacks.put(name, callback);
    }
}