Index: third_party/protobuf/objectivec/Tests/GPBCodedOuputStreamTests.m |
diff --git a/third_party/protobuf/objectivec/Tests/GPBCodedOuputStreamTests.m b/third_party/protobuf/objectivec/Tests/GPBCodedOuputStreamTests.m |
index 77d88033db2be603ee7ed30d9aa1616aec8ae62c..0723b645f75c2c79733dee19fbb95a3b43d42a5f 100644 |
--- a/third_party/protobuf/objectivec/Tests/GPBCodedOuputStreamTests.m |
+++ b/third_party/protobuf/objectivec/Tests/GPBCodedOuputStreamTests.m |
@@ -30,11 +30,30 @@ |
#import "GPBTestUtilities.h" |
-#import "GPBCodedOutputStream.h" |
+#import "GPBCodedOutputStream_PackagePrivate.h" |
#import "GPBCodedInputStream.h" |
#import "GPBUtilities_PackagePrivate.h" |
#import "google/protobuf/Unittest.pbobjc.h" |
+@interface GPBCodedOutputStream (InternalMethods) |
+// Declared in the .m file, expose for testing. |
+- (instancetype)initWithOutputStream:(NSOutputStream *)output |
+ data:(NSMutableData *)data; |
+@end |
+ |
+@interface GPBCodedOutputStream (Helper) |
++ (instancetype)streamWithOutputStream:(NSOutputStream *)output |
+ bufferSize:(size_t)bufferSize; |
+@end |
+ |
+@implementation GPBCodedOutputStream (Helper) |
++ (instancetype)streamWithOutputStream:(NSOutputStream *)output |
+ bufferSize:(size_t)bufferSize { |
+ NSMutableData *data = [NSMutableData dataWithLength:bufferSize]; |
+ return [[[self alloc] initWithOutputStream:output data:data] autorelease]; |
+} |
+@end |
+ |
@interface CodedOutputStreamTests : GPBTestCase |
@end |