1 files changed, 99 insertions, 0 deletions
diff --git a/filedistribution/src/tests/zkfiledbmodel/test-zkfiledbmodel.cpp b/filedistribution/src/tests/zkfiledbmodel/test-zkfiledbmodel.cpp
new file mode 100644
index 00000000000..b385949bb98
--- /dev/null
+++ b/filedistribution/src/tests/zkfiledbmodel/test-zkfiledbmodel.cpp
@@ -0,0 +1,99 @@
+// Copyright 2016 Yahoo Inc. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+#define BOOST_TEST_DYN_LINK
+#define BOOST_TEST_MAIN
+#define BOOST_TEST_MODULE zkfiledbmodel test
+#include <vespa/fastos/fastos.h>
+#include <boost/test/unit_test.hpp>
+
+#include <iostream>
+
+#include <boost/thread/barrier.hpp>
+#include <boost/thread/thread.hpp>
+#include <boost/checked_delete.hpp>
+
+#include <vespa/filedistribution/common/componentsdeleter.h>
+#include <vespa/filedistribution/model/zkfacade.h>
+#include <vespa/filedistribution/model/zkfiledbmodel.h>
+
+#include <zookeeper/zookeeper.h>
+
+
+using namespace filedistribution;
+
+typedef boost::filesystem::path Path;
+
+namespace {
+
+
+struct Fixture {
+    boost::shared_ptr<ExceptionRethrower> _exceptionRethrower;
+    ComponentsDeleter _componentsDeleter;
+    boost::shared_ptr<ZKFacade> zk;
+    boost::shared_ptr<ZKFileDBModel> model;
+
+    Fixture() {
+        _exceptionRethrower.reset(new ExceptionRethrower());
+
+        zoo_set_debug_level(ZOO_LOG_LEVEL_WARN);
+        zk = _componentsDeleter.track(new ZKFacade("test1-tonyv:2181", _exceptionRethrower));
+        zk->setData("/vespa", "", 0);
+
+        model = _componentsDeleter.track(new ZKFileDBModel(zk));
+    }
+};
+
+} //anonymous namespace
+
+
+BOOST_FIXTURE_TEST_SUITE(ZKFileDBModelTests, Fixture)
+
+BOOST_AUTO_TEST_CASE(retainOnlyHostsForTenant)
+{
+    Path path = "/vespa/filedistribution/hosts";
+    std::vector<std::string> files = {"myfile"};
+    BOOST_CHECK(zk->hasNode("/vespa"));
+    BOOST_CHECK(zk->hasNode("/vespa/filedistribution"));
+    BOOST_CHECK(zk->hasNode(path));
+    model->setDeployedFilesToDownload("testhost", "myapp:so:cool", files);
+    model->setDeployedFilesToDownload("testhost2", "myapp:so:cool", files);
+    model->setDeployedFilesToDownload("testhost3", "myapp:so:cool", files);
+    model->setDeployedFilesToDownload("testhost3", "myapp:legacyid:so:cool", files);
+    model->setDeployedFilesToDownload("testhost3", "yourapp:so:cool", files);
+    BOOST_CHECK(zk->getChildren(path / "testhost").size() == 1);
+    BOOST_CHECK(zk->getChildren(path / "testhost2").size() == 1);
+    BOOST_CHECK(zk->getChildren(path / "testhost3").size() == 3);
+
+    model->cleanDeployedFilesToDownload({"testhost3"}, "yourapp:so:cool");
+    model->removeDeploymentsThatHaveDifferentApplicationId({"testhost3"}, "yourapp:so:cool");
+    BOOST_CHECK(zk->hasNode(path / "testhost"));
+    BOOST_CHECK(zk->hasNode(path / "testhost2"));
+    BOOST_CHECK(zk->hasNode(path / "testhost3"));
+    BOOST_CHECK(zk->getChildren(path / "testhost").size() == 1);
+    BOOST_CHECK(zk->getChildren(path / "testhost2").size() == 1);
+    BOOST_CHECK(zk->getChildren(path / "testhost3").size() == 1);
+
+    model->cleanDeployedFilesToDownload({"testhost"}, "myapp:not:cool");
+    model->removeDeploymentsThatHaveDifferentApplicationId({"testhost"}, "myapp:not:cool");
+    BOOST_CHECK(zk->hasNode(path / "testhost"));
+    BOOST_CHECK(zk->hasNode(path / "testhost2"));
+    BOOST_CHECK(zk->hasNode(path / "testhost3"));
+    BOOST_CHECK(zk->getChildren(path / "testhost").size() == 0);
+    BOOST_CHECK(zk->getChildren(path / "testhost2").size() == 1);
+    BOOST_CHECK(zk->getChildren(path / "testhost3").size() == 1);
+
+    model->cleanDeployedFilesToDownload({"testhost2"}, "myapp:so:cool");
+    model->removeDeploymentsThatHaveDifferentApplicationId({"testhost2"}, "myapp:so:cool");
+
+    BOOST_CHECK(!zk->hasNode(path / "testhost"));
+    BOOST_CHECK(zk->hasNode(path / "testhost2"));
+    BOOST_CHECK(zk->hasNode(path / "testhost3"));
+    BOOST_CHECK(zk->getChildren(path / "testhost2").size() == 1);
+    BOOST_CHECK(zk->getChildren(path / "testhost3").size() == 1);
+
+    model->cleanDeployedFilesToDownload({"testhost2"}, "yourapp:so:cool");
+    BOOST_CHECK(!zk->hasNode(path / "testhost"));
+    BOOST_CHECK(zk->hasNode(path / "testhost2"));
+    BOOST_CHECK(!zk->hasNode(path / "testhost3"));
+}
+
+BOOST_AUTO_TEST_SUITE_END()