Publish

1 files changed, 202 insertions, 0 deletions

diff --git a/document/src/tests/documentcalculatortestcase.cpp b/document/src/tests/documentcalculatortestcase.cpp
new file mode 100644
index 00000000000..00de5427961
--- /dev/null
+++ b/document/src/tests/documentcalculatortestcase.cpp
@@ -0,0 +1,202 @@
+// Copyright 2016 Yahoo Inc. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+
+
+#include <vespa/fastos/fastos.h>
+#include <vespa/document/base/testdocrepo.h>
+#include <vespa/document/fieldvalue/fieldvalues.h>
+#include <vespa/vespalib/io/fileutil.h>
+#include <vespa/vdstestlib/cppunit/macros.h>
+#include <vespa/document/base/documentcalculator.h>
+#include <vespa/document/fieldvalue/intfieldvalue.h>
+#include <vespa/document/fieldvalue/longfieldvalue.h>
+#include <vespa/document/fieldvalue/floatfieldvalue.h>
+
+#include <fstream>
+
+namespace document {
+
+class DocumentCalculatorTest : public CppUnit::TestFixture {
+    TestDocRepo _testRepo;
+
+public:
+    const DocumentTypeRepo &getRepo() { return _testRepo.getTypeRepo(); }
+
+    void setUp() {}
+    void tearDown() {}
+
+    void testConstant();
+    void testSimple();
+    void testVariables();
+    void testFields();
+    void testDivideByZero();
+    void testModByZero();
+    void testFieldsDivZero();
+    void testFieldNotSet();
+    void testFieldNotFound();
+    void testByteSubtractionZeroResult();
+
+    CPPUNIT_TEST_SUITE(DocumentCalculatorTest);
+    CPPUNIT_TEST(testConstant);
+    CPPUNIT_TEST(testSimple);
+    CPPUNIT_TEST(testVariables);
+    CPPUNIT_TEST(testFields);
+    CPPUNIT_TEST(testDivideByZero);
+    CPPUNIT_TEST(testModByZero);
+    CPPUNIT_TEST(testFieldsDivZero);
+    CPPUNIT_TEST(testFieldNotSet);
+    CPPUNIT_TEST(testFieldNotFound);
+    CPPUNIT_TEST(testByteSubtractionZeroResult);
+    CPPUNIT_TEST_SUITE_END();
+};
+
+CPPUNIT_TEST_SUITE_REGISTRATION(DocumentCalculatorTest);
+
+
+void
+DocumentCalculatorTest::testConstant() {
+    DocumentCalculator::VariableMap variables;
+    DocumentCalculator calc(getRepo(), "4.0");
+
+    Document doc(*_testRepo.getDocumentType("testdoctype1"),
+                 DocumentId("doc:test:foo"));
+    CPPUNIT_ASSERT_EQUAL(4.0, calc.evaluate(doc, variables));
+}
+
+void
+DocumentCalculatorTest::testSimple() {
+    DocumentCalculator::VariableMap variables;
+    DocumentCalculator calc(getRepo(), "(3 + 5) / 2");
+
+    Document doc(*_testRepo.getDocumentType("testdoctype1"),
+                 DocumentId("doc:test:foo"));
+    CPPUNIT_ASSERT_EQUAL(4.0, calc.evaluate(doc, variables));
+}
+
+void
+DocumentCalculatorTest::testVariables() {
+    DocumentCalculator::VariableMap variables;
+    variables["x"] = 3.0;
+    variables["y"] = 5.0;
+    DocumentCalculator calc(getRepo(), "($x + $y) / 2");
+
+    Document doc(*_testRepo.getDocumentType("testdoctype1"),
+                 DocumentId("doc:test:foo"));
+    CPPUNIT_ASSERT_EQUAL(4.0, calc.evaluate(doc, variables));
+}
+
+void
+DocumentCalculatorTest::testFields() {
+    DocumentCalculator::VariableMap variables;
+    variables["x"] = 3.0;
+    variables["y"] = 5.0;
+    DocumentCalculator calc(getRepo(), "(testdoctype1.headerval + testdoctype1"
+                            ".hfloatval) / testdoctype1.headerlongval");
+
+    Document doc(*_testRepo.getDocumentType("testdoctype1"),
+                 DocumentId("doc:test:foo"));
+    doc.setValue(doc.getField("headerval"), IntFieldValue(5));
+    doc.setValue(doc.getField("hfloatval"), FloatFieldValue(3.0));
+    doc.setValue(doc.getField("headerlongval"), LongFieldValue(2));
+    CPPUNIT_ASSERT_EQUAL(4.0, calc.evaluate(doc, variables));
+}
+
+void
+DocumentCalculatorTest::testFieldsDivZero() {
+    DocumentCalculator::VariableMap variables;
+    variables["x"] = 3.0;
+    variables["y"] = 5.0;
+    DocumentCalculator calc(getRepo(), "(testdoctype1.headerval + testdoctype1"
+                            ".hfloatval) / testdoctype1.headerlongval");
+
+    Document doc(*_testRepo.getDocumentType("testdoctype1"),
+                 DocumentId("doc:test:foo"));
+    doc.setValue(doc.getField("headerval"), IntFieldValue(5));
+    doc.setValue(doc.getField("hfloatval"), FloatFieldValue(3.0));
+    doc.setValue(doc.getField("headerlongval"), LongFieldValue(0));
+    try {
+        calc.evaluate(doc, variables);
+        CPPUNIT_ASSERT(false);
+    } catch (const vespalib::IllegalArgumentException& e) {
+        // OK
+    }
+}
+
+void
+DocumentCalculatorTest::testDivideByZero() {
+    DocumentCalculator::VariableMap variables;
+    DocumentCalculator calc(getRepo(), "(3 + 5) / 0");
+
+    Document doc(*_testRepo.getDocumentType("testdoctype1"),
+                 DocumentId("doc:test:foo"));
+    try {
+        calc.evaluate(doc, variables);
+        CPPUNIT_ASSERT(false);
+    } catch (const vespalib::IllegalArgumentException& e) {
+        // OK
+    }
+}
+
+void
+DocumentCalculatorTest::testModByZero() {
+    DocumentCalculator::VariableMap variables;
+    DocumentCalculator calc(getRepo(), "(3 + 5) % 0");
+
+    Document doc(*_testRepo.getDocumentType("testdoctype1"),
+                 DocumentId("doc:test:foo"));
+    try {
+        calc.evaluate(doc, variables);
+        CPPUNIT_ASSERT(false);
+    } catch (const vespalib::IllegalArgumentException& e) {
+        // OK
+    }
+}
+
+void
+DocumentCalculatorTest::testFieldNotSet() {
+    DocumentCalculator::VariableMap variables;
+    DocumentCalculator calc(getRepo(), "(testdoctype1.headerval + testdoctype1"
+                            ".hfloatval) / testdoctype1.headerlongval");
+
+    Document doc(*_testRepo.getDocumentType("testdoctype1"),
+                 DocumentId("doc:test:foo"));
+    doc.setValue(doc.getField("hfloatval"), FloatFieldValue(3.0));
+    doc.setValue(doc.getField("headerlongval"), LongFieldValue(2));
+    try {
+        calc.evaluate(doc, variables);
+        CPPUNIT_ASSERT(false);
+    } catch (const vespalib::IllegalArgumentException&) {
+        // OK
+    }
+}
+
+void
+DocumentCalculatorTest::testFieldNotFound() {
+    DocumentCalculator::VariableMap variables;
+    DocumentCalculator calc(getRepo(),
+                            "(testdoctype1.mynotfoundfield + testdoctype1"
+                            ".hfloatval) / testdoctype1.headerlongval");
+
+    Document doc(*_testRepo.getDocumentType("testdoctype1"),
+                 DocumentId("doc:test:foo"));
+    doc.setValue(doc.getField("hfloatval"), FloatFieldValue(3.0));
+    doc.setValue(doc.getField("headerlongval"), LongFieldValue(2));
+    try {
+        calc.evaluate(doc, variables);
+        CPPUNIT_ASSERT(false);
+    } catch (const vespalib::IllegalArgumentException&) {
+        // OK
+    }
+}
+
+void
+DocumentCalculatorTest::testByteSubtractionZeroResult() {
+    DocumentCalculator::VariableMap variables;
+    DocumentCalculator calc(getRepo(), "testdoctype1.byteval - 3");
+
+    Document doc(*_testRepo.getDocumentType("testdoctype1"),
+                 DocumentId("doc:test:foo"));
+    doc.setValue(doc.getField("byteval"), ByteFieldValue(3));
+    CPPUNIT_ASSERT_EQUAL(0.0, calc.evaluate(doc, variables));
+}
+
+}