PowerShell/src/libpsl-native/test/test-createsymlink.cpp

//! @file test-createsymlink.cpp
//! @author George Fleming <v-geflem@microsoft.com>
//! @brief Implements test for CreateSymLink() and FollowSymLink()

#include <gtest/gtest.h>
#include <errno.h>
#include <unistd.h>
#include "issymlink.h"
#include "createsymlink.h"
#include "followsymlink.h"

using namespace std;

class CreateSymLinkTest : public ::testing::Test
{
protected:

    static const int bufSize = 64;
    const string fileTemplate = "/tmp/symlinktest.fXXXXXX";
    const string dirTemplate = "/tmp/symlinktest.dXXXXXX";
    const string fileSymLink = "/tmp/symlinktest.flink";
    const string dirSymLink = "/tmp/symlinktest.dlink";
    char *file, *dir;
    char fileTemplateBuf[bufSize], dirTemplateBuf[bufSize];

    CreateSymLinkTest()
    {
        // since mkstemp and mkdtemp modifies the template string, let's give them writable buffers
        strcpy(fileTemplateBuf, fileTemplate.c_str());
        strcpy(dirTemplateBuf, dirTemplate.c_str());

        // First create a temp file
        int fd = mkstemp(fileTemplateBuf);
        EXPECT_TRUE(fd != -1);
	file = fileTemplateBuf;

	// Create a temp directory
	dir = mkdtemp(dirTemplateBuf);
        EXPECT_TRUE(dir != NULL);

        // Create symbolic link to file
	int ret1 = CreateSymLink(fileSymLink.c_str(), file);
	EXPECT_EQ(ret1, 1);
        
        // Create symbolic link to directory
	int ret2 = CreateSymLink(dirSymLink.c_str(), dir);
	EXPECT_EQ(ret2, 1);
    }

    ~CreateSymLinkTest()
    {
        int ret;

        ret = unlink(fileSymLink.c_str());
	EXPECT_EQ(0, ret);

        ret = unlink(dirSymLink.c_str());
	EXPECT_EQ(0, ret);

        ret = unlink(file);
	EXPECT_EQ(0, ret);

	ret = rmdir(dir);
	EXPECT_EQ(0, ret);       
    }
};

TEST_F(CreateSymLinkTest, FilePathNameIsNull)
{
    int retVal = CreateSymLink(NULL, NULL);
    EXPECT_EQ(retVal, 0);
    EXPECT_EQ(ERROR_INVALID_PARAMETER, errno);
}

TEST_F(CreateSymLinkTest, FilePathNameDoesNotExist)
{
    std::string invalidFile = "/tmp/symlinktest_invalidFile";
    std::string invalidLink = "/tmp/symlinktest_invalidLink";

    // make sure neither exists    
    unlink(invalidFile.c_str());
    unlink(invalidLink.c_str());

    // Linux allows creation of symbolic link that points to an invalid file
    int retVal = CreateSymLink(invalidLink.c_str(), invalidFile.c_str());
    EXPECT_EQ(retVal, 1);

    std::string target = FollowSymLink(invalidLink.c_str());
    EXPECT_EQ(target, invalidFile);

    unlink(invalidLink.c_str());
}

TEST_F(CreateSymLinkTest, SymLinkToFile)
{
    int retVal = IsSymLink(fileSymLink.c_str());
    EXPECT_EQ(1, retVal);

    std::string target = FollowSymLink(fileSymLink.c_str());
    EXPECT_EQ(target, file);
}

TEST_F(CreateSymLinkTest, SymLinkToDirectory)
{
    int retVal = IsSymLink(dirSymLink.c_str());
    EXPECT_EQ(1, retVal);

    std::string target = FollowSymLink(dirSymLink.c_str());
    EXPECT_EQ(target, dir);
}

TEST_F(CreateSymLinkTest, SymLinkAgain)
{
    int retVal = CreateSymLink(fileSymLink.c_str(), file);
    EXPECT_EQ(0, retVal);
    EXPECT_EQ(ERROR_FILE_EXISTS, errno);
}