name:
src/test/java/io/protobase/f7/formulas/logic/IFNATest.java
-rw-r--r--
2599
1package io.protobase.f7.formulas.logic;
2
3import io.protobase.f7.errors.NAException;
4import io.protobase.f7.errors.NumException;
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.verifyNoMoreInteractions;
13import static org.mockito.Mockito.when;
14
15public class IFNATest extends TestFormula {
16 @Test
17 public void testApply() {
18 assertThat(IFNA.SELF.apply(null, "Not NA.", "Not returned.")).isEqualTo("Not NA.");
19 assertThat(IFNA.SELF.apply(null, 10.1, "Not returned.")).isEqualTo(10.1);
20 assertThat(IFNA.SELF.apply(null, true, "Not returned.")).isEqualTo(true);
21 assertThat(IFNA.SELF.apply(null, false, "Not returned.")).isEqualTo(false);
22 assertThat(IFNA.SELF.apply(null, 99.1, new ValueException())).isEqualTo(99.1);
23 }
24
25 @Test
26 public void testApply_error() {
27 assertThat(IFNA.SELF.apply(null, new NAException(), "Hello")).isEqualTo("Hello");
28 assertThat(IFNA.SELF.apply(null, new NumException(), "Hello")).isEqualTo(new NumException());
29 assertThat(IFNA.SELF.apply(null, new ValueException(), 10.11)).isEqualTo(new ValueException());
30 }
31
32 @Test
33 public void test_lookup_first() {
34 IFNA F = new IFNA(lookup, collateralLookup);
35 when(collateralLookup.apply(A1, M22_RANGE)).thenReturn(2.0);
36 assertThat(F.apply(A1, M22_RANGE, G19_RANGE)).isEqualTo(2.0);
37 verify(collateralLookup).apply(A1, M22_RANGE);
38 verifyNoMoreInteractions(collateralLookup);
39 }
40
41 @Test
42 public void test_lookup_second() {
43 IFNA F = new IFNA(lookup, collateralLookup);
44 when(collateralLookup.apply(A1, M22_RANGE)).thenReturn(new NAException());
45 when(collateralLookup.apply(A1, G19_RANGE)).thenReturn(2.0);
46 assertThat(F.apply(A1, M22_RANGE, G19_RANGE)).isEqualTo(2.0);
47 verify(collateralLookup).apply(A1, M22_RANGE);
48 verify(collateralLookup).apply(A1, G19_RANGE);
49 verifyNoMoreInteractions(collateralLookup);
50 }
51
52 @Test
53 public void testApply_gird() {
54 Grid<Object> grid = Grid.builder()
55 .add(0, 0, new NAException())
56 .add(0, 1, 1.0)
57 .add(0, 2, 2.0)
58 .add(0, 3, 3.0)
59 .build();
60 assertThat(IFNA.SELF.apply(null, grid, "Yes NA.")).isEqualTo("Yes NA.");
61 }
62
63 @Test
64 public void test_errorFromArgumentsMismatch() {
65 assertThat(IFNA.SELF.apply(null, 1.0)).isEqualTo(new NAException());
66 assertThat(IFNA.SELF.apply(null, 1.0, 2.0, 3.0)).isEqualTo(new NAException());
67 }
68}