/************************************************************************
  * EOS - the CERN Disk Storage System                                   *
  * Copyright (C) 2022 CERN/Switzerland                           *
  *                                                                      *
  * This program is free software: you can redistribute it and/or modify *
  * it under the terms of the GNU General Public License as published by *
  * the Free Software Foundation, either version 3 of the License, or    *
  * (at your option) any later version.                                  *
  *                                                                      *
  * This program is distributed in the hope that it will be useful,      *
  * but WITHOUT ANY WARRANTY; without even the implied warranty of       *
  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the        *
  * GNU General Public License for more details.                         *
  *                                                                      *
  * You should have received a copy of the GNU General Public License    *
  * along with this program.  If not, see <http://www.gnu.org/licenses/>.*
  ************************************************************************
*/

#include "mgm/utils/AttrHelper.hh"
#include "mgm/Constants.hh"
#include "gtest/gtest.h"

using namespace eos::mgm;
using eos::mgm::attr::checkDirOwner;

TEST(checkDirOwner, EmptyMap)
{
  eos::common::VirtualIdentity vid;
  bool sticky_owner;
  ASSERT_FALSE(checkDirOwner({}, {}, {}, vid, sticky_owner, nullptr));
}

TEST(checkDirOwner, StickyOwner)
{
  eos::common::VirtualIdentity vid;
  eos::IContainerMD::XAttrMap xattrs {{SYS_OWNER_AUTH, "*"}};
  bool sticky_owner;
  ASSERT_TRUE(checkDirOwner(xattrs, {}, {}, vid, sticky_owner, nullptr));
  ASSERT_TRUE(sticky_owner);
}

TEST(checkDirOwner, dirOwner)
{
  eos::common::VirtualIdentity vid;
  vid.uid = 23;
  vid.gid = 23;
  uid_t dir_uid = 46;
  gid_t dir_gid = 46;
  vid.prot = "krb5";
  vid.uid_string = "testuser";
  bool sticky_owner;
  eos::IContainerMD::XAttrMap  xattrs {{SYS_OWNER_AUTH, "sss:operator,krb5:testuser"}};
  EXPECT_TRUE(
    checkDirOwner(xattrs, dir_uid, dir_gid, vid, sticky_owner, nullptr));
  ASSERT_FALSE(sticky_owner);
  ASSERT_EQ(vid.uid, dir_uid);
  ASSERT_EQ(vid.gid, dir_gid);
}

TEST(checkAtomicUpload, EmptyMap)
{
  ASSERT_FALSE(attr::checkAtomicUpload({}));
}

TEST(checkAtomicUpload, sys)
{
  eos::IContainerMD::XAttrMap xattrs {{SYS_FORCED_ATOMIC, "1"}};
  ASSERT_TRUE(attr::checkAtomicUpload(xattrs));
  xattrs[SYS_FORCED_ATOMIC] = "0";
  ASSERT_FALSE(attr::checkAtomicUpload(xattrs));
  xattrs[SYS_FORCED_ATOMIC] = "-1";
  ASSERT_TRUE(attr::checkAtomicUpload(xattrs));
  xattrs[SYS_FORCED_ATOMIC] = "garbage";
  ASSERT_FALSE(attr::checkAtomicUpload(xattrs));
}

TEST(checkAtomicUpload, user)
{
  eos::IContainerMD::XAttrMap xattrs {{USER_FORCED_ATOMIC, "1"}};
  ASSERT_TRUE(attr::checkAtomicUpload(xattrs));
  xattrs[USER_FORCED_ATOMIC] = "0";
  ASSERT_FALSE(attr::checkAtomicUpload(xattrs));
  xattrs[USER_FORCED_ATOMIC] = "-1";
  ASSERT_TRUE(attr::checkAtomicUpload(xattrs));
  xattrs[USER_FORCED_ATOMIC] = "garbage";
  ASSERT_FALSE(attr::checkAtomicUpload(xattrs));
}

TEST(checkAtomicUpload, cgi)
{
  eos::IContainerMD::XAttrMap xattrs{};
  ASSERT_TRUE(attr::checkAtomicUpload(xattrs, "foo"));
  // sys overrides everything!
  xattrs[SYS_FORCED_ATOMIC] = "0";
  ASSERT_FALSE(attr::checkAtomicUpload(xattrs, "foo"));
  // usr overrides cgi
  xattrs.clear();
  xattrs[USER_FORCED_ATOMIC] = "0";
  ASSERT_FALSE(attr::checkAtomicUpload(xattrs, "foo"));
}

TEST(checkAtomicUpload, sysoverride)
{
  eos::IContainerMD::XAttrMap xattrs;
  xattrs[SYS_FORCED_ATOMIC] = "0";
  xattrs[USER_FORCED_ATOMIC] = "1";
  ASSERT_FALSE(attr::checkAtomicUpload(xattrs));
  ASSERT_FALSE(attr::checkAtomicUpload(xattrs, "foo"));
}

TEST(getVersioning, cgi)
{
  eos::IContainerMD::XAttrMap xattrs;
  std::string version {"1"};
  ASSERT_EQ(attr::getVersioning(xattrs, version), 1);
  version = "2";
  ASSERT_EQ(attr::getVersioning(xattrs, version), 2);
}

TEST(getVersioning, invalid_cgi)
{
  ASSERT_FALSE(attr::getVersioning({}, "garbage"));
  eos::IContainerMD::XAttrMap xattrs {{SYS_VERSIONING, "0"},
    {USER_VERSIONING, "1"}};
  ASSERT_FALSE(attr::getVersioning(xattrs, "garbage"));
  xattrs.clear();
  xattrs[SYS_VERSIONING] = "1";
  xattrs[USER_VERSIONING] = "0";
  ASSERT_FALSE(attr::getVersioning(xattrs, "garbage"));
}

TEST(getVersioning, cgi_overrides)
{
  eos::IContainerMD::XAttrMap xattrs {{SYS_VERSIONING, "0"},
    {USER_VERSIONING, "1"}};
  std::string version {"2"};
  ASSERT_EQ(attr::getVersioning(xattrs, version), 2);
}

TEST(getVersioning, sys_overrides)
{
  eos::IContainerMD::XAttrMap xattrs;
  xattrs[SYS_VERSIONING] = "1";
  xattrs[USER_VERSIONING] = "0";
  ASSERT_TRUE(attr::getVersioning(xattrs));
  xattrs[SYS_VERSIONING] = "10";
  ASSERT_EQ(attr::getVersioning(xattrs), 10);
  // sys overrides usr, so a garbage sys value will mean 0 versions!
  xattrs[SYS_VERSIONING] = "garbage";
  xattrs[USER_VERSIONING] = "1";
  ASSERT_FALSE(attr::getVersioning(xattrs));
}

TEST(getVersioning, user)
{
  eos::IContainerMD::XAttrMap xattrs {{USER_VERSIONING, "1"}};
  ASSERT_TRUE(attr::getVersioning(xattrs));
  xattrs[USER_VERSIONING] = "10";
  ASSERT_EQ(attr::getVersioning(xattrs), 10);
}