com.googlecode.gwt.test.MockManager Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of gwt-test-utils Show documentation

The newest version!

package com.googlecode.gwt.test;

import com.googlecode.gwt.test.internal.AfterTestCallback;
import com.googlecode.gwt.test.internal.AfterTestCallbackManager;

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

class MockManager implements AfterTestCallback {

    private static final MockManager INSTANCE = new MockManager();

    public static MockManager get() {
        return INSTANCE;
    }

    private Map, Object> mockObjects = new HashMap<>();

    private MockManager() {
        AfterTestCallbackManager.get().registerCallback(this);
    }

     T registerMock(Class clazz, T mock) {
        mockObjects.put(clazz, mock);
        return mock;
    }

    void registerMocks(Map, Object> mocks) {
        mockObjects.putAll(mocks);
    }

     T getMock(Class clazz) {
        return (T) mockObjects.get(clazz);
    }

    Map, Object> getAllMocksByType() {
        return mockObjects;
    }

    @Override
    public void afterTest() {
        this.mockObjects.clear();
    }
}