com.codenjoy.dojo.tetris.model.GameSetupRule.java Source code

Introduction

Here is the source code for com.codenjoy.dojo.tetris.model.GameSetupRule.java

Source

package com.codenjoy.dojo.tetris.model;

/*-
 * #%L
 * Codenjoy - it's a dojo-like platform from developers to developers.
 * %%
 * Copyright (C) 2016 Codenjoy
 * %%
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as
 * published by the Free Software Foundation, either version 3 of the
 * License, or (at your option) any later version.
 * 
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 * 
 * You should have received a copy of the GNU General Public
 * License along with this program.  If not, see
 * <http://www.gnu.org/licenses/gpl-3.0.html>.
 * #L%
 */

import com.codenjoy.dojo.services.PrinterFactory;
import com.codenjoy.dojo.services.PrinterFactoryImpl;
import org.apache.commons.lang.ArrayUtils;
import org.junit.rules.MethodRule;
import org.junit.runners.model.FrameworkMethod;
import org.junit.runners.model.Statement;
import org.mockito.Matchers;

import java.lang.reflect.Field;
import java.util.ArrayList;
import java.util.List;

import static org.mockito.Matchers.anyInt;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;

public class GameSetupRule implements MethodRule {

    private final Class<? extends TetrisGame> gameClass;

    public GameSetupRule(Class<? extends TetrisGame> gameClass) {
        this.gameClass = gameClass;
    }

    public Statement apply(final Statement base, final FrameworkMethod method, final Object target) {
        GivenFiguresInQueue givenAnnotation = method.getAnnotation(GivenFiguresInQueue.class);
        if (givenAnnotation != null) {
            FigureProperties[] figures = givenAnnotation.value();

            FigureQueue figureQueue = mock(FigureQueue.class);
            initQueueWithFigures(figures, figureQueue);
            Field gameField = findField(gameClass, target);

            Glass glass = (Glass) getFieldValue(Glass.class, target);
            when(glass.accept(Matchers.<Figure>anyObject(), anyInt(), anyInt())).thenReturn(true);

            try {
                gameField.set(target, gameClass.getConstructor(FigureQueue.class, Glass.class, PrinterFactory.class)
                        .newInstance(figureQueue, glass, new PrinterFactoryImpl()));
            } catch (Exception e) {
                throw new RuntimeException(e);
            }
        }
        return base;
    }

    private Object getFieldValue(Class fieldType, Object target) {
        Field field = findField(fieldType, target);
        try {
            return field.get(target);
        } catch (IllegalAccessException e) {
            throw new RuntimeException(e);
        }
    }

    private List<Figure> initQueueWithFigures(FigureProperties[] figures, FigureQueue figureQueue) {
        List<Figure> result = new ArrayList<Figure>();
        for (FigureProperties figure : figures) {
            Figure figureMock = mock(Figure.class);
            when(figureMock.getLeft()).thenReturn(figure.left());
            when(figureMock.getRight()).thenReturn(figure.right());
            when(figureMock.getTop()).thenReturn(figure.top());
            when(figureMock.getBottom()).thenReturn(figure.bottom());
            when(figureMock.getType()).thenReturn(figure.type());
            result.add(figureMock);
        }
        Figure[] values = result.toArray(new Figure[result.size()]);
        when(figureQueue.next()).thenReturn(values.length > 0 ? values[0] : null,
                values.length > 1 ? (Figure[]) ArrayUtils.subarray(values, 1, values.length) : null);
        return result;
    }

    private Field findField(Class fieldType, Object target) {
        Field[] fields = target.getClass().getDeclaredFields();
        for (Field field : fields) {
            if (field.getType().isAssignableFrom(fieldType)) {
                field.setAccessible(true);
                return field;
            }
        }
        return null;
    }
}