StructMatchTest.java

package scotch.compiler.syntax.pattern;

import static java.util.Arrays.asList;
import static org.hamcrest.Matchers.is;
import static org.junit.Assert.assertThat;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;
import static scotch.util.TestUtil.access;
import static scotch.util.TestUtil.arg;
import static scotch.util.TestUtil.capture;
import static scotch.util.TestUtil.field;
import static scotch.util.TestUtil.struct;
import static scotch.compiler.syntax.type.Types.t;

import org.junit.Test;
import scotch.compiler.syntax.scope.Scope;
import scotch.compiler.syntax.type.VariableType;

public class StructMatchTest {

    @Test
    public void shouldBindStructMatch() {
        VariableType t = t(0);
        Scope scope = mock(Scope.class);
        when(scope.reserveType()).thenReturn(t);
        PatternMatch struct = struct("(,)", t, asList(
            field("_0", t, capture("a", t(1))),
            field("_1", t, capture("b", t(2)))
        ));
        assertThat(struct.bind(arg("#0", t), scope), is(struct(arg("#0", t), "(,)", t, asList(
            field("_0", t, capture(access(arg("#0", t), "_0", t), "a", t(1))),
            field("_1", t, capture(access(arg("#0", t), "_1", t), "b", t(2)))
        ))));
    }
}