288 lines
10 KiB
C++
288 lines
10 KiB
C++
|
/*
|
||
|
|
* Copyright (c) Huawei Technologies Co., Ltd. 2020. All rights reserved.
|
||
|
|
* iSulad licensed under the Mulan PSL v1.
|
||
|
|
* You can use this software according to the terms and conditions of the Mulan PSL v1.
|
||
|
|
* You may obtain a copy of Mulan PSL v1 at:
|
||
|
|
* http://license.coscl.org.cn/MulanPSL
|
||
|
|
* THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND, EITHER EXPRESS OR
|
||
|
|
* IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT, MERCHANTABILITY OR FIT FOR A PARTICULAR
|
||
|
|
* PURPOSE.
|
||
|
|
* See the Mulan PSL v1 for more details.
|
||
|
|
* Description: specs extend llt
|
||
|
|
* Author: lifeng
|
||
|
|
* Create: 2020-02-18
|
||
|
|
*/
|
||
|
|
|
||
|
|
#include <stdlib.h>
|
||
|
|
#include <stdio.h>
|
||
|
|
#include <gtest/gtest.h>
|
||
|
|
#include "mock.h"
|
||
|
|
#include "oci_runtime_spec.h"
|
||
|
|
#include "specs.h"
|
||
|
|
#include "host_config.h"
|
||
|
|
#include "container_config.h"
|
||
|
|
#include "oci_llt_common.h"
|
||
|
|
#include <gtest/gtest.h>
|
||
|
|
#include <gmock/gmock.h>
|
||
|
|
#include "isulad_config_mock.h"
|
||
|
|
#include "oci_runtime_hooks.h"
|
||
|
|
#include "utils.h"
|
||
|
|
#include "specs_extend.h"
|
||
|
|
|
||
|
|
using ::testing::Args;
|
||
|
|
using ::testing::ByRef;
|
||
|
|
using ::testing::SetArgPointee;
|
||
|
|
using ::testing::DoAll;
|
||
|
|
using ::testing::NiceMock;
|
||
|
|
using ::testing::Return;
|
||
|
|
using ::testing::NotNull;
|
||
|
|
using ::testing::AtLeast;
|
||
|
|
using ::testing::Invoke;
|
||
|
|
using ::testing::_;
|
||
|
|
|
||
|
|
using namespace std;
|
||
|
|
|
||
|
|
#define HOOKS_CONFIG_FILE "specs/specs_extend/hooks.json"
|
||
|
|
|
||
|
|
TEST(make_sure_oci_spec_linux_llt, test_make_sure_oci_spec_linux)
|
||
|
|
{
|
||
|
|
oci_runtime_spec *oci_spec = NULL;
|
||
|
|
oci_spec = (oci_runtime_spec *) util_common_calloc_s(sizeof(oci_runtime_spec));
|
||
|
|
ASSERT_TRUE(oci_spec != NULL);
|
||
|
|
ASSERT_EQ(make_sure_oci_spec_linux(oci_spec), 0);
|
||
|
|
ASSERT_TRUE(oci_spec->linux != NULL);
|
||
|
|
free_oci_runtime_spec(oci_spec);
|
||
|
|
oci_spec = NULL;
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST(make_sure_oci_spec_process_llt, test_make_sure_oci_spec_process)
|
||
|
|
{
|
||
|
|
oci_runtime_spec *oci_spec = NULL;
|
||
|
|
oci_spec = (oci_runtime_spec *) util_common_calloc_s(sizeof(oci_runtime_spec));
|
||
|
|
ASSERT_TRUE(oci_spec != NULL);
|
||
|
|
ASSERT_EQ(make_sure_oci_spec_process(oci_spec), 0);
|
||
|
|
ASSERT_TRUE(oci_spec->process != NULL);
|
||
|
|
free_oci_runtime_spec(oci_spec);
|
||
|
|
oci_spec = NULL;
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST(make_sure_oci_spec_linux_resources_llt, test_make_sure_oci_spec_linux_resources)
|
||
|
|
{
|
||
|
|
oci_runtime_spec *oci_spec = NULL;
|
||
|
|
oci_spec = (oci_runtime_spec *) util_common_calloc_s(sizeof(oci_runtime_spec));
|
||
|
|
ASSERT_TRUE(oci_spec != NULL);
|
||
|
|
ASSERT_EQ(make_sure_oci_spec_linux_resources(oci_spec), 0);
|
||
|
|
ASSERT_TRUE(oci_spec->linux != NULL);
|
||
|
|
ASSERT_TRUE(oci_spec->linux->resources != NULL);
|
||
|
|
free_oci_runtime_spec(oci_spec);
|
||
|
|
oci_spec = NULL;
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST(make_sure_oci_spec_linux_resources_blkio_llt, test_make_sure_oci_spec_linux_resources_blkio)
|
||
|
|
{
|
||
|
|
oci_runtime_spec *oci_spec = NULL;
|
||
|
|
oci_spec = (oci_runtime_spec *) util_common_calloc_s(sizeof(oci_runtime_spec));
|
||
|
|
ASSERT_TRUE(oci_spec != NULL);
|
||
|
|
ASSERT_EQ(make_sure_oci_spec_linux_resources_blkio(oci_spec), 0);
|
||
|
|
ASSERT_TRUE(oci_spec->linux != NULL);
|
||
|
|
ASSERT_TRUE(oci_spec->linux->resources != NULL);
|
||
|
|
ASSERT_TRUE(oci_spec->linux->resources->block_io != NULL);
|
||
|
|
free_oci_runtime_spec(oci_spec);
|
||
|
|
oci_spec = NULL;
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST(merge_hooks_llt, test_merge_hooks_invalid)
|
||
|
|
{
|
||
|
|
ASSERT_NE(merge_hooks(nullptr, nullptr), 0);
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST(merge_hooks_llt, test_merge_hooks_llt_2)
|
||
|
|
{
|
||
|
|
oci_runtime_spec *oci_spec = NULL;
|
||
|
|
|
||
|
|
oci_spec = (oci_runtime_spec *) util_common_calloc_s(sizeof(oci_runtime_spec));
|
||
|
|
ASSERT_TRUE(oci_spec != NULL);
|
||
|
|
oci_spec->hooks = (oci_runtime_spec_hooks*)util_common_calloc_s(sizeof(oci_runtime_spec_hooks));
|
||
|
|
ASSERT_NE(merge_hooks(oci_spec->hooks, NULL), 0);
|
||
|
|
free_oci_runtime_spec(oci_spec);
|
||
|
|
oci_spec = NULL;
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST(merge_hooks_llt, test_merge_hooks_llt_3)
|
||
|
|
{
|
||
|
|
char *hooks_config_file = NULL;
|
||
|
|
oci_runtime_spec_hooks *hooks_spec = NULL;
|
||
|
|
char *err = NULL;
|
||
|
|
|
||
|
|
hooks_config_file = json_path(HOOKS_CONFIG_FILE);
|
||
|
|
ASSERT_TRUE(hooks_config_file != NULL);
|
||
|
|
hooks_spec = oci_runtime_spec_hooks_parse_file(hooks_config_file, NULL, &err);
|
||
|
|
ASSERT_TRUE(hooks_spec != NULL);
|
||
|
|
free(err);
|
||
|
|
err = NULL;
|
||
|
|
free(hooks_config_file);
|
||
|
|
hooks_config_file = NULL;
|
||
|
|
ASSERT_NE(merge_hooks(NULL, hooks_spec), 0);
|
||
|
|
free_oci_runtime_spec_hooks(hooks_spec);
|
||
|
|
hooks_spec = NULL;
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST(merge_hooks_llt, test_merge_hooks_llt_4)
|
||
|
|
{
|
||
|
|
char *hooks_config_file = NULL;
|
||
|
|
oci_runtime_spec_hooks *hooks_spec = NULL;
|
||
|
|
oci_runtime_spec *oci_spec = NULL;
|
||
|
|
char *err = NULL;
|
||
|
|
|
||
|
|
// All parameter correct
|
||
|
|
hooks_config_file = json_path(HOOKS_CONFIG_FILE);
|
||
|
|
ASSERT_TRUE(hooks_config_file != NULL);
|
||
|
|
hooks_spec = oci_runtime_spec_hooks_parse_file(hooks_config_file, NULL, &err);
|
||
|
|
ASSERT_TRUE(hooks_spec != NULL);
|
||
|
|
free(err);
|
||
|
|
err = NULL;
|
||
|
|
free(hooks_config_file);
|
||
|
|
hooks_config_file = NULL;
|
||
|
|
|
||
|
|
oci_spec = (oci_runtime_spec *) util_common_calloc_s(sizeof(oci_runtime_spec));
|
||
|
|
ASSERT_TRUE(oci_spec != NULL);
|
||
|
|
oci_spec->hooks = (oci_runtime_spec_hooks*)util_common_calloc_s(sizeof(oci_runtime_spec_hooks));
|
||
|
|
|
||
|
|
ASSERT_EQ(merge_hooks(oci_spec->hooks, hooks_spec), 0);
|
||
|
|
|
||
|
|
free_oci_runtime_spec_hooks(hooks_spec);
|
||
|
|
hooks_spec = NULL;
|
||
|
|
free_oci_runtime_spec(oci_spec);
|
||
|
|
oci_spec = NULL;
|
||
|
|
}
|
||
|
|
TEST(merge_hooks_llt, test_merge_hooks_llt_prestart)
|
||
|
|
{
|
||
|
|
char *hooks_config_file = NULL;
|
||
|
|
oci_runtime_spec_hooks *hooks_spec = NULL;
|
||
|
|
oci_runtime_spec *oci_spec = NULL;
|
||
|
|
char *err = NULL;
|
||
|
|
|
||
|
|
// All parameter correct
|
||
|
|
hooks_config_file = json_path(HOOKS_CONFIG_FILE);
|
||
|
|
ASSERT_TRUE(hooks_config_file != NULL);
|
||
|
|
hooks_spec = oci_runtime_spec_hooks_parse_file(hooks_config_file, NULL, &err);
|
||
|
|
ASSERT_TRUE(hooks_spec != NULL);
|
||
|
|
free(err);
|
||
|
|
err = NULL;
|
||
|
|
free(hooks_config_file);
|
||
|
|
hooks_config_file = NULL;
|
||
|
|
|
||
|
|
oci_spec = (oci_runtime_spec *) util_common_calloc_s(sizeof(oci_runtime_spec));
|
||
|
|
ASSERT_TRUE(oci_spec != NULL);
|
||
|
|
oci_spec->hooks = (oci_runtime_spec_hooks*)util_common_calloc_s(sizeof(oci_runtime_spec_hooks));
|
||
|
|
|
||
|
|
ASSERT_EQ(merge_hooks(oci_spec->hooks, hooks_spec), 0);
|
||
|
|
ASSERT_EQ(oci_spec->hooks->prestart_len, 1);
|
||
|
|
ASSERT_STREQ(oci_spec->hooks->prestart[0]->path, "/home/hooks/start.bash");
|
||
|
|
ASSERT_EQ(oci_spec->hooks->prestart[0]->args_len, 3);
|
||
|
|
ASSERT_STREQ(oci_spec->hooks->prestart[0]->args[0], "arg0");
|
||
|
|
ASSERT_STREQ(oci_spec->hooks->prestart[0]->args[1], "arg1");
|
||
|
|
ASSERT_STREQ(oci_spec->hooks->prestart[0]->args[2], "arg2");
|
||
|
|
ASSERT_EQ(oci_spec->hooks->prestart[0]->env_len, 1);
|
||
|
|
ASSERT_STREQ(oci_spec->hooks->prestart[0]->env[0], "key1=value1");
|
||
|
|
ASSERT_EQ(oci_spec->hooks->prestart[0]->timeout, 40);
|
||
|
|
|
||
|
|
free_oci_runtime_spec_hooks(hooks_spec);
|
||
|
|
hooks_spec = NULL;
|
||
|
|
free_oci_runtime_spec(oci_spec);
|
||
|
|
oci_spec = NULL;
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST(merge_hooks_llt, test_merge_hooks_llt_poststart)
|
||
|
|
{
|
||
|
|
char *hooks_config_file = NULL;
|
||
|
|
oci_runtime_spec_hooks *hooks_spec = NULL;
|
||
|
|
oci_runtime_spec *oci_spec = NULL;
|
||
|
|
char *err = NULL;
|
||
|
|
|
||
|
|
// All parameter correct
|
||
|
|
hooks_config_file = json_path(HOOKS_CONFIG_FILE);
|
||
|
|
ASSERT_TRUE(hooks_config_file != NULL);
|
||
|
|
hooks_spec = oci_runtime_spec_hooks_parse_file(hooks_config_file, NULL, &err);
|
||
|
|
ASSERT_TRUE(hooks_spec != NULL);
|
||
|
|
free(err);
|
||
|
|
err = NULL;
|
||
|
|
free(hooks_config_file);
|
||
|
|
hooks_config_file = NULL;
|
||
|
|
|
||
|
|
oci_spec = (oci_runtime_spec *) util_common_calloc_s(sizeof(oci_runtime_spec));
|
||
|
|
ASSERT_TRUE(oci_spec != NULL);
|
||
|
|
oci_spec->hooks = (oci_runtime_spec_hooks*)util_common_calloc_s(sizeof(oci_runtime_spec_hooks));
|
||
|
|
|
||
|
|
ASSERT_EQ(merge_hooks(oci_spec->hooks, hooks_spec), 0);
|
||
|
|
ASSERT_EQ(oci_spec->hooks->poststart_len, 2);
|
||
|
|
ASSERT_STREQ(oci_spec->hooks->poststart[0]->path, "/home/hooks/post1.bash");
|
||
|
|
ASSERT_EQ(oci_spec->hooks->poststart[0]->args_len, 3);
|
||
|
|
ASSERT_STREQ(oci_spec->hooks->poststart[0]->args[0], "arg5");
|
||
|
|
ASSERT_STREQ(oci_spec->hooks->poststart[0]->args[1], "arg6");
|
||
|
|
ASSERT_STREQ(oci_spec->hooks->poststart[0]->args[2], "arg7");
|
||
|
|
ASSERT_EQ(oci_spec->hooks->poststart[0]->env_len, 1);
|
||
|
|
ASSERT_STREQ(oci_spec->hooks->poststart[0]->env[0], "key2=value221");
|
||
|
|
ASSERT_EQ(oci_spec->hooks->poststart[0]->timeout, 60);
|
||
|
|
|
||
|
|
ASSERT_STREQ(oci_spec->hooks->poststart[1]->path, "/home/hooks/post2.bash");
|
||
|
|
ASSERT_EQ(oci_spec->hooks->poststart[1]->args_len, 3);
|
||
|
|
ASSERT_STREQ(oci_spec->hooks->poststart[1]->args[0], "arg51");
|
||
|
|
ASSERT_STREQ(oci_spec->hooks->poststart[1]->args[1], "arg61");
|
||
|
|
ASSERT_STREQ(oci_spec->hooks->poststart[1]->args[2], "arg71");
|
||
|
|
ASSERT_EQ(oci_spec->hooks->poststart[1]->env_len, 1);
|
||
|
|
ASSERT_STREQ(oci_spec->hooks->poststart[1]->env[0], "key3=value3");
|
||
|
|
ASSERT_EQ(oci_spec->hooks->poststart[1]->timeout, 61);
|
||
|
|
|
||
|
|
free_oci_runtime_spec_hooks(hooks_spec);
|
||
|
|
hooks_spec = NULL;
|
||
|
|
free_oci_runtime_spec(oci_spec);
|
||
|
|
oci_spec = NULL;
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST(merge_hooks_llt, test_merge_hooks_llt_poststop)
|
||
|
|
{
|
||
|
|
char *hooks_config_file = NULL;
|
||
|
|
oci_runtime_spec_hooks *hooks_spec = NULL;
|
||
|
|
oci_runtime_spec *oci_spec = NULL;
|
||
|
|
char *err = NULL;
|
||
|
|
|
||
|
|
// All parameter correct
|
||
|
|
hooks_config_file = json_path(HOOKS_CONFIG_FILE);
|
||
|
|
ASSERT_TRUE(hooks_config_file != NULL);
|
||
|
|
hooks_spec = oci_runtime_spec_hooks_parse_file(hooks_config_file, NULL, &err);
|
||
|
|
ASSERT_TRUE(hooks_spec != NULL);
|
||
|
|
free(err);
|
||
|
|
err = NULL;
|
||
|
|
free(hooks_config_file);
|
||
|
|
hooks_config_file = NULL;
|
||
|
|
|
||
|
|
oci_spec = (oci_runtime_spec *) util_common_calloc_s(sizeof(oci_runtime_spec));
|
||
|
|
ASSERT_TRUE(oci_spec != NULL);
|
||
|
|
oci_spec->hooks = (oci_runtime_spec_hooks*)util_common_calloc_s(sizeof(oci_runtime_spec_hooks));
|
||
|
|
|
||
|
|
ASSERT_EQ(merge_hooks(oci_spec->hooks, hooks_spec), 0);
|
||
|
|
ASSERT_EQ(oci_spec->hooks->poststop_len, 2);
|
||
|
|
ASSERT_STREQ(oci_spec->hooks->poststop[0]->path, "/home/hooks/stop1.bash");
|
||
|
|
ASSERT_EQ(oci_spec->hooks->poststop[0]->args_len, 3);
|
||
|
|
ASSERT_STREQ(oci_spec->hooks->poststop[0]->args[0], "arg11");
|
||
|
|
ASSERT_STREQ(oci_spec->hooks->poststop[0]->args[1], "arg12");
|
||
|
|
ASSERT_STREQ(oci_spec->hooks->poststop[0]->args[2], "arg13");
|
||
|
|
ASSERT_EQ(oci_spec->hooks->poststop[0]->env_len, 1);
|
||
|
|
ASSERT_STREQ(oci_spec->hooks->poststop[0]->env[0], "key2=value221");
|
||
|
|
ASSERT_EQ(oci_spec->hooks->poststop[0]->timeout, 60);
|
||
|
|
|
||
|
|
ASSERT_STREQ(oci_spec->hooks->poststop[1]->path, "/home/hooks/stop2.bash");
|
||
|
|
ASSERT_EQ(oci_spec->hooks->poststop[1]->args_len, 3);
|
||
|
|
ASSERT_STREQ(oci_spec->hooks->poststop[1]->args[0], "arg52");
|
||
|
|
ASSERT_STREQ(oci_spec->hooks->poststop[1]->args[1], "arg62");
|
||
|
|
ASSERT_STREQ(oci_spec->hooks->poststop[1]->args[2], "arg72");
|
||
|
|
ASSERT_EQ(oci_spec->hooks->poststop[1]->env_len, 1);
|
||
|
|
ASSERT_STREQ(oci_spec->hooks->poststop[1]->env[0], "key4=value4");
|
||
|
|
ASSERT_EQ(oci_spec->hooks->poststop[1]->timeout, 62);
|
||
|
|
|
||
|
|
free_oci_runtime_spec_hooks(hooks_spec);
|
||
|
|
hooks_spec = NULL;
|
||
|
|
free_oci_runtime_spec(oci_spec);
|
||
|
|
oci_spec = NULL;
|
||
|
|
}
|