| // Licensed to the Apache Software Foundation (ASF) under one |
| // or more contributor license agreements. See the NOTICE file |
| // distributed with this work for additional information |
| // regarding copyright ownership. The ASF licenses this file |
| // to you under the Apache License, Version 2.0 (the |
| // "License"); you may not use this file except in compliance |
| // with the License. You may obtain a copy of the License at |
| // |
| // http://www.apache.org/licenses/LICENSE-2.0 |
| // |
| // Unless required by applicable law or agreed to in writing, software |
| // distributed under the License is distributed on an "AS IS" BASIS, |
| // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| // See the License for the specific language governing permissions and |
| // limitations under the License. |
| |
| #include <gtest/gtest.h> |
| |
| #include <stout/check.hpp> |
| #include <stout/os.hpp> |
| #include <stout/path.hpp> |
| #include <stout/stringify.hpp> |
| #include <stout/strings.hpp> |
| #include <stout/uuid.hpp> |
| |
| #include "messages/messages.hpp" |
| |
| #include "slave/paths.hpp" |
| #include "slave/state.hpp" |
| |
| #include "tests/mesos.hpp" |
| |
| namespace mesos { |
| namespace internal { |
| namespace slave { |
| namespace paths { |
| |
| using std::string; |
| using strings::format; |
| |
| class PathsTest : public ::testing::Test |
| { |
| public: |
| virtual void SetUp() |
| { |
| slaveId.set_value("agent1"); |
| frameworkId.set_value("framework1"); |
| executorId.set_value("executor1"); |
| taskId.set_value("task1"); |
| containerId.set_value(UUID::random().toString()); |
| role = "role1"; |
| persistenceId = "persistenceId1"; |
| |
| Try<string> path = os::mkdtemp(); |
| ASSERT_SOME(path) << "Failed to mkdtemp"; |
| rootDir = path.get(); |
| |
| path = os::mkdtemp(); |
| ASSERT_SOME(path) << "Failed to mkdtemp"; |
| diskSourceDir = path.get(); |
| |
| imageType = Image::APPC; |
| } |
| |
| virtual void TearDown() |
| { |
| os::rmdir(rootDir); |
| os::rmdir(diskSourceDir); |
| } |
| |
| protected: |
| SlaveID slaveId; |
| FrameworkID frameworkId; |
| ExecutorID executorId; |
| TaskID taskId; |
| ContainerID containerId; |
| string role; |
| string persistenceId; |
| string rootDir; |
| string diskSourceDir; |
| Image::Type imageType; |
| }; |
| |
| |
| TEST_F(PathsTest, CreateExecutorDirectory) |
| { |
| const string& result = paths::createExecutorDirectory( |
| rootDir, slaveId, frameworkId, executorId, containerId); |
| |
| // Expected directory layout. |
| string dir = path::join( |
| rootDir, |
| "slaves", |
| slaveId.value(), |
| "frameworks", |
| frameworkId.value(), |
| "executors", |
| executorId.value(), |
| "runs", |
| containerId.value()); |
| |
| ASSERT_EQ(dir, result); |
| } |
| |
| |
| TEST_F(PathsTest, ParseExecutorRunPath) |
| { |
| string goodDir = paths::getExecutorRunPath( |
| rootDir, |
| slaveId, |
| frameworkId, |
| executorId, |
| containerId); |
| |
| string badDir1 = paths::getExecutorPath( |
| "/some/other/root", |
| slaveId, |
| frameworkId, |
| executorId); |
| |
| string badDir2 = paths::getExecutorPath( |
| rootDir, |
| slaveId, |
| frameworkId, |
| executorId); |
| |
| string badDir3 = path::join( |
| rootDir, |
| "notslaves", |
| slaveId.value(), |
| "notframeworks", |
| frameworkId.value(), |
| "notexecutors", |
| executorId.value(), |
| "notruns", |
| containerId.value()); |
| |
| Try<ExecutorRunPath> path = paths::parseExecutorRunPath(rootDir, goodDir); |
| ASSERT_SOME(path); |
| |
| EXPECT_EQ(slaveId, path->slaveId); |
| EXPECT_EQ(frameworkId, path->frameworkId); |
| EXPECT_EQ(executorId, path->executorId); |
| EXPECT_EQ(containerId, path->containerId); |
| |
| path = paths::parseExecutorRunPath(rootDir, badDir1); |
| ASSERT_ERROR(path); |
| |
| path = paths::parseExecutorRunPath(rootDir, badDir2); |
| ASSERT_ERROR(path); |
| |
| path = paths::parseExecutorRunPath(rootDir, badDir3); |
| ASSERT_ERROR(path); |
| } |
| |
| |
| TEST_F(PathsTest, Meta) |
| { |
| EXPECT_EQ(path::join(rootDir, "meta"), paths::getMetaRootDir(rootDir)); |
| } |
| |
| |
| TEST_F(PathsTest, ProvisionerDir) |
| { |
| EXPECT_EQ(path::join(rootDir, "provisioner"), |
| paths::getProvisionerDir(rootDir)); |
| } |
| |
| |
| TEST_F(PathsTest, BootId) |
| { |
| EXPECT_EQ(path::join(rootDir, "boot_id"), paths::getBootIdPath(rootDir)); |
| } |
| |
| |
| TEST_F(PathsTest, Slave) |
| { |
| const string slavesRoot = path::join(rootDir, "slaves"); |
| |
| EXPECT_EQ(path::join(slavesRoot, "latest"), |
| paths::getLatestSlavePath(rootDir)); |
| |
| EXPECT_EQ(path::join(slavesRoot, slaveId.value()), |
| paths::getSlavePath(rootDir, slaveId)); |
| } |
| |
| |
| TEST_F(PathsTest, Framework) |
| { |
| const string frameworksRoot = |
| path::join(paths::getSlavePath(rootDir, slaveId), "frameworks"); |
| |
| EXPECT_EQ(path::join(frameworksRoot, frameworkId.value()), |
| paths::getFrameworkPath(rootDir, slaveId, frameworkId)); |
| } |
| |
| |
| TEST_F(PathsTest, Executor) |
| { |
| const string executorsRoot = |
| path::join( |
| paths::getFrameworkPath( |
| rootDir, |
| slaveId, |
| frameworkId), |
| "executors"); |
| |
| EXPECT_EQ(path::join(executorsRoot, executorId.value()), |
| paths::getExecutorPath(rootDir, slaveId, frameworkId, executorId)); |
| |
| EXPECT_EQ( |
| path::join( |
| executorsRoot, |
| executorId.value(), |
| "runs", |
| containerId.value()), |
| paths::getExecutorRunPath( |
| rootDir, |
| slaveId, |
| frameworkId, |
| executorId, |
| containerId)); |
| } |
| |
| |
| TEST_F(PathsTest, LibProcessPid) |
| { |
| EXPECT_EQ( |
| path::join( |
| getExecutorRunPath( |
| rootDir, |
| slaveId, |
| frameworkId, |
| executorId, |
| containerId), |
| "pids", |
| "libprocess.pid"), |
| paths::getLibprocessPidPath( |
| rootDir, |
| slaveId, |
| frameworkId, |
| executorId, |
| containerId)); |
| } |
| |
| |
| TEST_F(PathsTest, Task) |
| { |
| const string tasksRoot = |
| path::join( |
| paths::getExecutorRunPath( |
| rootDir, |
| slaveId, |
| frameworkId, |
| executorId, |
| containerId), |
| "tasks"); |
| |
| EXPECT_EQ( |
| path::join(tasksRoot, taskId.value()), |
| paths::getTaskPath( |
| rootDir, |
| slaveId, |
| frameworkId, |
| executorId, |
| containerId, |
| taskId)); |
| } |
| |
| |
| TEST_F(PathsTest, PersistentVolume) |
| { |
| string dir = path::join(rootDir, "volumes", "roles", role, persistenceId); |
| |
| EXPECT_EQ(dir, paths::getPersistentVolumePath(rootDir, role, persistenceId)); |
| |
| dir = path::join(diskSourceDir, "volumes", "roles", role, persistenceId); |
| |
| Resource disk = tests::createPersistentVolume( |
| Megabytes(1024), |
| "role1", |
| persistenceId, |
| "path1", |
| None()); |
| |
| disk.mutable_disk()->mutable_source()->set_type( |
| Resource::DiskInfo::Source::PATH); |
| |
| disk.mutable_disk()->mutable_source()->mutable_path()->set_root( |
| diskSourceDir); |
| |
| EXPECT_EQ( |
| dir, |
| paths::getPersistentVolumePath(rootDir, disk)); |
| |
| disk.mutable_disk()->mutable_source()->set_type( |
| Resource::DiskInfo::Source::MOUNT); |
| |
| disk.mutable_disk()->mutable_source()->clear_path(); |
| disk.mutable_disk()->mutable_source()->mutable_mount()->set_root( |
| diskSourceDir); |
| |
| EXPECT_EQ( |
| diskSourceDir, |
| paths::getPersistentVolumePath(rootDir, disk)); |
| } |
| |
| |
| } // namespace paths { |
| } // namespace slave { |
| } // namespace internal { |
| } // namespace mesos { |