Files
wehub-resource-sync c8a779b1bb
Docker Image CI / build-ubuntu2004 (push) Waiting to run
chore: import upstream snapshot with attribution
2026-07-13 13:36:55 +08:00

156 lines
4.3 KiB
C++

/*
* SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
* SPDX-License-Identifier: Apache-2.0
*
* Licensed 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 "sampleUtils.h"
#include <gtest/gtest.h>
#include <string_view>
using namespace sample;
using namespace std::string_view_literals;
TEST(RoundUp, ExactMultiple)
{
EXPECT_EQ(roundUp(4, 4), 4);
EXPECT_EQ(roundUp(8, 4), 8);
EXPECT_EQ(roundUp(0, 4), 0);
}
TEST(RoundUp, NeedsRounding)
{
EXPECT_EQ(roundUp(1, 4), 4);
EXPECT_EQ(roundUp(5, 4), 8);
EXPECT_EQ(roundUp(7, 4), 8);
}
TEST(SplitToStringVec, SingleToken)
{
auto const v = splitToStringVec("hello", ',');
ASSERT_EQ(v.size(), 1U);
EXPECT_EQ(v[0], "hello"sv);
}
TEST(SplitToStringVec, MultipleTokens)
{
auto const v = splitToStringVec("a,b,c", ',');
ASSERT_EQ(v.size(), 3U);
EXPECT_EQ(v[0], "a"sv);
EXPECT_EQ(v[1], "b"sv);
EXPECT_EQ(v[2], "c"sv);
}
TEST(SplitToStringVec, EmptyString)
{
auto const v = splitToStringVec("", ',');
EXPECT_TRUE(v.empty());
}
TEST(SplitToStringVec, MaxSplit)
{
// maxSplit=1 means at most one split; the rest of the string is the second element.
auto const v = splitToStringVec("a:b:c", ':', 1);
ASSERT_EQ(v.size(), 2U);
EXPECT_EQ(v[0], "a"sv);
EXPECT_EQ(v[1], "b:c"sv);
}
TEST(SplitToStringVec, TrailingSeparator)
{
auto const v = splitToStringVec("a,b,", ',');
ASSERT_EQ(v.size(), 3U);
EXPECT_EQ(v[0], "a"sv);
EXPECT_EQ(v[1], "b"sv);
EXPECT_EQ(v[2], ""sv);
}
TEST(MatchStringWithOneWildcard, ExactMatch)
{
EXPECT_TRUE(matchStringWithOneWildcard("hello", "hello"));
EXPECT_FALSE(matchStringWithOneWildcard("hello", "world"));
EXPECT_FALSE(matchStringWithOneWildcard("hello", "hello2"));
}
TEST(MatchStringWithOneWildcard, WildcardMatchesAnything)
{
EXPECT_TRUE(matchStringWithOneWildcard("*", "anything"));
EXPECT_TRUE(matchStringWithOneWildcard("*", ""));
}
TEST(MatchStringWithOneWildcard, PrefixWildcard)
{
EXPECT_TRUE(matchStringWithOneWildcard("hello*", "hello"));
EXPECT_TRUE(matchStringWithOneWildcard("hello*", "hello world"));
EXPECT_FALSE(matchStringWithOneWildcard("hello*", "world"));
}
TEST(MatchStringWithOneWildcard, SuffixWildcard)
{
EXPECT_TRUE(matchStringWithOneWildcard("*world", "world"));
EXPECT_TRUE(matchStringWithOneWildcard("*world", "hello world"));
EXPECT_FALSE(matchStringWithOneWildcard("*world", "hello"));
}
TEST(MatchStringWithOneWildcard, MiddleWildcard)
{
EXPECT_TRUE(matchStringWithOneWildcard("he*ld", "held"));
EXPECT_TRUE(matchStringWithOneWildcard("he*ld", "hello world"));
EXPECT_FALSE(matchStringWithOneWildcard("he*ld", "hello"));
}
TEST(NormalizeDirectoryPath, AlreadyNormalized)
{
EXPECT_EQ(normalizeDirectoryPath("/some/path/"), "/some/path/"sv);
}
TEST(NormalizeDirectoryPath, MissingTrailingSlash)
{
EXPECT_EQ(normalizeDirectoryPath("/some/path"), "/some/path/"sv);
}
TEST(NormalizeDirectoryPath, EmptyString)
{
EXPECT_EQ(normalizeDirectoryPath(""), ""sv);
}
TEST(SanitizeRemoteAutoTuningConfig, Empty)
{
EXPECT_EQ(sanitizeRemoteAutoTuningConfig(""), ""sv);
}
TEST(SanitizeRemoteAutoTuningConfig, NoCredentials)
{
// No @ means no credentials section; returned as-is.
EXPECT_EQ(sanitizeRemoteAutoTuningConfig("ssh://host:22"), "ssh://host:22"sv);
}
TEST(SanitizeRemoteAutoTuningConfig, UsernameOnly)
{
EXPECT_EQ(sanitizeRemoteAutoTuningConfig("ssh://user@host:22"), "ssh://***@host:22"sv);
}
TEST(SanitizeRemoteAutoTuningConfig, UsernameAndPassword)
{
EXPECT_EQ(sanitizeRemoteAutoTuningConfig("ssh://user:pass@host:22"), "ssh://***@host:22"sv);
}
TEST(SanitizeRemoteAutoTuningConfig, WithQueryParams)
{
EXPECT_EQ(sanitizeRemoteAutoTuningConfig("ssh://admin:secret@server.com:22?timeout=30"),
"ssh://***@server.com:22?timeout=30"sv);
}