net.serenitybdd.core.sessions.TestSessionVariables Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of serenity-core Show documentation

There is a newer version: 4.2.1

package net.serenitybdd.core.sessions;

import net.serenitybdd.core.collect.NewMap;
import net.serenitybdd.core.SessionMap;

import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;

public class TestSessionVariables extends ConcurrentHashMap implements SessionMap {

    private final Map metadata = new ConcurrentHashMap();

    @Override
    public void shouldContainKey(Object key) {
        Object result = super.get(key);
        if (result == null) {
            throw new AssertionError("Session variable " + key + " expected but not found.");
        }
    }


    @Override
    public Object put(Object key, Object value) {
        if (value == null) {
            return remove(key);
        } else {
            return super.put(key, value);
        }
    }



    @Override
    public Map getMetaData() {
        return NewMap.copyOf(metadata);
    }

    @Override
    public void addMetaData(String key, String value) {
        metadata.put(key, value);
    }

    @Override
    public void clearMetaData() {
        metadata.clear();
    }

    @Override
    public void clear() {
        clearMetaData();
        super.clear();
    }
}