1package io.protobase.f7.formulas.logic;
 2
 3import io.protobase.f7.errors.DivException;
 4import io.protobase.f7.errors.NAException;
 5import io.protobase.f7.errors.ValueException;
 6import io.protobase.f7.models.Grid;
 7import io.protobase.f7.testutils.TestFormula;
 8import org.junit.Test;
 9
10import static com.google.common.truth.Truth.assertThat;
11import static org.mockito.Mockito.verify;
12import static org.mockito.Mockito.when;
13
14public class EQTest extends TestFormula {
15
16  @Test
17  public void test_sameType() {
18    assertThat(EQ.SELF.apply(null, 10.0, 10.0)).isEqualTo(true);
19    assertThat(EQ.SELF.apply(null, 10.0, 0.0)).isEqualTo(false);
20    assertThat(EQ.SELF.apply(null, "Same", "Same")).isEqualTo(true);
21    assertThat(EQ.SELF.apply(null, "Same", "Different")).isEqualTo(false);
22    assertThat(EQ.SELF.apply(null, true, true)).isEqualTo(true);
23    assertThat(EQ.SELF.apply(null, true, false)).isEqualTo(false);
24    assertThat(EQ.SELF.apply(null, false, false)).isEqualTo(true);
25  }
26
27  @Test
28  public void test_error() {
29    assertThat(EQ.SELF.apply(null, new ValueException(), new ValueException())).isEqualTo(new ValueException());
30    assertThat(EQ.SELF.apply(null, new DivException(), new ValueException())).isEqualTo(new DivException());
31    assertThat(EQ.SELF.apply(null, 4.4444, new ValueException())).isEqualTo(new ValueException());
32  }
33
34  @Test
35  public void test_lookup() {
36    EQ F = new EQ(lookup, collateralLookup);
37    when(collateralLookup.apply(A1, M22_RANGE)).thenReturn("A");
38    when(collateralLookup.apply(A1, G19_RANGE)).thenReturn("B");
39    assertThat(F.apply(A1, M22_RANGE, G19_RANGE)).isEqualTo(false);
40    verify(collateralLookup).apply(A1, M22_RANGE);
41  }
42
43  @Test
44  public void test_grids() {
45    Grid one = Grid.builder()
46        .add(0, 0, 44.0)
47        .add(0, 1, "A")
48        .build();
49    Grid two = Grid.builder()
50        .add(0, 0, 44.0)
51        .add(0, 1, "B")
52        .build();
53    assertThat(EQ.SELF.apply(null, one, two)).isEqualTo(true);
54  }
55
56  @Test
57  public void test_errorFromArgumentsMismatch() {
58    assertThat(EQ.SELF.apply(null, "Too few")).isEqualTo(new NAException());
59    assertThat(EQ.SELF.apply(null, "A", "B", "Too many")).isEqualTo(new NAException());
60  }
61}