blob: 84acc3af3ac193545ac10d07819e027e4685a5db [file] [log] [blame]
// Copyright 2022 The PDFium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "core/fpdfapi/parser/cpdf_dictionary.h"
#include <utility>
#include "core/fpdfapi/parser/cpdf_array.h"
#include "core/fpdfapi/parser/cpdf_number.h"
#include "core/fpdfapi/parser/cpdf_stream.h"
#include "testing/gtest/include/gtest/gtest.h"
TEST(DictionaryTest, Iterators) {
auto dict = pdfium::MakeRetain<CPDF_Dictionary>();
dict->SetNewFor<CPDF_Dictionary>("the-dictionary");
dict->SetNewFor<CPDF_Array>("the-array");
dict->SetNewFor<CPDF_Stream>("the-stream");
dict->SetNewFor<CPDF_Number>("the-number", 42);
CPDF_DictionaryLocker locked_dict(dict);
auto it = locked_dict.begin();
EXPECT_NE(it, locked_dict.end());
EXPECT_EQ(it->first, ByteString("the-array"));
EXPECT_TRUE(it->second->IsArray());
++it;
EXPECT_NE(it, locked_dict.end());
EXPECT_EQ(it->first, ByteString("the-dictionary"));
EXPECT_TRUE(it->second->IsDictionary());
++it;
EXPECT_NE(it, locked_dict.end());
EXPECT_EQ(it->first, ByteString("the-number"));
EXPECT_TRUE(it->second->IsNumber());
++it;
EXPECT_NE(it, locked_dict.end());
EXPECT_EQ(it->first, ByteString("the-stream"));
EXPECT_TRUE(it->second->IsStream());
++it;
EXPECT_EQ(it, locked_dict.end());
}