package com.infomancers.tests;

import com.infomancers.collections.yield.asmtree.enhancers.*;
import static org.junit.Assert.assertEquals;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.junit.runners.Parameterized;
import org.objectweb.asm.Opcodes;
import org.objectweb.asm.tree.*;

import java.util.Arrays;
import java.util.Collection;

/**
 * Created by IntelliJ IDEA.
 * User: aviadbendov
 * Date: Apr 3, 2009
 * Time: 11:50:10 PM
 * To change this template use File | Settings | File Templates.
 */
@RunWith(Parameterized.class)
public class EnhancersFactoryTests {

    @Parameterized.Parameters
    public static Collection<Object[]> parameters() {
        return Arrays.asList(
                new Object[]{new IincInsnNode(1, 1), IincEnhancer.class},
                new Object[]{new InsnNode(Opcodes.ARRAYLENGTH), ArraylengthEnhancer.class},

                new Object[]{new InsnNode(Opcodes.IALOAD), ArrayLoadEnhancer.class},
                new Object[]{new InsnNode(Opcodes.LALOAD), ArrayLoadEnhancer.class},
                new Object[]{new InsnNode(Opcodes.FALOAD), ArrayLoadEnhancer.class},
                new Object[]{new InsnNode(Opcodes.DALOAD), ArrayLoadEnhancer.class},
                new Object[]{new InsnNode(Opcodes.AALOAD), ArrayLoadEnhancer.class},
                new Object[]{new InsnNode(Opcodes.BALOAD), ArrayLoadEnhancer.class},
                new Object[]{new InsnNode(Opcodes.CALOAD), ArrayLoadEnhancer.class},
                new Object[]{new InsnNode(Opcodes.SALOAD), ArrayLoadEnhancer.class},

                new Object[]{new InsnNode(Opcodes.IASTORE), ArrayStoreEnhancer.class},
                new Object[]{new InsnNode(Opcodes.LASTORE), ArrayStoreEnhancer.class},
                new Object[]{new InsnNode(Opcodes.FASTORE), ArrayStoreEnhancer.class},
                new Object[]{new InsnNode(Opcodes.DASTORE), ArrayStoreEnhancer.class},
                new Object[]{new InsnNode(Opcodes.AASTORE), ArrayStoreEnhancer.class},
                new Object[]{new InsnNode(Opcodes.BASTORE), ArrayStoreEnhancer.class},
                new Object[]{new InsnNode(Opcodes.CASTORE), ArrayStoreEnhancer.class},
                new Object[]{new InsnNode(Opcodes.SASTORE), ArrayStoreEnhancer.class},

                new Object[]{new VarInsnNode(Opcodes.ISTORE, 1), StoreEnhancer.class},
                new Object[]{new VarInsnNode(Opcodes.LSTORE, 1), StoreEnhancer.class},
                new Object[]{new VarInsnNode(Opcodes.DSTORE, 1), StoreEnhancer.class},
                new Object[]{new VarInsnNode(Opcodes.FSTORE, 1), StoreEnhancer.class},
                new Object[]{new VarInsnNode(Opcodes.ASTORE, 1), StoreEnhancer.class},

                new Object[]{new VarInsnNode(Opcodes.ILOAD, 1), LoadEnhancer.class},
                new Object[]{new VarInsnNode(Opcodes.LLOAD, 1), LoadEnhancer.class},
                new Object[]{new VarInsnNode(Opcodes.DLOAD, 1), LoadEnhancer.class},
                new Object[]{new VarInsnNode(Opcodes.FLOAD, 1), LoadEnhancer.class},
                new Object[]{new VarInsnNode(Opcodes.ALOAD, 1), LoadEnhancer.class},
                new Object[]{new VarInsnNode(Opcodes.ALOAD, 0), NullEnhancer.class},

                new Object[]{new MethodInsnNode(Opcodes.INVOKEVIRTUAL, "owner", "yieldReturn", "(Ljava/lang/Object;)V"), YieldReturnEnhancer.class},
                new Object[]{new MethodInsnNode(Opcodes.INVOKEVIRTUAL, "owner", "yieldReturnCore", "(Ljava/lang/Object;)V"), MethodInvocationEnhancer.class},
                new Object[]{new MethodInsnNode(Opcodes.INVOKEVIRTUAL, "owner", "yieldReturn", "()V"), MethodInvocationEnhancer.class},

                new Object[]{new MethodInsnNode(Opcodes.INVOKEVIRTUAL, "owner", "yieldBreak", "(Ljava/lang/Object;)V"), MethodInvocationEnhancer.class},
                new Object[]{new MethodInsnNode(Opcodes.INVOKEVIRTUAL, "owner", "yieldBreakCore", "(Ljava/lang/Object;)V"), MethodInvocationEnhancer.class},
                new Object[]{new MethodInsnNode(Opcodes.INVOKEVIRTUAL, "owner", "yieldBreak", "()V"), YieldBreakEnhancer.class},

                new Object[]{new MethodInsnNode(Opcodes.INVOKEVIRTUAL, "java/lang/String", "length", "()I"), MethodInvocationEnhancer.class},
                new Object[]{new MethodInsnNode(Opcodes.INVOKEVIRTUAL, "java/lang/String", "wait", "()V"), MethodInvocationEnhancer.class},
                new Object[]{new MethodInsnNode(Opcodes.INVOKEVIRTUAL, "java/lang/Object", "wait", "()V"), NullEnhancer.class},

                new Object[]{new MethodInsnNode(Opcodes.INVOKEINTERFACE, "java/io/Closeable", "close", "()V"), MethodInvocationEnhancer.class},

                new Object[]{new MethodInsnNode(Opcodes.INVOKESTATIC, "java/lang/Integer", "valueOf", "(I)Ljava/lang/Integer;"), NullEnhancer.class},

                new Object[]{null, null}
        );
    }

    private final Class expected;
    private final AbstractInsnNode node;

    public EnhancersFactoryTests(AbstractInsnNode node, Class expected) {
        this.node = node;
        this.expected = expected;
    }

    @Test
    public void test() {
        if (expected == null) return;

        assertEquals("opcode: " + node.getOpcode(), expected, EnhancersFactory.instnace().createEnhancer(node).getClass());
    }
}