1fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// Protocol Buffers - Google's data interchange format 2fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// Copyright 2008 Google Inc. All rights reserved. 3fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// http://code.google.com/p/protobuf/ 4fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// 5fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// Redistribution and use in source and binary forms, with or without 6fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// modification, are permitted provided that the following conditions are 7fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// met: 8fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// 9fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// * Redistributions of source code must retain the above copyright 10fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// notice, this list of conditions and the following disclaimer. 11fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// * Redistributions in binary form must reproduce the above 12fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// copyright notice, this list of conditions and the following disclaimer 13fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// in the documentation and/or other materials provided with the 14fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// distribution. 15fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// * Neither the name of Google Inc. nor the names of its 16fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// contributors may be used to endorse or promote products derived from 17fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// this software without specific prior written permission. 18fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// 19fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 20fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 21fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 22fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 23fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 24fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 25fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 26fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 27fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 28fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 29fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 30fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 31fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// Author: kenton@google.com (Kenton Varda) 32fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// Based on original Protocol Buffers design by 33fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// Sanjay Ghemawat, Jeff Dean, and others. 34fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// 35fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// This test insures that google/protobuf/descriptor.pb.{h,cc} match exactly 36fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// what would be generated by the protocol compiler. These files are not 37fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// generated automatically at build time because they are compiled into the 38fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// protocol compiler itself. So, if they were auto-generated, you'd have a 39fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// chicken-and-egg problem. 40fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// 41fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// If this test fails, run the script 42fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// "generate_descriptor_proto.sh" and add 43fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville// descriptor.pb.{h,cc} to your changelist. 44fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 45fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville#include <map> 46fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 47fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville#include <google/protobuf/compiler/cpp/cpp_generator.h> 48fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville#include <google/protobuf/compiler/importer.h> 49fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville#include <google/protobuf/descriptor.h> 50fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville#include <google/protobuf/io/zero_copy_stream_impl.h> 51fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville#include <google/protobuf/stubs/stl_util-inl.h> 52fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville#include <google/protobuf/stubs/map-util.h> 53fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville#include <google/protobuf/stubs/strutil.h> 54fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville#include <google/protobuf/stubs/substitute.h> 55fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 56fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville#include <google/protobuf/testing/file.h> 57fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville#include <google/protobuf/testing/googletest.h> 58fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville#include <gtest/gtest.h> 59fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 60fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillenamespace google { 61fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillenamespace protobuf { 62fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillenamespace compiler { 63fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillenamespace cpp { 64fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 65fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savillenamespace { 66fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 67fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleclass MockErrorCollector : public MultiFileErrorCollector { 68fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville public: 69fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville MockErrorCollector() {} 70fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville ~MockErrorCollector() {} 71fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 72fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville string text_; 73fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 74fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville // implements ErrorCollector --------------------------------------- 75fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville void AddError(const string& filename, int line, int column, 76fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville const string& message) { 77fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville strings::SubstituteAndAppend(&text_, "$0:$1:$2: $3\n", 78fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville filename, line, column, message); 79fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville } 80fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}; 81fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 82fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Savilleclass MockOutputDirectory : public OutputDirectory { 83fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville public: 84fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville MockOutputDirectory() {} 85fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville ~MockOutputDirectory() { 86fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville STLDeleteValues(&files_); 87fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville } 88fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 89fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville void ExpectFileMatches(const string& virtual_filename, 90fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville const string& physical_filename) { 91fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville string* expected_contents = FindPtrOrNull(files_, virtual_filename); 92fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville ASSERT_TRUE(expected_contents != NULL) 93fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville << "Generator failed to generate file: " << virtual_filename; 94fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 95fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville string actual_contents; 96fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville File::ReadFileToStringOrDie( 97fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville TestSourceDir() + "/" + physical_filename, 98fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville &actual_contents); 99fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville EXPECT_TRUE(actual_contents == *expected_contents) 100fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville << physical_filename << " needs to be regenerated. Please run " 101fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville "generate_descriptor_proto.sh and add this file " 102fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville "to your CL."; 103fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville } 104fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 105fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville // implements OutputDirectory -------------------------------------- 106fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 107fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville virtual io::ZeroCopyOutputStream* Open(const string& filename) { 108fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville string** map_slot = &files_[filename]; 109fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville if (*map_slot != NULL) delete *map_slot; 110fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville *map_slot = new string; 111fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 112fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville return new io::StringOutputStream(*map_slot); 113fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville } 114fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 115fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville private: 116fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville map<string, string*> files_; 117fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville}; 118fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 119fbaaef999ba563838ebd00874ed8a1c01fbf286dWink SavilleTEST(BootstrapTest, GeneratedDescriptorMatches) { 120fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville MockErrorCollector error_collector; 121fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville DiskSourceTree source_tree; 122fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville source_tree.MapPath("", TestSourceDir()); 123fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville Importer importer(&source_tree, &error_collector); 124fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville const FileDescriptor* proto_file = 125fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville importer.Import("google/protobuf/descriptor.proto"); 126d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville const FileDescriptor* plugin_proto_file = 127d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville importer.Import("google/protobuf/compiler/plugin.proto"); 128fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville EXPECT_EQ("", error_collector.text_); 129fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville ASSERT_TRUE(proto_file != NULL); 130d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville ASSERT_TRUE(plugin_proto_file != NULL); 131fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 132fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville CppGenerator generator; 133fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville MockOutputDirectory output_directory; 134fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville string error; 135fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville string parameter; 136fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville parameter = "dllexport_decl=LIBPROTOBUF_EXPORT"; 137fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville ASSERT_TRUE(generator.Generate(proto_file, parameter, 138fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville &output_directory, &error)); 139d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville parameter = "dllexport_decl=LIBPROTOC_EXPORT"; 140d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville ASSERT_TRUE(generator.Generate(plugin_proto_file, parameter, 141d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville &output_directory, &error)); 142fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 143fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville output_directory.ExpectFileMatches("google/protobuf/descriptor.pb.h", 144fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville "google/protobuf/descriptor.pb.h"); 145fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville output_directory.ExpectFileMatches("google/protobuf/descriptor.pb.cc", 146fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville "google/protobuf/descriptor.pb.cc"); 147d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville output_directory.ExpectFileMatches("google/protobuf/compiler/plugin.pb.h", 148d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville "google/protobuf/compiler/plugin.pb.h"); 149d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville output_directory.ExpectFileMatches("google/protobuf/compiler/plugin.pb.cc", 150d0332953cda33fb4f8e24ebff9c49159b69c43d6Wink Saville "google/protobuf/compiler/plugin.pb.cc"); 151fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville} 152fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 153fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville} // namespace 154fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville 155fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville} // namespace cpp 156fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville} // namespace compiler 157fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville} // namespace protobuf 158fbaaef999ba563838ebd00874ed8a1c01fbf286dWink Saville} // namespace google 159