1 files changed, 42 insertions, 0 deletions
diff --git a/vendor_suffix_test.cpp b/vendor_suffix_test.cpp
new file mode 100644
index 000000000..bd0642234
--- /dev/null
+++ b/vendor_suffix_test.cpp
@@ -0,0 +1,42 @@
+#include <link.h>
+
+#include <string>
+#include <vector>
+
+#include <android-base/stringprintf.h>
+
+#include <gtest/gtest.h>
+
+#include <google/protobuf/message_lite.h>
+
+TEST(vendor_suffix, suffix) {
+    std::vector<std::string> libs;
+    dl_iterate_phdr([](dl_phdr_info* info, size_t, void* data) -> int {
+        auto local_libs = static_cast<decltype(&libs)>(data);
+        std::string name = info->dlpi_name;
+        size_t libprotobuf = name.find("libprotobuf-cpp");
+        if (libprotobuf != name.npos) {
+            local_libs->push_back(name.substr(libprotobuf, name.size()));
+        }
+        return 0;
+    }, &libs);
+
+    std::sort(libs.begin(), libs.end());
+
+    std::string version = android::base::StringPrintf("-%d.%d.%d",
+       GOOGLE_PROTOBUF_VERSION / 1000000,
+       GOOGLE_PROTOBUF_VERSION / 1000 % 1000,
+       GOOGLE_PROTOBUF_VERSION % 1000);
+
+    std::string suffix = GOOGLE_PROTOBUF_VERSION_SUFFIX;
+    if (suffix != "") {
+        version += "-" + suffix;
+    }
+
+    std::vector<std::string> expect = {
+        "libprotobuf-cpp-full" + version + ".so",
+        "libprotobuf-cpp-lite" + version + ".so",
+    };
+
+    ASSERT_EQ(expect, libs);
+}