blob: 1bb785f727c1aa27de074680864eb9b56cc33fea [file] [log] [blame]
// Copyright 2016 The PDFium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "fpdfsdk/cpdfsdk_annot.h"
#include "fpdfsdk/cpdfsdk_annotiterator.h"
#include "fpdfsdk/cpdfsdk_formfillenvironment.h"
#include "fpdfsdk/cpdfsdk_helpers.h"
#include "testing/embedder_test.h"
#include "testing/embedder_test_mock_delegate.h"
#include "testing/embedder_test_timer_handling_delegate.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace {
void CheckRect(const CFX_FloatRect& actual, const CFX_FloatRect& expected) {
EXPECT_EQ(expected.left, actual.left);
EXPECT_EQ(expected.bottom, actual.bottom);
EXPECT_EQ(expected.right, actual.right);
EXPECT_EQ(expected.top, actual.top);
}
} // namespace
class CPDFSDK_AnnotIteratorTest : public EmbedderTest {};
TEST_F(CPDFSDK_AnnotIteratorTest, CPDFSDK_AnnotIterator) {
ASSERT_TRUE(OpenDocument("annotiter.pdf"));
FPDF_PAGE page0 = LoadPage(0);
FPDF_PAGE page1 = LoadPage(1);
FPDF_PAGE page2 = LoadPage(2);
ASSERT_TRUE(page0);
ASSERT_TRUE(page1);
ASSERT_TRUE(page2);
CFX_FloatRect LeftBottom(200, 200, 220, 220);
CFX_FloatRect RightBottom(400, 201, 420, 221);
CFX_FloatRect LeftTop(201, 400, 221, 420);
CFX_FloatRect RightTop(401, 401, 421, 421);
CPDFSDK_FormFillEnvironment* pFormFillEnv =
CPDFSDKFormFillEnvironmentFromFPDFFormHandle(form_handle());
{
// Page 0 specifies "row order".
CPDFSDK_AnnotIterator iter(pFormFillEnv->GetPageViewAtIndex(0),
{CPDF_Annot::Subtype::WIDGET});
CPDFSDK_Annot* pAnnot = iter.GetFirstAnnot();
CheckRect(pAnnot->GetRect(), RightTop);
pAnnot = iter.GetNextAnnot(pAnnot);
CheckRect(pAnnot->GetRect(), LeftTop);
pAnnot = iter.GetNextAnnot(pAnnot);
CheckRect(pAnnot->GetRect(), RightBottom);
pAnnot = iter.GetNextAnnot(pAnnot);
CheckRect(pAnnot->GetRect(), LeftBottom);
pAnnot = iter.GetNextAnnot(pAnnot);
EXPECT_FALSE(pAnnot);
pAnnot = iter.GetLastAnnot();
CheckRect(pAnnot->GetRect(), LeftBottom);
pAnnot = iter.GetPrevAnnot(pAnnot);
CheckRect(pAnnot->GetRect(), RightBottom);
pAnnot = iter.GetPrevAnnot(pAnnot);
CheckRect(pAnnot->GetRect(), LeftTop);
pAnnot = iter.GetPrevAnnot(pAnnot);
CheckRect(pAnnot->GetRect(), RightTop);
pAnnot = iter.GetPrevAnnot(pAnnot);
EXPECT_FALSE(pAnnot);
}
{
// Page 1 specifies "column order"
CPDFSDK_AnnotIterator iter(pFormFillEnv->GetPageViewAtIndex(1),
{CPDF_Annot::Subtype::WIDGET});
CPDFSDK_Annot* pAnnot = iter.GetFirstAnnot();
CheckRect(pAnnot->GetRect(), RightTop);
pAnnot = iter.GetNextAnnot(pAnnot);
CheckRect(pAnnot->GetRect(), RightBottom);
pAnnot = iter.GetNextAnnot(pAnnot);
CheckRect(pAnnot->GetRect(), LeftTop);
pAnnot = iter.GetNextAnnot(pAnnot);
CheckRect(pAnnot->GetRect(), LeftBottom);
pAnnot = iter.GetNextAnnot(pAnnot);
EXPECT_FALSE(pAnnot);
pAnnot = iter.GetLastAnnot();
CheckRect(pAnnot->GetRect(), LeftBottom);
pAnnot = iter.GetPrevAnnot(pAnnot);
CheckRect(pAnnot->GetRect(), LeftTop);
pAnnot = iter.GetPrevAnnot(pAnnot);
CheckRect(pAnnot->GetRect(), RightBottom);
pAnnot = iter.GetPrevAnnot(pAnnot);
CheckRect(pAnnot->GetRect(), RightTop);
pAnnot = iter.GetPrevAnnot(pAnnot);
EXPECT_FALSE(pAnnot);
}
{
// Page 2 specifies "struct order"
CPDFSDK_AnnotIterator iter(pFormFillEnv->GetPageViewAtIndex(2),
{CPDF_Annot::Subtype::WIDGET});
CPDFSDK_Annot* pAnnot = iter.GetFirstAnnot();
CheckRect(pAnnot->GetRect(), LeftBottom);
pAnnot = iter.GetNextAnnot(pAnnot);
CheckRect(pAnnot->GetRect(), RightTop);
pAnnot = iter.GetNextAnnot(pAnnot);
CheckRect(pAnnot->GetRect(), LeftTop);
pAnnot = iter.GetNextAnnot(pAnnot);
CheckRect(pAnnot->GetRect(), RightBottom);
pAnnot = iter.GetNextAnnot(pAnnot);
EXPECT_FALSE(pAnnot);
pAnnot = iter.GetLastAnnot();
CheckRect(pAnnot->GetRect(), RightBottom);
pAnnot = iter.GetPrevAnnot(pAnnot);
CheckRect(pAnnot->GetRect(), LeftTop);
pAnnot = iter.GetPrevAnnot(pAnnot);
CheckRect(pAnnot->GetRect(), RightTop);
pAnnot = iter.GetPrevAnnot(pAnnot);
CheckRect(pAnnot->GetRect(), LeftBottom);
pAnnot = iter.GetPrevAnnot(pAnnot);
EXPECT_FALSE(pAnnot);
}
UnloadPage(page2);
UnloadPage(page1);
UnloadPage(page0);
}