/* Copyright (C) CJ Affiliate
 *
 * You may use, distribute and modify this code under  the
 * terms of the  GNU General Public License  version 2  or
 * later.
 *
 * You should have received a copy of the license with this
 * file. If not, you will find a copy in the "LICENSE" file
 * at https://github.com/cjdev/dual-control.
 */

#include "become.h"
#include "sys_unistd.h"
#include "sys_pwd.h"
#include "user.h"
#include "test_util.h"

template<class T>
std::shared_ptr<T> share (T *t)
{
    return std::shared_ptr<T> (t);
}

class mock_unistd : public unistd_ifc
{
private:
    uid_t expected_euid_;
public:
    mutable uid_t actual_uid;
    mock_unistd (uid_t expected_euid)
        : expected_euid_ (expected_euid)
    {}

    int seteuid (uid_t euid) const override
    {
        actual_uid = euid;
        return euid == expected_euid_ ? 0 : -1;
    }
};

class fake_user : public user_ifc
{
private:
    uid_t uid_;
public:
    fake_user (uid_t uid) : uid_ (uid)
    {}
    uid_t uid() const override
    {
        return uid_;
    }
};

int become_calls_seteuid_with_right_arguments ()
{
    uid_t expected_uid = 1000;
    std::shared_ptr<mock_unistd> mock_unistd  = share (new class mock_unistd (
                expected_uid));
    unistd fake_unistd (mock_unistd);
    user fake_user (share (new class fake_user (expected_uid)));

    become become_ (fake_user, fake_unistd);

    check (mock_unistd->actual_uid == expected_uid, "become does not seteuid");

    succeed();
}

bool run_tests()
{
    test (become_calls_seteuid_with_right_arguments);
    succeed();
}

int main (int argc, char *argv[])
{
    return !run_tests();
}