Source code
Revision control
Copy as Markdown
Other Tools
/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* 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 "SimpleMap.h"
#include "gtest/gtest.h"
#include "nsString.h"
namespace mozilla {
TEST(SimpleMapTest, Insert)
{
SimpleMap<int, nsCString> map;
map.Insert(1, "first"_ns);
ASSERT_EQ(map.Count(), size_t(1));
Maybe<nsCString> taken = map.Take(1);
ASSERT_TRUE(taken.isSome());
ASSERT_TRUE(taken.ref().EqualsLiteral("first"));
ASSERT_EQ(map.Count(), size_t(0));
Maybe<nsCString> shouldBeNothing = map.Take(1);
ASSERT_TRUE(shouldBeNothing.isNothing());
}
TEST(SimpleMapTest, DuplicateKeys)
{
SimpleMap<int, nsCString> map;
map.Insert(1, "a1"_ns);
map.Insert(2, "b1"_ns);
map.Insert(1, "a2"_ns);
map.Insert(2, "b2"_ns);
ASSERT_EQ(map.Count(), size_t(4));
// Take returns entries in FIFO order within each key.
Maybe<nsCString> v = map.Take(1);
ASSERT_TRUE(v.isSome());
ASSERT_TRUE(v.ref().EqualsLiteral("a1"));
v = map.Take(2);
ASSERT_TRUE(v.isSome());
ASSERT_TRUE(v.ref().EqualsLiteral("b1"));
v = map.Take(1);
ASSERT_TRUE(v.isSome());
ASSERT_TRUE(v.ref().EqualsLiteral("a2"));
v = map.Take(2);
ASSERT_TRUE(v.isSome());
ASSERT_TRUE(v.ref().EqualsLiteral("b2"));
ASSERT_EQ(map.Count(), size_t(0));
}
TEST(SimpleMapTest, Contains)
{
SimpleMap<int, nsCString> map;
ASSERT_FALSE(map.Contains(1));
map.Insert(1, "one"_ns);
ASSERT_TRUE(map.Contains(1));
ASSERT_FALSE(map.Contains(2));
}
TEST(SimpleMapTest, Take)
{
SimpleMap<int, nsCString> map;
ASSERT_EQ(map.Count(), size_t(0));
map.Insert(1, "one"_ns);
ASSERT_EQ(map.Count(), size_t(1));
map.Insert(2, "two"_ns);
ASSERT_EQ(map.Count(), size_t(2));
Maybe<nsCString> taken = map.Take(1);
ASSERT_TRUE(taken.isSome());
ASSERT_TRUE(taken.ref().EqualsLiteral("one"));
ASSERT_FALSE(map.Contains(1));
ASSERT_TRUE(map.Contains(2));
ASSERT_EQ(map.Count(), size_t(1));
Maybe<nsCString> notTaken = map.Take(3);
ASSERT_TRUE(notTaken.isNothing());
}
TEST(SimpleMapTest, Clear)
{
SimpleMap<int, nsCString> map;
ASSERT_EQ(map.Count(), size_t(0));
map.Insert(1, "one"_ns);
map.Insert(2, "two"_ns);
ASSERT_EQ(map.Count(), size_t(2));
ASSERT_TRUE(map.Contains(1));
ASSERT_TRUE(map.Contains(2));
map.Clear();
ASSERT_FALSE(map.Contains(1));
ASSERT_FALSE(map.Contains(2));
ASSERT_EQ(map.Count(), size_t(0));
}
TEST(SimpleMapTest, ClearWithCallback)
{
SimpleMap<int, nsCString> map;
map.Insert(1, "one"_ns);
map.Insert(2, "two"_ns);
map.Insert(3, "three"_ns);
nsTArray<int> keys;
map.Clear([&](int key, const nsCString& value) { keys.AppendElement(key); });
ASSERT_EQ(map.Count(), size_t(0));
ASSERT_EQ(keys.Length(), size_t(3));
ASSERT_TRUE(keys.Contains(1) && keys.Contains(2) && keys.Contains(3));
}
} // namespace mozilla