chromium/components/user_notes/browser/frame_user_note_changes_unittest.cc

// Copyright 2022 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "components/user_notes/browser/frame_user_note_changes.h"

#include <memory>
#include <vector>

#include "base/memory/safe_ref.h"
#include "base/test/bind.h"
#include "base/test/gmock_callback_support.h"
#include "base/unguessable_token.h"
#include "components/user_notes/browser/user_note_base_test.h"
#include "components/user_notes/browser/user_note_instance.h"
#include "components/user_notes/model/user_note.h"
#include "content/public/browser/render_frame_host.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h"

_;
Mock;

namespace user_notes {

namespace {

// A mock for a note instance that synchronously invokes the callback when
// initializing the text highlight.
class MockUserNoteInstance : public UserNoteInstance { … };

// Partially mock the object under test so calls to `MakeNoteInstance` can be
// intercepted, allowing the tests to create mocked instances.
class MockFrameUserNoteChanges : public FrameUserNoteChanges { … };

void MockInitializeHighlightIfNeeded(base::OnceClosure callback) { … }

std::unique_ptr<UserNoteInstance> MockMakeNoteInstance(
    const UserNote* note_model,
    UserNoteManager* manager) { … }

}  // namespace

class FrameUserNoteChangesTest : public UserNoteBaseTest { … };

// Tests that added notes correctly kick off highlight initialization on the
// renderer side, and new instances are correctly added to the note manager.
TEST_F(FrameUserNoteChangesTest, ApplyAddedNotes) { … }

// Tests that modified notes don't impact instances in the note manager.
TEST_F(FrameUserNoteChangesTest, ApplyModifiedNotes) { … }

// Tests that removed notes correctly have their instances removed from the
// note manager.
TEST_F(FrameUserNoteChangesTest, ApplyRemovedNotes) { … }

}  // namespace user_notes