1 files changed, 77 insertions, 0 deletions
diff --git a/config-model/src/test/java/com/yahoo/searchdefinition/processing/VespaMlModelTestCase.java b/config-model/src/test/java/com/yahoo/searchdefinition/processing/VespaMlModelTestCase.java
new file mode 100644
index 00000000000..a75699d2a1d
--- /dev/null
+++ b/config-model/src/test/java/com/yahoo/searchdefinition/processing/VespaMlModelTestCase.java
@@ -0,0 +1,77 @@
+// Copyright 2019 Oath Inc. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+package com.yahoo.searchdefinition.processing;
+
+import com.yahoo.config.application.api.ApplicationPackage;
+import com.yahoo.io.IOUtils;
+import com.yahoo.path.Path;
+import com.yahoo.searchdefinition.derived.RawRankProfile;
+import com.yahoo.vespa.model.VespaModel;
+import com.yahoo.vespa.model.ml.ImportedModelTester;
+import org.junit.After;
+import org.junit.Test;
+
+import java.io.IOException;
+import java.util.Optional;
+
+import static org.junit.Assert.assertEquals;
+
+/**
+ * Tests adding Vespa ranking expression based models in the models/ dir
+ *
+ * @author bratseth
+ */
+public class VespaMlModelTestCase {
+
+    private final Path applicationDir = Path.fromString("src/test/integration/vespa/");
+
+    private final String expectedRankConfig =
+            "constant(constant1).type : tensor(x[3])\n" +
+            "constant(constant1).value : tensor(x[3]):{{x:0}:0.5,{x:1}:1.5,{x:2}:2.5}\n" +
+            "rankingExpression(foo1).rankingScript : reduce(reduce(input1 * input2, sum, name) * constant(constant1), max, x) * 3.0\n" +
+            "rankingExpression(foo1).input2.type : tensor(x[3])\n" +
+            "rankingExpression(foo1).input1.type : tensor(name{},x[3])\n" +
+            "rankingExpression(foo2).rankingScript : max(reduce(input1 * input2, sum, name) * constant1asLarge,x) * 3.0\n" +
+            "rankingExpression(foo2).input2.type : tensor(x[3])\n" +
+            "rankingExpression(foo2).input1.type : tensor(name{},x[3])\n";
+
+    /** The model name */
+    private final String name = "example";
+
+    @After
+    public void removeGeneratedModelFiles() {
+        IOUtils.recursiveDeleteDir(applicationDir.append(ApplicationPackage.MODELS_GENERATED_DIR).toFile());
+    }
+
+    @Test
+    public void testGlobalVespaModel() throws IOException {
+        ImportedModelTester tester = new ImportedModelTester(name, applicationDir);
+        VespaModel model = tester.createVespaModel();
+        tester.assertLargeConstant("constant1asLarge", model, Optional.of(3L));
+        assertEquals(expectedRankConfig, rankConfigOf("example", model));
+
+        // At this point the expression is stored - copy application to another location which do not have a models dir
+        Path storedAppDir = applicationDir.append("copy");
+        try {
+            storedAppDir.toFile().mkdirs();
+            IOUtils.copy(applicationDir.append("services.xml").toString(), storedAppDir.append("services.xml").toString());
+            IOUtils.copyDirectory(applicationDir.append(ApplicationPackage.MODELS_GENERATED_DIR).toFile(),
+                                  storedAppDir.append(ApplicationPackage.MODELS_GENERATED_DIR).toFile());
+            ImportedModelTester storedTester = new ImportedModelTester(name, storedAppDir);
+            VespaModel storedModel = storedTester.createVespaModel();
+            storedTester.assertLargeConstant("constant1asLarge", model, Optional.of(3L));
+            assertEquals(expectedRankConfig, rankConfigOf("example", storedModel));
+        }
+        finally {
+            IOUtils.recursiveDeleteDir(storedAppDir.toFile());
+        }
+    }
+
+    private String rankConfigOf(String rankProfileName, VespaModel model) {
+        StringBuilder b = new StringBuilder();
+        RawRankProfile profile = model.rankProfileList().getRankProfile(rankProfileName);
+        for (var property : profile.configProperties())
+            b.append(property.getFirst()).append(" : ").append(property.getSecond()).append("\n");
+        return b.toString();
+    }
+
+}