From 7da29c6a138c90bc37b6559ce908d1c2a67edfaf Mon Sep 17 00:00:00 2001
From: Protobuf Team Bot <protobuf-github-bot@google.com>
Date: Thu, 8 Feb 2024 17:34:26 -0800
Subject: [PATCH] Minor cleanup: IWYU, etc.

PiperOrigin-RevId: 605475996
---
 src/google/protobuf/BUILD.bazel                   | 1 +
 src/google/protobuf/message_unittest.cc           | 7 ++++++-
 src/google/protobuf/sample_messages_edition.proto | 5 +----
 3 files changed, 8 insertions(+), 5 deletions(-)

diff --git a/src/google/protobuf/BUILD.bazel b/src/google/protobuf/BUILD.bazel
index bc0f3e5345..4e8d8b843d 100644
--- a/src/google/protobuf/BUILD.bazel
+++ b/src/google/protobuf/BUILD.bazel
@@ -1487,6 +1487,7 @@ cc_test(
     deps = [
         ":arena",
         ":cc_test_protos",
+        ":internal_visibility",
         ":port",
         ":protobuf",
         ":protobuf_lite",
diff --git a/src/google/protobuf/message_unittest.cc b/src/google/protobuf/message_unittest.cc
index c663165991..e8fe7f796d 100644
--- a/src/google/protobuf/message_unittest.cc
+++ b/src/google/protobuf/message_unittest.cc
@@ -9,6 +9,12 @@
 //  Based on original Protocol Buffers design by
 //  Sanjay Ghemawat, Jeff Dean, and others.
 
+#include <gtest/gtest.h>
+#include "absl/log/absl_check.h"
+#include "google/protobuf/arena.h"
+#include "google/protobuf/explicitly_constructed.h"
+#include "google/protobuf/has_bits.h"
+#include "google/protobuf/internal_visibility.h"
 #include "google/protobuf/unittest.pb.h"
 
 #define MESSAGE_TEST_NAME MessageTest
@@ -21,5 +27,4 @@
 // clang-format off
 #include "google/protobuf/test_util.inc"
 #include "google/protobuf/message_unittest.inc"
-#include "google/protobuf/arena.h"
 // clang-format on
diff --git a/src/google/protobuf/sample_messages_edition.proto b/src/google/protobuf/sample_messages_edition.proto
index c0310dac39..464a816d9e 100644
--- a/src/google/protobuf/sample_messages_edition.proto
+++ b/src/google/protobuf/sample_messages_edition.proto
@@ -5,10 +5,7 @@
 // license that can be found in the LICENSE file or at
 // https://developers.google.com/open-source/licenses/bsd
 //
-// Test schema for proto2 messages.  This test schema is used by:
-//
-// - conformance tests
-//
+// Sample messages to generate example code.
 
 edition = "2023";