JavascriptValueSourceTest.java

/*
 * Copyright (c) 2017 OBiBa. All rights reserved.
 *
 * This program and the accompanying materials
 * are made available under the terms of the GNU Public License v3.0.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

package org.obiba.magma.js;

import java.util.Date;

import org.junit.Before;
import org.junit.Test;
import org.mozilla.javascript.EvaluatorException;
import org.obiba.magma.Value;
import org.obiba.magma.ValueSet;
import org.obiba.magma.ValueTable;
import org.obiba.magma.VariableEntity;
import org.obiba.magma.type.DateTimeType;
import org.obiba.magma.type.DecimalType;
import org.obiba.magma.type.IntegerType;
import org.obiba.magma.type.TextType;

import static org.fest.assertions.api.Assertions.assertThat;
import static org.junit.Assert.fail;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;

public class JavascriptValueSourceTest extends AbstractJsTest {

  private ValueSet mockValueSet;

  @Before
  @Override
  public void before() {
    super.before();
    mockValueSet = mock(ValueSet.class);
    when(mockValueSet.getValueTable()).thenReturn(mock(ValueTable.class));
    when(mockValueSet.getVariableEntity()).thenReturn(mock(VariableEntity.class));
  }

  @Test
  public void test_simple_script() {
    JavascriptValueSource source = new JavascriptValueSource(DecimalType.get(), "1");
    source.initialise();

    Value value = source.getValue(mockValueSet);
    assertThat(value.getValue()).isEqualTo(1d);
  }

  @Test
  public void test_script_newValue() {
    JavascriptValueSource source = new JavascriptValueSource(DecimalType.get(), "newValue('1')");
    source.initialise();

    Value value = source.getValue(mockValueSet);
    assertThat(value.getValue()).isEqualTo(1d);
  }

  @Test
  public void test_script_newSequence_to_Value() {
    JavascriptValueSource source = new JavascriptValueSource(DecimalType.get(), "newSequence(['1','2','3'])");
    source.initialise();

    Value value = source.getValue(mockValueSet);
    assertThat(value.getValue()).isEqualTo(1d);
  }

  @Test
  public void test_script_newSequence_to_ValueSequence() {
    JavascriptValueSource source = new JavascriptValueSource(DecimalType.get(), "newSequence(['1',null,'3'])") {
      @Override
      protected boolean isSequence() {
        return true;
      }
    };
    source.initialise();

    Value value = source.getValue(mockValueSet);
    assertThat(value.isSequence()).isTrue();
    assertThat(value.asSequence().getSize()).isEqualTo(3);
    assertThat(value.asSequence().get(0).getValue()).isEqualTo(1d);
    assertThat(value.asSequence().get(1).isNull()).isTrue();
  }

  @Test
  public void test_script_null_to_ValueSequence() {
    JavascriptValueSource source = new JavascriptValueSource(DecimalType.get(), "null") {
      @Override
      protected boolean isSequence() {
        return true;
      }
    };
    source.initialise();

    Value value = source.getValue(mockValueSet);
    assertThat(value.isSequence()).isTrue();
    assertThat(value.asSequence().getSize()).isEqualTo(0);
  }

  @Test
  public void test_script_nullValue_to_ValueSequence() {
    JavascriptValueSource source = new JavascriptValueSource(DecimalType.get(), "newValue(null, 'text')") {
      @Override
      protected boolean isSequence() {
        return true;
      }
    };
    source.initialise();

    Value value = source.getValue(mockValueSet);
    assertThat(value.isSequence()).isTrue();
    assertThat(value.asSequence().getSize()).isEqualTo(1);
    assertThat(value.asSequence().get(0).isNull()).isTrue();
  }

  @Test
  public void test_engine_method() {
    JavascriptValueSource source = new JavascriptValueSource(DateTimeType.get(), "now()");
    source.initialise();

    Value value = source.getValue(mockValueSet);
    assertThat(value).isNotNull();
    assertThat(value.isNull()).isFalse();
    assertThat(value.getValueType()).isEqualTo(DateTimeType.get());

    Date dateValue = (Date) value.getValue();
    // Make sure both dates are within 1 second of one-another
    assertThat(System.currentTimeMillis() - dateValue.getTime()).isLessThan(1000);
  }

  @Test
  public void test_compile_error() {
    // Error is on second line of script
    String script = "var i = 1+1;\nERROR!";
    JavascriptValueSource source = new JavascriptValueSource(IntegerType.get(), script);
    source.setScriptName("Bogus");
    try {
      source.initialise();
      source.getValue(null);
      fail("EvaluatorException was expected");
    } catch(MagmaJsRuntimeException e) {
      assertThat(e.getCause() instanceof EvaluatorException);
      EvaluatorException cause = (EvaluatorException)e.getCause();
      assertThat(cause.sourceName()).isEqualTo("Bogus");
      assertThat(cause.lineNumber()).isEqualTo(2);
      assertThat(cause.lineSource()).isEqualTo("ERROR!");
    }
  }

  @Test
  public void test_OPAL_1110() {
    JavascriptValueSource source = new JavascriptValueSource(TextType.get(), "1");
    source.initialise();

    Value value = source.getValue(mockValueSet);
    assertThat(value.getValue()).isEqualTo("1");
  }

}