blob: 553905649ff04c70ce231773f144fd6e33367334 [file] [log] [blame]
// Copyright 2019 The PDFium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include <vector>
#include "fpdfsdk/cpdfsdk_annotiterator.h"
#include "fpdfsdk/cpdfsdk_baannothandler.h"
#include "fpdfsdk/cpdfsdk_formfillenvironment.h"
#include "fpdfsdk/cpdfsdk_helpers.h"
#include "fpdfsdk/cpdfsdk_pageview.h"
#include "testing/embedder_test.h"
class CPDFSDK_BAAnnotHandlerTest : public EmbedderTest {
public:
void SetUp() override {
EmbedderTest::SetUp();
SetUpBAAnnotHandler();
}
void TearDown() override {
UnloadPage(m_page);
EmbedderTest::TearDown();
}
void SetUpBAAnnotHandler() {
ASSERT_TRUE(OpenDocument("links_highlights_annots.pdf"));
m_page = LoadPage(0);
ASSERT_TRUE(m_page);
m_pFormFillEnv =
CPDFSDKFormFillEnvironmentFromFPDFFormHandle(form_handle());
ASSERT_TRUE(m_pFormFillEnv);
m_pPageView =
m_pFormFillEnv->GetPageView(IPDFPageFromFPDFPage(m_page), true);
ASSERT_TRUE(m_pPageView);
CPDFSDK_AnnotHandlerMgr* pAnnotHandlerMgr =
m_pFormFillEnv->GetAnnotHandlerMgr();
ASSERT_TRUE(pAnnotHandlerMgr);
m_pBAAnnotHandler = pAnnotHandlerMgr->m_pBAAnnotHandler.get();
ASSERT_TRUE(m_pBAAnnotHandler);
}
CPDFSDK_FormFillEnvironment* GetFormFillEnv() const { return m_pFormFillEnv; }
CPDFSDK_PageView* GetPageView() const { return m_pPageView; }
CPDFSDK_BAAnnotHandler* GetBAAnnotHandler() const {
return m_pBAAnnotHandler;
}
CPDFSDK_Annot* GetNthFocusableAnnot(size_t n) {
DCHECK_NE(n, 0);
CPDFSDK_AnnotIterator ai(GetPageView(),
m_pFormFillEnv->GetFocusableAnnotSubtypes());
CPDFSDK_Annot* pAnnot = ai.GetFirstAnnot();
ASSERT(pAnnot);
for (size_t i = 1; i < n; i++) {
pAnnot = ai.GetNextAnnot(pAnnot);
ASSERT(pAnnot);
}
return pAnnot;
}
private:
FPDF_PAGE m_page = nullptr;
CPDFSDK_PageView* m_pPageView = nullptr;
CPDFSDK_FormFillEnvironment* m_pFormFillEnv = nullptr;
CPDFSDK_BAAnnotHandler* m_pBAAnnotHandler = nullptr;
};
TEST_F(CPDFSDK_BAAnnotHandlerTest, TabToLinkOrHighlightAnnot) {
std::vector<CPDF_Annot::Subtype> focusable_annot_types = {
CPDF_Annot::Subtype::WIDGET, CPDF_Annot::Subtype::LINK,
CPDF_Annot::Subtype::HIGHLIGHT};
GetFormFillEnv()->SetFocusableAnnotSubtypes(focusable_annot_types);
// Get link annot.
CPDFSDK_Annot* pAnnot = GetNthFocusableAnnot(2);
ASSERT_TRUE(pAnnot);
EXPECT_EQ(pAnnot->GetAnnotSubtype(), CPDF_Annot::Subtype::LINK);
ObservedPtr<CPDFSDK_Annot> pLinkAnnot(pAnnot);
EXPECT_TRUE(GetBAAnnotHandler()->OnSetFocus(&pLinkAnnot, 0));
EXPECT_TRUE(GetBAAnnotHandler()->OnKillFocus(&pLinkAnnot, 0));
// Get highlight annot.
pAnnot = GetNthFocusableAnnot(4);
ASSERT_TRUE(pAnnot);
EXPECT_EQ(pAnnot->GetAnnotSubtype(), CPDF_Annot::Subtype::HIGHLIGHT);
ObservedPtr<CPDFSDK_Annot> pHighlightAnnot(pAnnot);
EXPECT_TRUE(GetBAAnnotHandler()->OnSetFocus(&pHighlightAnnot, 0));
EXPECT_TRUE(GetBAAnnotHandler()->OnKillFocus(&pHighlightAnnot, 0));
}