TestContext.java

/*
 * Copyright (C) 2011 Benoit GUEROUT <bguerout at gmail dot com> and Yves AMSELLEM <amsellem dot yves at gmail dot com>
 *
 * 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 org.jongo.util.compatibility;

import org.jongo.Mapper;
import org.jongo.util.JongoTestBase;
import org.reflections.Reflections;

import java.util.*;

public class TestContext {

    private final String contextName;
    private final Mapper mapper;
    private final List<Class<? extends JongoTestBase>> ignoredTestCases;
    private final Map<Class<? extends JongoTestBase>, String> ignoredTests;

    public TestContext(String contextName, Mapper mapper) {
        this.contextName = contextName;
        this.mapper = mapper;
        this.ignoredTestCases = new ArrayList<Class<? extends JongoTestBase>>();
        this.ignoredTests = new HashMap<Class<? extends JongoTestBase>, String>();
    }

    public String getContextName() {
        return contextName;
    }

    public boolean mustIgnoreTestCase(Class<?> clazz) {
        return ignoredTestCases.contains(clazz);
    }

    public void ignoreTestCase(Class<? extends JongoTestBase> clazz) {
        ignoredTestCases.add(clazz);
    }

    public boolean mustIgnoreTest(Class<?> clazz, String methodName) {
        return ignoredTests.containsKey(clazz) && ignoredTests.get(clazz).equals(methodName);
    }

    public void ignoreTest(Class<? extends JongoTestBase> clazz, String methodName) {
        ignoredTests.put(clazz, methodName);
    }

    public List<Class<?>> findTestCases() {
        Set<Class<? extends JongoTestBase>> jongoTestCases = new Reflections("org.jongo").getSubTypesOf(JongoTestBase.class);
        return new ArrayList<Class<?>>(jongoTestCases);
    }

    public void prepareTestCase(Object testCase) throws Exception {
        if (testCase instanceof JongoTestBase) {
            JongoTestBase test = (JongoTestBase) testCase;
            test.prepareMarshallingStrategy(mapper);
        }
    }

}