43 lines
987 B
C++
43 lines
987 B
C++
|
#include "../lib/src/HttpUtils.h"
|
||
|
|
#include <gtest/gtest.h>
|
||
|
|
#include <string>
|
||
|
|
using namespace drogon;
|
||
|
|
|
||
|
|
TEST(ExtensionTest, normal)
|
||
|
|
{
|
||
|
|
std::string str{"drogon.jpg"};
|
||
|
|
EXPECT_TRUE(getFileExtension(str) == "jpg");
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST(ExtensionTest, negative)
|
||
|
|
{
|
||
|
|
std::string str{"drogon."};
|
||
|
|
EXPECT_TRUE(getFileExtension(str) == "");
|
||
|
|
str = "drogon";
|
||
|
|
EXPECT_TRUE(getFileExtension(str) == "");
|
||
|
|
str = "";
|
||
|
|
EXPECT_TRUE(getFileExtension(str) == "");
|
||
|
|
str = "....";
|
||
|
|
EXPECT_TRUE(getFileExtension(str) == "");
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST(FileTypeTest, normal)
|
||
|
|
{
|
||
|
|
EXPECT_EQ(parseFileType("jpg"), FT_IMAGE);
|
||
|
|
EXPECT_EQ(parseFileType("mp4"), FT_MEDIA);
|
||
|
|
EXPECT_EQ(parseFileType("csp"), FT_CUSTOM);
|
||
|
|
EXPECT_EQ(parseFileType("html"), FT_DOCUMENT);
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST(FileTypeTest, negative)
|
||
|
|
{
|
||
|
|
EXPECT_EQ(parseFileType(""), FT_UNKNOWN);
|
||
|
|
EXPECT_EQ(parseFileType("don'tknow"), FT_CUSTOM);
|
||
|
|
}
|
||
|
|
|
||
|
|
int main(int argc, char **argv)
|
||
|
|
{
|
||
|
|
testing::InitGoogleTest(&argc, argv);
|
||
|
|
return RUN_ALL_TESTS();
|
||
|
|
}
|