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

Go to download

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

Serenity core libraries

There is a newer version: 4.2.8

package net.serenitybdd.core.sessions;

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

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 V put(K key, V 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();
    }
}