/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this file,
 * You can obtain one at http://mozilla.org/MPL/2.0/. */

#include "OriginParser.h"
#include "gtest/gtest.h"
#include "mozilla/OriginAttributes.h"
#include "nsFmtString.h"

namespace mozilla::dom::quota {

TEST(DOM_Quota_OriginParser_IsUUIDOrigin, Valid)
{
  EXPECT_TRUE(IsUUIDOrigin("uuid://1ef9867c-e754-4303-a18b-684f0321f6e2"_ns));
}

TEST(DOM_Quota_OriginParser_IsUUIDOrigin, Invalid)
{
  EXPECT_FALSE(IsUUIDOrigin("Invalid UUID Origin"_ns));

  EXPECT_FALSE(IsUUIDOrigin("1ef9867c-e754-4303-a18b-684f0321f6e2"_ns));

  EXPECT_FALSE(IsUUIDOrigin("uuid://1ef9867c-e754-4303-a18b"_ns));

  EXPECT_FALSE(IsUUIDOrigin("uuid+++1ef9867c-e754-4303-a18b-684f0321f6e2"_ns));
}

TEST(DOM_Quota_OriginParser_IsUserContextSuffix, True)
{
  const uint32_t userContextId = 5;

  EXPECT_TRUE(IsUserContextSuffix(
      nsFmtCString("^userContextId={}", userContextId), userContextId));
  EXPECT_TRUE(IsUserContextSuffix(
      nsFmtCString("^inBrowser=1&userContextId={}", userContextId),
      userContextId));
}

TEST(DOM_Quota_OriginParser_IsUserContextSuffix, False)
{
  const uint32_t userContextId = 5;

  EXPECT_FALSE(IsUserContextSuffix(""_ns, userContextId));
  EXPECT_FALSE(IsUserContextSuffix("^inBrowser=1"_ns, userContextId));
  EXPECT_FALSE(IsUserContextSuffix("^userContextId=1"_ns, userContextId));
  EXPECT_FALSE(
      IsUserContextSuffix("^inBrowser=1&userContextId=1"_ns, userContextId));
}

TEST(DOM_Quota_OriginParser_IsUserContextPattern, True)
{
  const uint32_t userContextId = 5;

  {
    OriginAttributesPattern pattern;
    pattern.Init(nsFmtString(u"{{ \"userContextId\": {} }}", userContextId));
    EXPECT_TRUE(IsUserContextPattern(pattern, userContextId));
  }

  {
    OriginAttributesPattern pattern;
    pattern.Init(
        nsFmtString(u"{{ \"userContextId\": 5, \"privateBrowsingId\": 1 }}",
                    userContextId));
    EXPECT_TRUE(IsUserContextPattern(pattern, userContextId));
  }
}

TEST(DOM_Quota_OriginParser_IsUserContextPattern, False)
{
  const uint32_t userContextId = 5;

  {
    OriginAttributesPattern pattern;
    pattern.Init(u"{ \"inBrowser\": 1 }"_ns);
    EXPECT_FALSE(IsUserContextPattern(pattern, userContextId));
  }

  {
    OriginAttributesPattern pattern;
    pattern.Init(u"{ \"userContextId\": 1 }"_ns);
    EXPECT_FALSE(IsUserContextPattern(pattern, userContextId));
  }
}

}  //  namespace mozilla::dom::quota