GPBMessageTests.m 86 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186
  1. // Protocol Buffers - Google's data interchange format
  2. // Copyright 2008 Google Inc. All rights reserved.
  3. // https://developers.google.com/protocol-buffers/
  4. //
  5. // Redistribution and use in source and binary forms, with or without
  6. // modification, are permitted provided that the following conditions are
  7. // met:
  8. //
  9. // * Redistributions of source code must retain the above copyright
  10. // notice, this list of conditions and the following disclaimer.
  11. // * Redistributions in binary form must reproduce the above
  12. // copyright notice, this list of conditions and the following disclaimer
  13. // in the documentation and/or other materials provided with the
  14. // distribution.
  15. // * Neither the name of Google Inc. nor the names of its
  16. // contributors may be used to endorse or promote products derived from
  17. // this software without specific prior written permission.
  18. //
  19. // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  20. // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  21. // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  22. // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  23. // OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  24. // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  25. // LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  26. // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  27. // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  28. // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  29. // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  30. #import "GPBTestUtilities.h"
  31. #import <objc/runtime.h>
  32. #import "GPBArray_PackagePrivate.h"
  33. #import "GPBDescriptor.h"
  34. #import "GPBDictionary_PackagePrivate.h"
  35. #import "GPBMessage_PackagePrivate.h"
  36. #import "GPBUnknownField_PackagePrivate.h"
  37. #import "GPBUnknownFieldSet_PackagePrivate.h"
  38. #import "google/protobuf/Unittest.pbobjc.h"
  39. #import "google/protobuf/UnittestObjc.pbobjc.h"
  40. #import "google/protobuf/UnittestObjcOptions.pbobjc.h"
  41. @interface MessageTests : GPBTestCase
  42. @end
  43. @implementation MessageTests
  44. // TODO(thomasvl): this should get split into a few files of logic junks, it is
  45. // a jumble of things at the moment (and the testutils have a bunch of the real
  46. // assertions).
  47. - (TestAllTypes *)mergeSource {
  48. TestAllTypes *message = [TestAllTypes message];
  49. [message setOptionalInt32:1];
  50. [message setOptionalString:@"foo"];
  51. [message setOptionalForeignMessage:[ForeignMessage message]];
  52. [message.repeatedStringArray addObject:@"bar"];
  53. return message;
  54. }
  55. - (TestAllTypes *)mergeDestination {
  56. TestAllTypes *message = [TestAllTypes message];
  57. [message setOptionalInt64:2];
  58. [message setOptionalString:@"baz"];
  59. ForeignMessage *foreignMessage = [ForeignMessage message];
  60. [foreignMessage setC:3];
  61. [message setOptionalForeignMessage:foreignMessage];
  62. [message.repeatedStringArray addObject:@"qux"];
  63. return message;
  64. }
  65. - (TestAllTypes *)mergeDestinationWithoutForeignMessageIvar {
  66. TestAllTypes *message = [TestAllTypes message];
  67. [message setOptionalInt64:2];
  68. [message setOptionalString:@"baz"];
  69. [message.repeatedStringArray addObject:@"qux"];
  70. return message;
  71. }
  72. - (TestAllTypes *)mergeResult {
  73. TestAllTypes *message = [TestAllTypes message];
  74. [message setOptionalInt32:1];
  75. [message setOptionalInt64:2];
  76. [message setOptionalString:@"foo"];
  77. ForeignMessage *foreignMessage = [ForeignMessage message];
  78. [foreignMessage setC:3];
  79. [message setOptionalForeignMessage:foreignMessage];
  80. [message.repeatedStringArray addObject:@"qux"];
  81. [message.repeatedStringArray addObject:@"bar"];
  82. return message;
  83. }
  84. - (TestAllTypes *)mergeResultForDestinationWithoutForeignMessageIvar {
  85. TestAllTypes *message = [TestAllTypes message];
  86. [message setOptionalInt32:1];
  87. [message setOptionalInt64:2];
  88. [message setOptionalString:@"foo"];
  89. ForeignMessage *foreignMessage = [ForeignMessage message];
  90. [message setOptionalForeignMessage:foreignMessage];
  91. [message.repeatedStringArray addObject:@"qux"];
  92. [message.repeatedStringArray addObject:@"bar"];
  93. return message;
  94. }
  95. - (TestAllExtensions *)mergeExtensionsDestination {
  96. TestAllExtensions *message = [TestAllExtensions message];
  97. [message setExtension:[UnittestRoot optionalInt32Extension] value:@5];
  98. [message setExtension:[UnittestRoot optionalStringExtension] value:@"foo"];
  99. ForeignMessage *foreignMessage = [ForeignMessage message];
  100. foreignMessage.c = 4;
  101. [message setExtension:[UnittestRoot optionalForeignMessageExtension]
  102. value:foreignMessage];
  103. TestAllTypes_NestedMessage *nestedMessage =
  104. [TestAllTypes_NestedMessage message];
  105. [message setExtension:[UnittestRoot optionalNestedMessageExtension]
  106. value:nestedMessage];
  107. return message;
  108. }
  109. - (TestAllExtensions *)mergeExtensionsSource {
  110. TestAllExtensions *message = [TestAllExtensions message];
  111. [message setExtension:[UnittestRoot optionalInt64Extension] value:@6];
  112. [message setExtension:[UnittestRoot optionalStringExtension] value:@"bar"];
  113. ForeignMessage *foreignMessage = [ForeignMessage message];
  114. [message setExtension:[UnittestRoot optionalForeignMessageExtension]
  115. value:foreignMessage];
  116. TestAllTypes_NestedMessage *nestedMessage =
  117. [TestAllTypes_NestedMessage message];
  118. nestedMessage.bb = 7;
  119. [message setExtension:[UnittestRoot optionalNestedMessageExtension]
  120. value:nestedMessage];
  121. return message;
  122. }
  123. - (TestAllExtensions *)mergeExtensionsResult {
  124. TestAllExtensions *message = [TestAllExtensions message];
  125. [message setExtension:[UnittestRoot optionalInt32Extension] value:@5];
  126. [message setExtension:[UnittestRoot optionalInt64Extension] value:@6];
  127. [message setExtension:[UnittestRoot optionalStringExtension] value:@"bar"];
  128. ForeignMessage *foreignMessage = [ForeignMessage message];
  129. foreignMessage.c = 4;
  130. [message setExtension:[UnittestRoot optionalForeignMessageExtension]
  131. value:foreignMessage];
  132. TestAllTypes_NestedMessage *nestedMessage =
  133. [TestAllTypes_NestedMessage message];
  134. nestedMessage.bb = 7;
  135. [message setExtension:[UnittestRoot optionalNestedMessageExtension]
  136. value:nestedMessage];
  137. return message;
  138. }
  139. - (void)testMergeFrom {
  140. TestAllTypes *result = [[self.mergeDestination copy] autorelease];
  141. [result mergeFrom:self.mergeSource];
  142. NSData *resultData = [result data];
  143. NSData *mergeResultData = [self.mergeResult data];
  144. XCTAssertEqualObjects(resultData, mergeResultData);
  145. XCTAssertEqualObjects(result, self.mergeResult);
  146. // Test when destination does not have an Ivar (type is an object) but source
  147. // has such Ivar.
  148. // The result must has the Ivar which is same as the one in source.
  149. result = [[self.mergeDestinationWithoutForeignMessageIvar copy] autorelease];
  150. [result mergeFrom:self.mergeSource];
  151. resultData = [result data];
  152. mergeResultData =
  153. [self.mergeResultForDestinationWithoutForeignMessageIvar data];
  154. XCTAssertEqualObjects(resultData, mergeResultData);
  155. XCTAssertEqualObjects(
  156. result, self.mergeResultForDestinationWithoutForeignMessageIvar);
  157. // Test when destination is empty.
  158. // The result must is same as the source.
  159. result = [TestAllTypes message];
  160. [result mergeFrom:self.mergeSource];
  161. resultData = [result data];
  162. mergeResultData = [self.mergeSource data];
  163. XCTAssertEqualObjects(resultData, mergeResultData);
  164. XCTAssertEqualObjects(result, self.mergeSource);
  165. }
  166. - (void)testMergeFromWithExtensions {
  167. TestAllExtensions *result = [self mergeExtensionsDestination];
  168. [result mergeFrom:[self mergeExtensionsSource]];
  169. NSData *resultData = [result data];
  170. NSData *mergeResultData = [[self mergeExtensionsResult] data];
  171. XCTAssertEqualObjects(resultData, mergeResultData);
  172. XCTAssertEqualObjects(result, [self mergeExtensionsResult]);
  173. // Test merging from data.
  174. result = [self mergeExtensionsDestination];
  175. NSData *data = [[self mergeExtensionsSource] data];
  176. XCTAssertNotNil(data);
  177. [result mergeFromData:data
  178. extensionRegistry:[UnittestRoot extensionRegistry]];
  179. resultData = [result data];
  180. XCTAssertEqualObjects(resultData, mergeResultData);
  181. XCTAssertEqualObjects(result, [self mergeExtensionsResult]);
  182. }
  183. - (void)testIsEquals {
  184. TestAllTypes *result = [[self.mergeDestination copy] autorelease];
  185. [result mergeFrom:self.mergeSource];
  186. XCTAssertEqualObjects(result.data, self.mergeResult.data);
  187. XCTAssertEqualObjects(result, self.mergeResult);
  188. TestAllTypes *result2 = [[self.mergeDestination copy] autorelease];
  189. XCTAssertNotEqualObjects(result2.data, self.mergeResult.data);
  190. XCTAssertNotEqualObjects(result2, self.mergeResult);
  191. }
  192. // =================================================================
  193. // Required-field-related tests.
  194. - (TestRequired *)testRequiredInitialized {
  195. TestRequired *message = [TestRequired message];
  196. [message setA:1];
  197. [message setB:2];
  198. [message setC:3];
  199. return message;
  200. }
  201. - (void)testRequired {
  202. TestRequired *message = [TestRequired message];
  203. XCTAssertFalse(message.initialized);
  204. [message setA:1];
  205. XCTAssertFalse(message.initialized);
  206. [message setB:1];
  207. XCTAssertFalse(message.initialized);
  208. [message setC:1];
  209. XCTAssertTrue(message.initialized);
  210. }
  211. - (void)testRequiredForeign {
  212. TestRequiredForeign *message = [TestRequiredForeign message];
  213. XCTAssertTrue(message.initialized);
  214. [message setOptionalMessage:[TestRequired message]];
  215. XCTAssertFalse(message.initialized);
  216. [message setOptionalMessage:self.testRequiredInitialized];
  217. XCTAssertTrue(message.initialized);
  218. [message.repeatedMessageArray addObject:[TestRequired message]];
  219. XCTAssertFalse(message.initialized);
  220. [message.repeatedMessageArray removeAllObjects];
  221. [message.repeatedMessageArray addObject:self.testRequiredInitialized];
  222. XCTAssertTrue(message.initialized);
  223. }
  224. - (void)testRequiredExtension {
  225. TestAllExtensions *message = [TestAllExtensions message];
  226. XCTAssertTrue(message.initialized);
  227. [message setExtension:[TestRequired single] value:[TestRequired message]];
  228. XCTAssertFalse(message.initialized);
  229. [message setExtension:[TestRequired single]
  230. value:self.testRequiredInitialized];
  231. XCTAssertTrue(message.initialized);
  232. [message addExtension:[TestRequired multi] value:[TestRequired message]];
  233. XCTAssertFalse(message.initialized);
  234. [message setExtension:[TestRequired multi]
  235. index:0
  236. value:self.testRequiredInitialized];
  237. XCTAssertTrue(message.initialized);
  238. }
  239. - (void)testDataFromUninitialized {
  240. TestRequired *message = [TestRequired message];
  241. NSData *data = [message data];
  242. // In DEBUG, the data generation will fail, but in non DEBUG, it passes
  243. // because the check isn't done (for speed).
  244. #ifdef DEBUG
  245. XCTAssertNil(data);
  246. #else
  247. XCTAssertNotNil(data);
  248. XCTAssertFalse(message.initialized);
  249. #endif // DEBUG
  250. }
  251. - (void)testInitialized {
  252. // We're mostly testing that no exception is thrown.
  253. TestRequired *message = [TestRequired message];
  254. XCTAssertFalse(message.initialized);
  255. }
  256. - (void)testDataFromNestedUninitialized {
  257. TestRequiredForeign *message = [TestRequiredForeign message];
  258. [message setOptionalMessage:[TestRequired message]];
  259. [message.repeatedMessageArray addObject:[TestRequired message]];
  260. [message.repeatedMessageArray addObject:[TestRequired message]];
  261. NSData *data = [message data];
  262. // In DEBUG, the data generation will fail, but in non DEBUG, it passes
  263. // because the check isn't done (for speed).
  264. #ifdef DEBUG
  265. XCTAssertNil(data);
  266. #else
  267. XCTAssertNotNil(data);
  268. XCTAssertFalse(message.initialized);
  269. #endif // DEBUG
  270. }
  271. - (void)testNestedInitialized {
  272. // We're mostly testing that no exception is thrown.
  273. TestRequiredForeign *message = [TestRequiredForeign message];
  274. [message setOptionalMessage:[TestRequired message]];
  275. [message.repeatedMessageArray addObject:[TestRequired message]];
  276. [message.repeatedMessageArray addObject:[TestRequired message]];
  277. XCTAssertFalse(message.initialized);
  278. }
  279. - (void)testParseUninitialized {
  280. NSError *error = nil;
  281. TestRequired *msg =
  282. [TestRequired parseFromData:GPBEmptyNSData() error:&error];
  283. // In DEBUG, the parse will fail, but in non DEBUG, it passes because
  284. // the check isn't done (for speed).
  285. #ifdef DEBUG
  286. XCTAssertNil(msg);
  287. XCTAssertNotNil(error);
  288. XCTAssertEqualObjects(error.domain, GPBMessageErrorDomain);
  289. XCTAssertEqual(error.code, GPBMessageErrorCodeMissingRequiredField);
  290. #else
  291. XCTAssertNotNil(msg);
  292. XCTAssertNil(error);
  293. XCTAssertFalse(msg.initialized);
  294. #endif // DEBUG
  295. }
  296. - (void)testCoding {
  297. NSData *data =
  298. [NSKeyedArchiver archivedDataWithRootObject:[self mergeResult]];
  299. id unarchivedObject = [NSKeyedUnarchiver unarchiveObjectWithData:data];
  300. XCTAssertEqualObjects(unarchivedObject, [self mergeResult]);
  301. // Intentionally doing a pointer comparison.
  302. XCTAssertNotEqual(unarchivedObject, [self mergeResult]);
  303. }
  304. - (void)testObjectReset {
  305. // Tests a failure where clearing out defaults values caused an over release.
  306. TestAllTypes *message = [TestAllTypes message];
  307. message.hasOptionalNestedMessage = NO;
  308. [message setOptionalNestedMessage:[TestAllTypes_NestedMessage message]];
  309. message.hasOptionalNestedMessage = NO;
  310. [message setOptionalNestedMessage:[TestAllTypes_NestedMessage message]];
  311. [message setOptionalNestedMessage:nil];
  312. message.hasOptionalNestedMessage = NO;
  313. }
  314. - (void)testSettingHasToYes {
  315. TestAllTypes *message = [TestAllTypes message];
  316. XCTAssertThrows([message setHasOptionalNestedMessage:YES]);
  317. }
  318. - (void)testRoot {
  319. XCTAssertNotNil([UnittestRoot extensionRegistry]);
  320. }
  321. - (void)testGPBMessageSize {
  322. // See the note in GPBMessage_PackagePrivate.h about why we want to keep the
  323. // base instance size pointer size aligned.
  324. size_t messageSize = class_getInstanceSize([GPBMessage class]);
  325. XCTAssertEqual((messageSize % sizeof(void *)), (size_t)0,
  326. @"Base size isn't pointer size aligned");
  327. // Since we add storage ourselves (see +allocWithZone: in GPBMessage), confirm
  328. // that the size of some generated classes is still the same as the base for
  329. // that logic to work as desired.
  330. size_t testMessageSize = class_getInstanceSize([TestAllTypes class]);
  331. XCTAssertEqual(testMessageSize, messageSize);
  332. }
  333. - (void)testInit {
  334. TestAllTypes *message = [TestAllTypes message];
  335. [self assertClear:message];
  336. }
  337. - (void)testAccessors {
  338. TestAllTypes *message = [TestAllTypes message];
  339. [self setAllFields:message repeatedCount:kGPBDefaultRepeatCount];
  340. [self assertAllFieldsSet:message repeatedCount:kGPBDefaultRepeatCount];
  341. }
  342. - (void)testKVC_ValueForKey {
  343. TestAllTypes *message = [TestAllTypes message];
  344. [self setAllFields:message repeatedCount:kGPBDefaultRepeatCount];
  345. [self assertAllFieldsKVCMatch:message];
  346. }
  347. - (void)testKVC_SetValue_ForKey {
  348. TestAllTypes *message = [TestAllTypes message];
  349. [self setAllFieldsViaKVC:message repeatedCount:kGPBDefaultRepeatCount];
  350. [self assertAllFieldsKVCMatch:message];
  351. [self assertAllFieldsSet:message repeatedCount:kGPBDefaultRepeatCount];
  352. [self assertAllFieldsKVCMatch:message];
  353. }
  354. - (void)testDescription {
  355. // No real test, just exercise code
  356. TestAllTypes *message = [TestAllTypes message];
  357. [self setAllFields:message repeatedCount:kGPBDefaultRepeatCount];
  358. GPBUnknownFieldSet *unknownFields =
  359. [[[GPBUnknownFieldSet alloc] init] autorelease];
  360. GPBUnknownField *field =
  361. [[[GPBUnknownField alloc] initWithNumber:2] autorelease];
  362. [field addVarint:2];
  363. [unknownFields addField:field];
  364. field = [[[GPBUnknownField alloc] initWithNumber:3] autorelease];
  365. [field addVarint:4];
  366. [unknownFields addField:field];
  367. [message setUnknownFields:unknownFields];
  368. NSString *description = [message description];
  369. XCTAssertGreaterThan([description length], 0U);
  370. GPBMessage *message2 = [TestAllExtensions message];
  371. [message2 setExtension:[UnittestRoot optionalInt32Extension] value:@1];
  372. [message2 addExtension:[UnittestRoot repeatedInt32Extension] value:@2];
  373. description = [message2 description];
  374. XCTAssertGreaterThan([description length], 0U);
  375. }
  376. - (void)testSetter {
  377. // Test to make sure that if we set a value that has a default value
  378. // with the default, that the has is set, and the value gets put into the
  379. // message correctly.
  380. TestAllTypes *message = [TestAllTypes message];
  381. GPBDescriptor *descriptor = [[message class] descriptor];
  382. XCTAssertNotNil(descriptor);
  383. GPBFieldDescriptor *fieldDescriptor =
  384. [descriptor fieldWithName:@"defaultInt32"];
  385. XCTAssertNotNil(fieldDescriptor);
  386. GPBGenericValue defaultValue = [fieldDescriptor defaultValue];
  387. [message setDefaultInt32:defaultValue.valueInt32];
  388. XCTAssertTrue(message.hasDefaultInt32);
  389. XCTAssertEqual(message.defaultInt32, defaultValue.valueInt32);
  390. // Do the same thing with an object type.
  391. message = [TestAllTypes message];
  392. fieldDescriptor = [descriptor fieldWithName:@"defaultString"];
  393. XCTAssertNotNil(fieldDescriptor);
  394. defaultValue = [fieldDescriptor defaultValue];
  395. [message setDefaultString:defaultValue.valueString];
  396. XCTAssertTrue(message.hasDefaultString);
  397. XCTAssertEqualObjects(message.defaultString, defaultValue.valueString);
  398. // Test default string type.
  399. message = [TestAllTypes message];
  400. XCTAssertEqualObjects(message.defaultString, @"hello");
  401. XCTAssertFalse(message.hasDefaultString);
  402. fieldDescriptor = [descriptor fieldWithName:@"defaultString"];
  403. XCTAssertNotNil(fieldDescriptor);
  404. defaultValue = [fieldDescriptor defaultValue];
  405. [message setDefaultString:defaultValue.valueString];
  406. XCTAssertEqualObjects(message.defaultString, @"hello");
  407. XCTAssertTrue(message.hasDefaultString);
  408. [message setDefaultString:nil];
  409. XCTAssertEqualObjects(message.defaultString, @"hello");
  410. XCTAssertFalse(message.hasDefaultString);
  411. message.hasDefaultString = NO;
  412. XCTAssertFalse(message.hasDefaultString);
  413. XCTAssertEqualObjects(message.defaultString, @"hello");
  414. // Test default bytes type.
  415. NSData *defaultBytes = [@"world" dataUsingEncoding:NSUTF8StringEncoding];
  416. XCTAssertEqualObjects(message.defaultBytes, defaultBytes);
  417. XCTAssertFalse(message.hasDefaultString);
  418. fieldDescriptor = [descriptor fieldWithName:@"defaultBytes"];
  419. XCTAssertNotNil(fieldDescriptor);
  420. defaultValue = [fieldDescriptor defaultValue];
  421. [message setDefaultBytes:defaultValue.valueData];
  422. XCTAssertEqualObjects(message.defaultBytes, defaultBytes);
  423. XCTAssertTrue(message.hasDefaultBytes);
  424. [message setDefaultBytes:nil];
  425. XCTAssertEqualObjects(message.defaultBytes, defaultBytes);
  426. XCTAssertFalse(message.hasDefaultBytes);
  427. message.hasDefaultBytes = NO;
  428. XCTAssertFalse(message.hasDefaultBytes);
  429. XCTAssertEqualObjects(message.defaultBytes, defaultBytes);
  430. // Test optional string.
  431. XCTAssertFalse(message.hasOptionalString);
  432. XCTAssertEqualObjects(message.optionalString, @"");
  433. XCTAssertFalse(message.hasOptionalString);
  434. message.optionalString = nil;
  435. XCTAssertFalse(message.hasOptionalString);
  436. XCTAssertEqualObjects(message.optionalString, @"");
  437. NSString *string = @"string";
  438. message.optionalString = string;
  439. XCTAssertEqualObjects(message.optionalString, string);
  440. XCTAssertTrue(message.hasOptionalString);
  441. message.optionalString = nil;
  442. XCTAssertFalse(message.hasOptionalString);
  443. XCTAssertEqualObjects(message.optionalString, @"");
  444. // Test optional data.
  445. XCTAssertFalse(message.hasOptionalBytes);
  446. XCTAssertEqualObjects(message.optionalBytes, GPBEmptyNSData());
  447. XCTAssertFalse(message.hasOptionalBytes);
  448. message.optionalBytes = nil;
  449. XCTAssertFalse(message.hasOptionalBytes);
  450. XCTAssertEqualObjects(message.optionalBytes, GPBEmptyNSData());
  451. NSData *data = [@"bytes" dataUsingEncoding:NSUTF8StringEncoding];
  452. message.optionalBytes = data;
  453. XCTAssertEqualObjects(message.optionalBytes, data);
  454. XCTAssertTrue(message.hasOptionalBytes);
  455. message.optionalBytes = nil;
  456. XCTAssertFalse(message.hasOptionalBytes);
  457. XCTAssertEqualObjects(message.optionalBytes, GPBEmptyNSData());
  458. // Test lazy message setting
  459. XCTAssertFalse(message.hasOptionalLazyMessage);
  460. XCTAssertNotNil(message.optionalLazyMessage);
  461. XCTAssertFalse(message.hasOptionalLazyMessage);
  462. message.hasOptionalLazyMessage = NO;
  463. XCTAssertFalse(message.hasOptionalLazyMessage);
  464. XCTAssertNotNil(message.optionalLazyMessage);
  465. XCTAssertFalse(message.hasOptionalLazyMessage);
  466. message.optionalLazyMessage = nil;
  467. XCTAssertFalse(message.hasOptionalLazyMessage);
  468. // Test nested messages
  469. XCTAssertFalse(message.hasOptionalLazyMessage);
  470. message.optionalLazyMessage.bb = 1;
  471. XCTAssertTrue(message.hasOptionalLazyMessage);
  472. XCTAssertEqual(message.optionalLazyMessage.bb, 1);
  473. XCTAssertNotNil(message.optionalLazyMessage);
  474. message.optionalLazyMessage = nil;
  475. XCTAssertFalse(message.hasOptionalLazyMessage);
  476. XCTAssertEqual(message.optionalLazyMessage.bb, 0);
  477. XCTAssertFalse(message.hasOptionalLazyMessage);
  478. XCTAssertNotNil(message.optionalLazyMessage);
  479. // -testDefaultSubMessages tests the "defaulting" handling of fields
  480. // containing messages.
  481. }
  482. - (void)testRepeatedSetters {
  483. TestAllTypes *message = [TestAllTypes message];
  484. [self setAllFields:message repeatedCount:kGPBDefaultRepeatCount];
  485. [self modifyRepeatedFields:message];
  486. [self assertRepeatedFieldsModified:message
  487. repeatedCount:kGPBDefaultRepeatCount];
  488. }
  489. - (void)testClear {
  490. TestAllTypes *message = [TestAllTypes message];
  491. [self setAllFields:message repeatedCount:kGPBDefaultRepeatCount];
  492. [self clearAllFields:message];
  493. [self assertClear:message];
  494. TestAllTypes *message2 = [TestAllTypes message];
  495. XCTAssertEqualObjects(message, message2);
  496. }
  497. - (void)testClearKVC {
  498. TestAllTypes *message = [TestAllTypes message];
  499. [self setAllFields:message repeatedCount:kGPBDefaultRepeatCount];
  500. [self clearAllFields:message];
  501. [self assertClear:message];
  502. [self assertClearKVC:message];
  503. }
  504. - (void)testClearExtension {
  505. // clearExtension() is not actually used in TestUtil, so try it manually.
  506. GPBMessage *message1 = [TestAllExtensions message];
  507. [message1 setExtension:[UnittestRoot optionalInt32Extension] value:@1];
  508. XCTAssertTrue([message1 hasExtension:[UnittestRoot optionalInt32Extension]]);
  509. [message1 clearExtension:[UnittestRoot optionalInt32Extension]];
  510. XCTAssertFalse([message1 hasExtension:[UnittestRoot optionalInt32Extension]]);
  511. GPBMessage *message2 = [TestAllExtensions message];
  512. [message2 addExtension:[UnittestRoot repeatedInt32Extension] value:@1];
  513. XCTAssertEqual(
  514. [[message2 getExtension:[UnittestRoot repeatedInt32Extension]] count],
  515. (NSUInteger)1);
  516. [message2 clearExtension:[UnittestRoot repeatedInt32Extension]];
  517. XCTAssertEqual(
  518. [[message2 getExtension:[UnittestRoot repeatedInt32Extension]] count],
  519. (NSUInteger)0);
  520. // Clearing an unset extension field shouldn't make the target message
  521. // visible.
  522. GPBMessage *message3 = [TestAllExtensions message];
  523. GPBMessage *extension_msg =
  524. [message3 getExtension:[UnittestObjcRoot recursiveExtension]];
  525. XCTAssertFalse([message3 hasExtension:[UnittestObjcRoot recursiveExtension]]);
  526. [extension_msg clearExtension:[UnittestRoot optionalInt32Extension]];
  527. XCTAssertFalse([message3 hasExtension:[UnittestObjcRoot recursiveExtension]]);
  528. }
  529. - (void)testDefaultingSubMessages {
  530. TestAllTypes *message = [TestAllTypes message];
  531. // Initially they should all not have values.
  532. XCTAssertFalse(message.hasOptionalGroup);
  533. XCTAssertFalse(message.hasOptionalNestedMessage);
  534. XCTAssertFalse(message.hasOptionalForeignMessage);
  535. XCTAssertFalse(message.hasOptionalImportMessage);
  536. XCTAssertFalse(message.hasOptionalPublicImportMessage);
  537. XCTAssertFalse(message.hasOptionalLazyMessage);
  538. // They should auto create something when fetched.
  539. TestAllTypes_OptionalGroup *optionalGroup = [message.optionalGroup retain];
  540. TestAllTypes_NestedMessage *optionalNestedMessage =
  541. [message.optionalNestedMessage retain];
  542. ForeignMessage *optionalForeignMessage =
  543. [message.optionalForeignMessage retain];
  544. ImportMessage *optionalImportMessage = [message.optionalImportMessage retain];
  545. PublicImportMessage *optionalPublicImportMessage =
  546. [message.optionalPublicImportMessage retain];
  547. TestAllTypes_NestedMessage *optionalLazyMessage =
  548. [message.optionalLazyMessage retain];
  549. XCTAssertNotNil(optionalGroup);
  550. XCTAssertNotNil(optionalNestedMessage);
  551. XCTAssertNotNil(optionalForeignMessage);
  552. XCTAssertNotNil(optionalImportMessage);
  553. XCTAssertNotNil(optionalPublicImportMessage);
  554. XCTAssertNotNil(optionalLazyMessage);
  555. // Although they were created, they should not respond to hasValue until that
  556. // submessage is mutated.
  557. XCTAssertFalse(message.hasOptionalGroup);
  558. XCTAssertFalse(message.hasOptionalNestedMessage);
  559. XCTAssertFalse(message.hasOptionalForeignMessage);
  560. XCTAssertFalse(message.hasOptionalImportMessage);
  561. XCTAssertFalse(message.hasOptionalPublicImportMessage);
  562. XCTAssertFalse(message.hasOptionalLazyMessage);
  563. // And they set that value back in to the message since the value created was
  564. // mutable (so a second fetch should give the same object).
  565. XCTAssertEqual(message.optionalGroup, optionalGroup);
  566. XCTAssertEqual(message.optionalNestedMessage, optionalNestedMessage);
  567. XCTAssertEqual(message.optionalForeignMessage, optionalForeignMessage);
  568. XCTAssertEqual(message.optionalImportMessage, optionalImportMessage);
  569. XCTAssertEqual(message.optionalPublicImportMessage,
  570. optionalPublicImportMessage);
  571. XCTAssertEqual(message.optionalLazyMessage, optionalLazyMessage);
  572. // And the default objects for a second message should be distinct (again,
  573. // since they are mutable, each needs their own copy).
  574. TestAllTypes *message2 = [TestAllTypes message];
  575. // Intentionally doing a pointer comparison.
  576. XCTAssertNotEqual(message2.optionalGroup, optionalGroup);
  577. XCTAssertNotEqual(message2.optionalNestedMessage, optionalNestedMessage);
  578. XCTAssertNotEqual(message2.optionalForeignMessage, optionalForeignMessage);
  579. XCTAssertNotEqual(message2.optionalImportMessage, optionalImportMessage);
  580. XCTAssertNotEqual(message2.optionalPublicImportMessage,
  581. optionalPublicImportMessage);
  582. XCTAssertNotEqual(message2.optionalLazyMessage, optionalLazyMessage);
  583. // Setting the values to nil will clear the has flag, and on next access you
  584. // get back new submessages.
  585. message.optionalGroup = nil;
  586. message.optionalNestedMessage = nil;
  587. message.optionalForeignMessage = nil;
  588. message.optionalImportMessage = nil;
  589. message.optionalPublicImportMessage = nil;
  590. message.optionalLazyMessage = nil;
  591. XCTAssertFalse(message.hasOptionalGroup);
  592. XCTAssertFalse(message.hasOptionalNestedMessage);
  593. XCTAssertFalse(message.hasOptionalForeignMessage);
  594. XCTAssertFalse(message.hasOptionalImportMessage);
  595. XCTAssertFalse(message.hasOptionalPublicImportMessage);
  596. XCTAssertFalse(message.hasOptionalLazyMessage);
  597. // Intentionally doing a pointer comparison.
  598. XCTAssertNotEqual(message.optionalGroup, optionalGroup);
  599. XCTAssertNotEqual(message.optionalNestedMessage, optionalNestedMessage);
  600. XCTAssertNotEqual(message.optionalForeignMessage, optionalForeignMessage);
  601. XCTAssertNotEqual(message.optionalImportMessage, optionalImportMessage);
  602. XCTAssertNotEqual(message.optionalPublicImportMessage,
  603. optionalPublicImportMessage);
  604. XCTAssertNotEqual(message.optionalLazyMessage, optionalLazyMessage);
  605. [optionalGroup release];
  606. [optionalNestedMessage release];
  607. [optionalForeignMessage release];
  608. [optionalImportMessage release];
  609. [optionalPublicImportMessage release];
  610. [optionalLazyMessage release];
  611. }
  612. - (void)testMultiplePointersToAutocreatedMessage {
  613. // Multiple objects pointing to the same autocreated message.
  614. TestAllTypes *message = [TestAllTypes message];
  615. TestAllTypes *message2 = [TestAllTypes message];
  616. message2.optionalGroup = message.optionalGroup;
  617. XCTAssertTrue([message2 hasOptionalGroup]);
  618. XCTAssertFalse([message hasOptionalGroup]);
  619. message2.optionalGroup.a = 42;
  620. XCTAssertTrue([message hasOptionalGroup]);
  621. XCTAssertTrue([message2 hasOptionalGroup]);
  622. }
  623. - (void)testCopyWithAutocreatedMessage {
  624. // Mutable copy should not copy autocreated messages.
  625. TestAllTypes *message = [TestAllTypes message];
  626. message.optionalGroup.a = 42;
  627. XCTAssertNotNil(message.optionalNestedMessage);
  628. TestAllTypes *message2 = [[message copy] autorelease];
  629. XCTAssertTrue([message2 hasOptionalGroup]);
  630. XCTAssertFalse([message2 hasOptionalNestedMessage]);
  631. // Intentionally doing a pointer comparison.
  632. XCTAssertNotEqual(message.optionalNestedMessage,
  633. message2.optionalNestedMessage);
  634. }
  635. - (void)testClearAutocreatedSubmessage {
  636. // Call clear on an intermediate submessage should cause it to get recreated
  637. // on the next call.
  638. TestRecursiveMessage *message = [TestRecursiveMessage message];
  639. TestRecursiveMessage *message_inner = [message.a.a.a retain];
  640. XCTAssertNotNil(message_inner);
  641. XCTAssertTrue(GPBWasMessageAutocreatedBy(message_inner, message.a.a));
  642. [message.a.a clear];
  643. XCTAssertFalse(GPBWasMessageAutocreatedBy(message_inner, message.a.a));
  644. // Intentionally doing a pointer comparison.
  645. XCTAssertNotEqual(message.a.a.a, message_inner);
  646. [message_inner release];
  647. }
  648. - (void)testRetainAutocreatedSubmessage {
  649. // Should be able to retain autocreated submessage while the creator is
  650. // dealloced.
  651. TestAllTypes *message = [TestAllTypes message];
  652. ForeignMessage *subMessage;
  653. @autoreleasepool {
  654. TestAllTypes *message2 = [TestAllTypes message];
  655. subMessage = message2.optionalForeignMessage; // Autocreated
  656. message.optionalForeignMessage = subMessage;
  657. XCTAssertTrue(GPBWasMessageAutocreatedBy(message.optionalForeignMessage,
  658. message2));
  659. }
  660. // Should be the same object, and should still be live.
  661. XCTAssertEqual(message.optionalForeignMessage, subMessage);
  662. XCTAssertNotNil([subMessage description]);
  663. }
  664. - (void)testSetNilAutocreatedSubmessage {
  665. TestRecursiveMessage *message = [TestRecursiveMessage message];
  666. TestRecursiveMessage *message_inner = [message.a.a retain];
  667. XCTAssertFalse([message hasA]);
  668. XCTAssertFalse([message.a hasA]);
  669. message.a.a = nil;
  670. // |message.a| has to be made visible, but |message.a.a| was set to nil so
  671. // shouldn't be.
  672. XCTAssertTrue([message hasA]);
  673. XCTAssertFalse([message.a hasA]);
  674. // Setting submessage to nil should cause it to lose its creator.
  675. XCTAssertFalse(GPBWasMessageAutocreatedBy(message_inner, message.a));
  676. // After setting to nil, getting it again should create a new autocreated
  677. // message.
  678. // Intentionally doing a pointer comparison.
  679. XCTAssertNotEqual(message.a.a, message_inner);
  680. [message_inner release];
  681. }
  682. - (void)testSetDoesntHaveAutocreatedSubmessage {
  683. // Clearing submessage (set has == NO) should NOT cause it to lose its
  684. // creator.
  685. TestAllTypes *message = [TestAllTypes message];
  686. TestAllTypes_NestedMessage *nestedMessage = message.optionalNestedMessage;
  687. XCTAssertFalse([message hasOptionalNestedMessage]);
  688. [message setHasOptionalNestedMessage:NO];
  689. XCTAssertFalse([message hasOptionalNestedMessage]);
  690. XCTAssertEqual(message.optionalNestedMessage, nestedMessage);
  691. }
  692. - (void)testSetAutocreatedMessageBecomesVisible {
  693. // Setting a value should cause the submessage to appear to its creator.
  694. // Test this several levels deep.
  695. TestRecursiveMessage *message = [TestRecursiveMessage message];
  696. message.a.a.a.a.i = 42;
  697. XCTAssertTrue([message hasA]);
  698. XCTAssertTrue([message.a hasA]);
  699. XCTAssertTrue([message.a.a hasA]);
  700. XCTAssertTrue([message.a.a.a hasA]);
  701. XCTAssertFalse([message.a.a.a.a hasA]);
  702. XCTAssertEqual(message.a.a.a.a.i, 42);
  703. }
  704. - (void)testClearUnsetFieldOfAutocreatedMessage {
  705. // Clearing an unset field should not cause the submessage to appear to its
  706. // creator.
  707. TestRecursiveMessage *message = [TestRecursiveMessage message];
  708. message.a.a.a.a.hasI = NO;
  709. XCTAssertFalse([message hasA]);
  710. XCTAssertFalse([message.a hasA]);
  711. XCTAssertFalse([message.a.a hasA]);
  712. XCTAssertFalse([message.a.a.a hasA]);
  713. }
  714. - (void)testAutocreatedSubmessageAssignSkip {
  715. TestRecursiveMessage *message = [TestRecursiveMessage message];
  716. TestRecursiveMessage *messageLevel1 = [message.a retain];
  717. TestRecursiveMessage *messageLevel2 = [message.a.a retain];
  718. TestRecursiveMessage *messageLevel3 = [message.a.a.a retain];
  719. TestRecursiveMessage *messageLevel4 = [message.a.a.a.a retain];
  720. XCTAssertTrue(GPBWasMessageAutocreatedBy(messageLevel1, message));
  721. XCTAssertTrue(GPBWasMessageAutocreatedBy(messageLevel2, messageLevel1));
  722. XCTAssertTrue(GPBWasMessageAutocreatedBy(messageLevel3, messageLevel2));
  723. XCTAssertTrue(GPBWasMessageAutocreatedBy(messageLevel4, messageLevel3));
  724. // Test skipping over an autocreated submessage and ensure it gets unset.
  725. message.a = message.a.a;
  726. XCTAssertEqual(message.a, messageLevel2);
  727. XCTAssertTrue([message hasA]);
  728. XCTAssertEqual(message.a.a, messageLevel3);
  729. XCTAssertFalse([message.a hasA]);
  730. XCTAssertEqual(message.a.a.a, messageLevel4);
  731. XCTAssertFalse(GPBWasMessageAutocreatedBy(messageLevel1,
  732. message)); // Because it was orphaned.
  733. XCTAssertTrue(GPBWasMessageAutocreatedBy(messageLevel2, messageLevel1));
  734. XCTAssertTrue(GPBWasMessageAutocreatedBy(messageLevel3, messageLevel2));
  735. [messageLevel1 release];
  736. [messageLevel2 release];
  737. [messageLevel3 release];
  738. [messageLevel4 release];
  739. }
  740. - (void)testAutocreatedSubmessageAssignLoop {
  741. TestRecursiveMessage *message = [TestRecursiveMessage message];
  742. TestRecursiveMessage *messageLevel1 = [message.a retain];
  743. TestRecursiveMessage *messageLevel2 = [message.a.a retain];
  744. TestRecursiveMessage *messageLevel3 = [message.a.a.a retain];
  745. TestRecursiveMessage *messageLevel4 = [message.a.a.a.a retain];
  746. XCTAssertTrue(GPBWasMessageAutocreatedBy(messageLevel1, message));
  747. XCTAssertTrue(GPBWasMessageAutocreatedBy(messageLevel2, messageLevel1));
  748. XCTAssertTrue(GPBWasMessageAutocreatedBy(messageLevel3, messageLevel2));
  749. XCTAssertTrue(GPBWasMessageAutocreatedBy(messageLevel4, messageLevel3));
  750. // Test a property with a loop. You'd never do this but at least ensure the
  751. // autocreated submessages behave sanely.
  752. message.a.a = message.a;
  753. XCTAssertTrue([message hasA]);
  754. XCTAssertEqual(message.a, messageLevel1);
  755. XCTAssertTrue([message.a hasA]);
  756. XCTAssertEqual(message.a.a, messageLevel1);
  757. XCTAssertTrue([message.a.a hasA]);
  758. XCTAssertEqual(message.a.a.a, messageLevel1);
  759. XCTAssertFalse(GPBWasMessageAutocreatedBy(messageLevel1,
  760. message)); // Because it was assigned.
  761. XCTAssertFalse(GPBWasMessageAutocreatedBy(messageLevel2,
  762. messageLevel1)); // Because it was orphaned.
  763. XCTAssertFalse([messageLevel2 hasA]);
  764. // Break the retain loop.
  765. message.a.a = nil;
  766. XCTAssertTrue([message hasA]);
  767. XCTAssertFalse([message.a hasA]);
  768. [messageLevel1 release];
  769. [messageLevel2 release];
  770. [messageLevel3 release];
  771. [messageLevel4 release];
  772. }
  773. - (void)testSetAutocreatedSubmessage {
  774. // Setting autocreated submessage to another value should cause the old one to
  775. // lose its creator.
  776. TestAllTypes *message = [TestAllTypes message];
  777. TestAllTypes_NestedMessage *nestedMessage =
  778. [message.optionalNestedMessage retain];
  779. message.optionalNestedMessage = [TestAllTypes_NestedMessage message];
  780. XCTAssertTrue([message hasOptionalNestedMessage]);
  781. XCTAssertTrue(message.optionalNestedMessage != nestedMessage);
  782. XCTAssertFalse(GPBWasMessageAutocreatedBy(nestedMessage, message));
  783. [nestedMessage release];
  784. }
  785. - (void)testAutocreatedUnknownFields {
  786. // Doing anything with (except reading) unknown fields should cause the
  787. // submessage to become visible.
  788. TestAllTypes *message = [TestAllTypes message];
  789. XCTAssertNotNil(message.optionalNestedMessage);
  790. XCTAssertFalse([message hasOptionalNestedMessage]);
  791. XCTAssertNil(message.optionalNestedMessage.unknownFields);
  792. XCTAssertFalse([message hasOptionalNestedMessage]);
  793. GPBUnknownFieldSet *unknownFields =
  794. [[[GPBUnknownFieldSet alloc] init] autorelease];
  795. message.optionalNestedMessage.unknownFields = unknownFields;
  796. XCTAssertTrue([message hasOptionalNestedMessage]);
  797. message.optionalNestedMessage = nil;
  798. XCTAssertFalse([message hasOptionalNestedMessage]);
  799. [message.optionalNestedMessage setUnknownFields:unknownFields];
  800. XCTAssertTrue([message hasOptionalNestedMessage]);
  801. }
  802. - (void)testSetAutocreatedSubmessageToSelf {
  803. // Setting submessage to itself should cause it to become visible.
  804. TestAllTypes *message = [TestAllTypes message];
  805. XCTAssertNotNil(message.optionalNestedMessage);
  806. XCTAssertFalse([message hasOptionalNestedMessage]);
  807. message.optionalNestedMessage = message.optionalNestedMessage;
  808. XCTAssertTrue([message hasOptionalNestedMessage]);
  809. }
  810. - (void)testAutocreatedSubmessageMemoryLeaks {
  811. // Test for memory leaks with autocreated submessages.
  812. TestRecursiveMessage *message;
  813. TestRecursiveMessage *messageLevel1;
  814. TestRecursiveMessage *messageLevel2;
  815. TestRecursiveMessage *messageLevel3;
  816. TestRecursiveMessage *messageLevel4;
  817. @autoreleasepool {
  818. message = [[TestRecursiveMessage alloc] init];
  819. messageLevel1 = [message.a retain];
  820. messageLevel2 = [message.a.a retain];
  821. messageLevel3 = [message.a.a.a retain];
  822. messageLevel4 = [message.a.a.a.a retain];
  823. message.a.i = 1;
  824. }
  825. XCTAssertEqual(message.retainCount, (NSUInteger)1);
  826. [message release];
  827. XCTAssertEqual(messageLevel1.retainCount, (NSUInteger)1);
  828. [messageLevel1 release];
  829. XCTAssertEqual(messageLevel2.retainCount, (NSUInteger)1);
  830. [messageLevel2 release];
  831. XCTAssertEqual(messageLevel3.retainCount, (NSUInteger)1);
  832. [messageLevel3 release];
  833. XCTAssertEqual(messageLevel4.retainCount, (NSUInteger)1);
  834. [messageLevel4 release];
  835. }
  836. - (void)testDefaultingArrays {
  837. // Basic tests for default creation of arrays in a message.
  838. TestRecursiveMessageWithRepeatedField *message =
  839. [TestRecursiveMessageWithRepeatedField message];
  840. TestRecursiveMessageWithRepeatedField *message2 =
  841. [TestRecursiveMessageWithRepeatedField message];
  842. // Simply accessing the array should not make any fields visible.
  843. XCTAssertNotNil(message.a.a.iArray);
  844. XCTAssertFalse([message hasA]);
  845. XCTAssertFalse([message.a hasA]);
  846. XCTAssertNotNil(message2.a.a.strArray);
  847. XCTAssertFalse([message2 hasA]);
  848. XCTAssertFalse([message2.a hasA]);
  849. // But adding an element to the array should.
  850. [message.a.a.iArray addValue:42];
  851. XCTAssertTrue([message hasA]);
  852. XCTAssertTrue([message.a hasA]);
  853. XCTAssertEqual([message.a.a.iArray count], (NSUInteger)1);
  854. [message2.a.a.strArray addObject:@"foo"];
  855. XCTAssertTrue([message2 hasA]);
  856. XCTAssertTrue([message2.a hasA]);
  857. XCTAssertEqual([message2.a.a.strArray count], (NSUInteger)1);
  858. }
  859. - (void)testAutocreatedArrayShared {
  860. // Multiple objects pointing to the same array.
  861. TestRecursiveMessageWithRepeatedField *message1a =
  862. [TestRecursiveMessageWithRepeatedField message];
  863. TestRecursiveMessageWithRepeatedField *message1b =
  864. [TestRecursiveMessageWithRepeatedField message];
  865. message1a.a.iArray = message1b.a.iArray;
  866. XCTAssertTrue([message1a hasA]);
  867. XCTAssertFalse([message1b hasA]);
  868. [message1a.a.iArray addValue:1];
  869. XCTAssertTrue([message1a hasA]);
  870. XCTAssertTrue([message1b hasA]);
  871. XCTAssertEqual(message1a.a.iArray, message1b.a.iArray);
  872. TestRecursiveMessageWithRepeatedField *message2a =
  873. [TestRecursiveMessageWithRepeatedField message];
  874. TestRecursiveMessageWithRepeatedField *message2b =
  875. [TestRecursiveMessageWithRepeatedField message];
  876. message2a.a.strArray = message2b.a.strArray;
  877. XCTAssertTrue([message2a hasA]);
  878. XCTAssertFalse([message2b hasA]);
  879. [message2a.a.strArray addObject:@"bar"];
  880. XCTAssertTrue([message2a hasA]);
  881. XCTAssertTrue([message2b hasA]);
  882. XCTAssertEqual(message2a.a.strArray, message2b.a.strArray);
  883. }
  884. - (void)testAutocreatedArrayCopy {
  885. // Copy should not copy autocreated arrays.
  886. TestAllTypes *message = [TestAllTypes message];
  887. XCTAssertNotNil(message.repeatedStringArray);
  888. XCTAssertNotNil(message.repeatedInt32Array);
  889. TestAllTypes *message2 = [[message copy] autorelease];
  890. // Pointer conparisions.
  891. XCTAssertNotEqual(message.repeatedStringArray, message2.repeatedStringArray);
  892. XCTAssertNotEqual(message.repeatedInt32Array, message2.repeatedInt32Array);
  893. // Mutable copy should copy empty arrays that were explicitly set (end up
  894. // with different objects that are equal).
  895. TestAllTypes *message3 = [TestAllTypes message];
  896. message3.repeatedInt32Array = [GPBInt32Array arrayWithValue:42];
  897. message3.repeatedStringArray = [NSMutableArray arrayWithObject:@"wee"];
  898. XCTAssertNotNil(message.repeatedInt32Array);
  899. XCTAssertNotNil(message.repeatedStringArray);
  900. TestAllTypes *message4 = [[message3 copy] autorelease];
  901. XCTAssertNotEqual(message3.repeatedInt32Array, message4.repeatedInt32Array);
  902. XCTAssertEqualObjects(message3.repeatedInt32Array,
  903. message4.repeatedInt32Array);
  904. XCTAssertNotEqual(message3.repeatedStringArray, message4.repeatedStringArray);
  905. XCTAssertEqualObjects(message3.repeatedStringArray,
  906. message4.repeatedStringArray);
  907. }
  908. - (void)testAutocreatedArrayRetain {
  909. // Should be able to retain autocreated array while the creator is dealloced.
  910. TestAllTypes *message = [TestAllTypes message];
  911. @autoreleasepool {
  912. TestAllTypes *message2 = [TestAllTypes message];
  913. message.repeatedInt32Array = message2.repeatedInt32Array;
  914. message.repeatedStringArray = message2.repeatedStringArray;
  915. // Pointer conparision
  916. XCTAssertEqual(message.repeatedInt32Array->_autocreator, message2);
  917. XCTAssertTrue([message.repeatedStringArray
  918. isKindOfClass:[GPBAutocreatedArray class]]);
  919. XCTAssertEqual(
  920. ((GPBAutocreatedArray *)message.repeatedStringArray)->_autocreator,
  921. message2);
  922. }
  923. XCTAssertNil(message.repeatedInt32Array->_autocreator);
  924. XCTAssertTrue(
  925. [message.repeatedStringArray isKindOfClass:[GPBAutocreatedArray class]]);
  926. XCTAssertNil(
  927. ((GPBAutocreatedArray *)message.repeatedStringArray)->_autocreator);
  928. }
  929. - (void)testSetNilAutocreatedArray {
  930. // Setting array to nil should cause it to lose its delegate.
  931. TestAllTypes *message = [TestAllTypes message];
  932. GPBInt32Array *repeatedInt32Array = [message.repeatedInt32Array retain];
  933. GPBAutocreatedArray *repeatedStringArray =
  934. (GPBAutocreatedArray *)[message.repeatedStringArray retain];
  935. XCTAssertTrue([repeatedStringArray isKindOfClass:[GPBAutocreatedArray class]]);
  936. XCTAssertEqual(repeatedInt32Array->_autocreator, message);
  937. XCTAssertEqual(repeatedStringArray->_autocreator, message);
  938. message.repeatedInt32Array = nil;
  939. message.repeatedStringArray = nil;
  940. XCTAssertNil(repeatedInt32Array->_autocreator);
  941. XCTAssertNil(repeatedStringArray->_autocreator);
  942. [repeatedInt32Array release];
  943. [repeatedStringArray release];
  944. }
  945. - (void)testSetOverAutocreatedArrayAndSetAgain {
  946. // Ensure when dealing with replacing an array it is handled being either
  947. // an autocreated one or a straight NSArray.
  948. // The real test here is that nothing crashes while doing the work.
  949. TestAllTypes *message = [TestAllTypes message];
  950. [message.repeatedStringArray addObject:@"foo"];
  951. XCTAssertEqual(message.repeatedStringArray_Count, (NSUInteger)1);
  952. message.repeatedStringArray = [NSMutableArray arrayWithObjects:@"bar", @"bar2", nil];
  953. XCTAssertEqual(message.repeatedStringArray_Count, (NSUInteger)2);
  954. message.repeatedStringArray = [NSMutableArray arrayWithObject:@"baz"];
  955. XCTAssertEqual(message.repeatedStringArray_Count, (NSUInteger)1);
  956. }
  957. - (void)testReplaceAutocreatedArray {
  958. // Replacing array should orphan the old one and cause its creator to become
  959. // visible.
  960. {
  961. TestRecursiveMessageWithRepeatedField *message =
  962. [TestRecursiveMessageWithRepeatedField message];
  963. XCTAssertNotNil(message.a);
  964. XCTAssertNotNil(message.a.iArray);
  965. XCTAssertFalse([message hasA]);
  966. GPBInt32Array *iArray = [message.a.iArray retain];
  967. XCTAssertEqual(iArray->_autocreator, message.a); // Pointer comparision
  968. message.a.iArray = [GPBInt32Array arrayWithValue:1];
  969. XCTAssertTrue([message hasA]);
  970. XCTAssertNotEqual(message.a.iArray, iArray); // Pointer comparision
  971. XCTAssertNil(iArray->_autocreator);
  972. [iArray release];
  973. }
  974. {
  975. TestRecursiveMessageWithRepeatedField *message =
  976. [TestRecursiveMessageWithRepeatedField message];
  977. XCTAssertNotNil(message.a);
  978. XCTAssertNotNil(message.a.strArray);
  979. XCTAssertFalse([message hasA]);
  980. GPBAutocreatedArray *strArray =
  981. (GPBAutocreatedArray *)[message.a.strArray retain];
  982. XCTAssertTrue([strArray isKindOfClass:[GPBAutocreatedArray class]]);
  983. XCTAssertEqual(strArray->_autocreator, message.a); // Pointer comparision
  984. message.a.strArray = [NSMutableArray arrayWithObject:@"foo"];
  985. XCTAssertTrue([message hasA]);
  986. XCTAssertNotEqual(message.a.strArray, strArray); // Pointer comparision
  987. XCTAssertNil(strArray->_autocreator);
  988. [strArray release];
  989. }
  990. }
  991. - (void)testSetAutocreatedArrayToSelf {
  992. // Setting array to itself should cause it to become visible.
  993. {
  994. TestRecursiveMessageWithRepeatedField *message =
  995. [TestRecursiveMessageWithRepeatedField message];
  996. XCTAssertNotNil(message.a);
  997. XCTAssertNotNil(message.a.iArray);
  998. XCTAssertFalse([message hasA]);
  999. message.a.iArray = message.a.iArray;
  1000. XCTAssertTrue([message hasA]);
  1001. XCTAssertNil(message.a.iArray->_autocreator);
  1002. }
  1003. {
  1004. TestRecursiveMessageWithRepeatedField *message =
  1005. [TestRecursiveMessageWithRepeatedField message];
  1006. XCTAssertNotNil(message.a);
  1007. XCTAssertNotNil(message.a.strArray);
  1008. XCTAssertFalse([message hasA]);
  1009. message.a.strArray = message.a.strArray;
  1010. XCTAssertTrue([message hasA]);
  1011. XCTAssertTrue([message.a.strArray isKindOfClass:[GPBAutocreatedArray class]]);
  1012. XCTAssertNil(((GPBAutocreatedArray *)message.a.strArray)->_autocreator);
  1013. }
  1014. }
  1015. - (void)testAutocreatedArrayRemoveAllValues {
  1016. // Calling removeAllValues on autocreated array should not cause it to be
  1017. // visible.
  1018. TestRecursiveMessageWithRepeatedField *message =
  1019. [TestRecursiveMessageWithRepeatedField message];
  1020. [message.a.iArray removeAll];
  1021. XCTAssertFalse([message hasA]);
  1022. [message.a.strArray removeAllObjects];
  1023. XCTAssertFalse([message hasA]);
  1024. }
  1025. - (void)testDefaultingMaps {
  1026. // Basic tests for default creation of maps in a message.
  1027. TestRecursiveMessageWithRepeatedField *message =
  1028. [TestRecursiveMessageWithRepeatedField message];
  1029. TestRecursiveMessageWithRepeatedField *message2 =
  1030. [TestRecursiveMessageWithRepeatedField message];
  1031. // Simply accessing the map should not make any fields visible.
  1032. XCTAssertNotNil(message.a.a.iToI);
  1033. XCTAssertFalse([message hasA]);
  1034. XCTAssertFalse([message.a hasA]);
  1035. XCTAssertNotNil(message2.a.a.strToStr);
  1036. XCTAssertFalse([message2 hasA]);
  1037. XCTAssertFalse([message2.a hasA]);
  1038. // But adding an element to the map should.
  1039. [message.a.a.iToI setInt32:100 forKey:200];
  1040. XCTAssertTrue([message hasA]);
  1041. XCTAssertTrue([message.a hasA]);
  1042. XCTAssertEqual([message.a.a.iToI count], (NSUInteger)1);
  1043. [message2.a.a.strToStr setObject:@"foo" forKey:@"bar"];
  1044. XCTAssertTrue([message2 hasA]);
  1045. XCTAssertTrue([message2.a hasA]);
  1046. XCTAssertEqual([message2.a.a.strToStr count], (NSUInteger)1);
  1047. }
  1048. - (void)testAutocreatedMapShared {
  1049. // Multiple objects pointing to the same map.
  1050. TestRecursiveMessageWithRepeatedField *message1a =
  1051. [TestRecursiveMessageWithRepeatedField message];
  1052. TestRecursiveMessageWithRepeatedField *message1b =
  1053. [TestRecursiveMessageWithRepeatedField message];
  1054. message1a.a.iToI = message1b.a.iToI;
  1055. XCTAssertTrue([message1a hasA]);
  1056. XCTAssertFalse([message1b hasA]);
  1057. [message1a.a.iToI setInt32:1 forKey:2];
  1058. XCTAssertTrue([message1a hasA]);
  1059. XCTAssertTrue([message1b hasA]);
  1060. XCTAssertEqual(message1a.a.iToI, message1b.a.iToI);
  1061. TestRecursiveMessageWithRepeatedField *message2a =
  1062. [TestRecursiveMessageWithRepeatedField message];
  1063. TestRecursiveMessageWithRepeatedField *message2b =
  1064. [TestRecursiveMessageWithRepeatedField message];
  1065. message2a.a.strToStr = message2b.a.strToStr;
  1066. XCTAssertTrue([message2a hasA]);
  1067. XCTAssertFalse([message2b hasA]);
  1068. [message2a.a.strToStr setObject:@"bar" forKey:@"foo"];
  1069. XCTAssertTrue([message2a hasA]);
  1070. XCTAssertTrue([message2b hasA]);
  1071. XCTAssertEqual(message2a.a.strToStr, message2b.a.strToStr);
  1072. }
  1073. - (void)testAutocreatedMapCopy {
  1074. // Copy should not copy autocreated maps.
  1075. TestRecursiveMessageWithRepeatedField *message =
  1076. [TestRecursiveMessageWithRepeatedField message];
  1077. XCTAssertNotNil(message.strToStr);
  1078. XCTAssertNotNil(message.iToI);
  1079. TestRecursiveMessageWithRepeatedField *message2 =
  1080. [[message copy] autorelease];
  1081. // Pointer conparisions.
  1082. XCTAssertNotEqual(message.strToStr, message2.strToStr);
  1083. XCTAssertNotEqual(message.iToI, message2.iToI);
  1084. // Mutable copy should copy empty arrays that were explicitly set (end up
  1085. // with different objects that are equal).
  1086. TestRecursiveMessageWithRepeatedField *message3 =
  1087. [TestRecursiveMessageWithRepeatedField message];
  1088. message3.iToI = [[[GPBInt32Int32Dictionary alloc] init] autorelease];
  1089. [message3.iToI setInt32:10 forKey:20];
  1090. message3.strToStr =
  1091. [NSMutableDictionary dictionaryWithObject:@"abc" forKey:@"123"];
  1092. XCTAssertNotNil(message.iToI);
  1093. XCTAssertNotNil(message.iToI);
  1094. TestRecursiveMessageWithRepeatedField *message4 =
  1095. [[message3 copy] autorelease];
  1096. XCTAssertNotEqual(message3.iToI, message4.iToI);
  1097. XCTAssertEqualObjects(message3.iToI, message4.iToI);
  1098. XCTAssertNotEqual(message3.strToStr, message4.strToStr);
  1099. XCTAssertEqualObjects(message3.strToStr, message4.strToStr);
  1100. }
  1101. - (void)testAutocreatedMapRetain {
  1102. // Should be able to retain autocreated map while the creator is dealloced.
  1103. TestRecursiveMessageWithRepeatedField *message =
  1104. [TestRecursiveMessageWithRepeatedField message];
  1105. @autoreleasepool {
  1106. TestRecursiveMessageWithRepeatedField *message2 =
  1107. [TestRecursiveMessageWithRepeatedField message];
  1108. message.iToI = message2.iToI;
  1109. message.strToStr = message2.strToStr;
  1110. // Pointer conparision
  1111. XCTAssertEqual(message.iToI->_autocreator, message2);
  1112. XCTAssertTrue([message.strToStr
  1113. isKindOfClass:[GPBAutocreatedDictionary class]]);
  1114. XCTAssertEqual(
  1115. ((GPBAutocreatedDictionary *)message.strToStr)->_autocreator,
  1116. message2);
  1117. }
  1118. XCTAssertNil(message.iToI->_autocreator);
  1119. XCTAssertTrue(
  1120. [message.strToStr isKindOfClass:[GPBAutocreatedDictionary class]]);
  1121. XCTAssertNil(
  1122. ((GPBAutocreatedDictionary *)message.strToStr)->_autocreator);
  1123. }
  1124. - (void)testSetNilAutocreatedMap {
  1125. // Setting map to nil should cause it to lose its delegate.
  1126. TestRecursiveMessageWithRepeatedField *message =
  1127. [TestRecursiveMessageWithRepeatedField message];
  1128. GPBInt32Int32Dictionary *iToI = [message.iToI retain];
  1129. GPBAutocreatedDictionary *strToStr =
  1130. (GPBAutocreatedDictionary *)[message.strToStr retain];
  1131. XCTAssertTrue([strToStr isKindOfClass:[GPBAutocreatedDictionary class]]);
  1132. XCTAssertEqual(iToI->_autocreator, message);
  1133. XCTAssertEqual(strToStr->_autocreator, message);
  1134. message.iToI = nil;
  1135. message.strToStr = nil;
  1136. XCTAssertNil(iToI->_autocreator);
  1137. XCTAssertNil(strToStr->_autocreator);
  1138. [iToI release];
  1139. [strToStr release];
  1140. }
  1141. - (void)testSetOverAutocreatedMapAndSetAgain {
  1142. // Ensure when dealing with replacing a map it is handled being either
  1143. // an autocreated one or a straight NSDictionary.
  1144. // The real test here is that nothing crashes while doing the work.
  1145. TestRecursiveMessageWithRepeatedField *message =
  1146. [TestRecursiveMessageWithRepeatedField message];
  1147. message.strToStr[@"foo"] = @"bar";
  1148. XCTAssertEqual(message.strToStr_Count, (NSUInteger)1);
  1149. message.strToStr =
  1150. [NSMutableDictionary dictionaryWithObjectsAndKeys:@"bar", @"key1", @"baz", @"key2", nil];
  1151. XCTAssertEqual(message.strToStr_Count, (NSUInteger)2);
  1152. message.strToStr =
  1153. [NSMutableDictionary dictionaryWithObject:@"baz" forKey:@"mumble"];
  1154. XCTAssertEqual(message.strToStr_Count, (NSUInteger)1);
  1155. }
  1156. - (void)testReplaceAutocreatedMap {
  1157. // Replacing map should orphan the old one and cause its creator to become
  1158. // visible.
  1159. {
  1160. TestRecursiveMessageWithRepeatedField *message =
  1161. [TestRecursiveMessageWithRepeatedField message];
  1162. XCTAssertNotNil(message.a);
  1163. XCTAssertNotNil(message.a.iToI);
  1164. XCTAssertFalse([message hasA]);
  1165. GPBInt32Int32Dictionary *iToI = [message.a.iToI retain];
  1166. XCTAssertEqual(iToI->_autocreator, message.a); // Pointer comparision
  1167. message.a.iToI = [[[GPBInt32Int32Dictionary alloc] init] autorelease];
  1168. [message.a.iToI setInt32:6 forKey:7];
  1169. XCTAssertTrue([message hasA]);
  1170. XCTAssertNotEqual(message.a.iToI, iToI); // Pointer comparision
  1171. XCTAssertNil(iToI->_autocreator);
  1172. [iToI release];
  1173. }
  1174. {
  1175. TestRecursiveMessageWithRepeatedField *message =
  1176. [TestRecursiveMessageWithRepeatedField message];
  1177. XCTAssertNotNil(message.a);
  1178. XCTAssertNotNil(message.a.strToStr);
  1179. XCTAssertFalse([message hasA]);
  1180. GPBAutocreatedDictionary *strToStr =
  1181. (GPBAutocreatedDictionary *)[message.a.strToStr retain];
  1182. XCTAssertTrue([strToStr isKindOfClass:[GPBAutocreatedDictionary class]]);
  1183. XCTAssertEqual(strToStr->_autocreator, message.a); // Pointer comparision
  1184. message.a.strToStr =
  1185. [NSMutableDictionary dictionaryWithObject:@"abc" forKey:@"def"];
  1186. XCTAssertTrue([message hasA]);
  1187. XCTAssertNotEqual(message.a.strToStr, strToStr); // Pointer comparision
  1188. XCTAssertNil(strToStr->_autocreator);
  1189. [strToStr release];
  1190. }
  1191. }
  1192. - (void)testSetAutocreatedMapToSelf {
  1193. // Setting map to itself should cause it to become visible.
  1194. {
  1195. TestRecursiveMessageWithRepeatedField *message =
  1196. [TestRecursiveMessageWithRepeatedField message];
  1197. XCTAssertNotNil(message.a);
  1198. XCTAssertNotNil(message.a.iToI);
  1199. XCTAssertFalse([message hasA]);
  1200. message.a.iToI = message.a.iToI;
  1201. XCTAssertTrue([message hasA]);
  1202. XCTAssertNil(message.a.iToI->_autocreator);
  1203. }
  1204. {
  1205. TestRecursiveMessageWithRepeatedField *message =
  1206. [TestRecursiveMessageWithRepeatedField message];
  1207. XCTAssertNotNil(message.a);
  1208. XCTAssertNotNil(message.a.strToStr);
  1209. XCTAssertFalse([message hasA]);
  1210. message.a.strToStr = message.a.strToStr;
  1211. XCTAssertTrue([message hasA]);
  1212. XCTAssertTrue([message.a.strToStr isKindOfClass:[GPBAutocreatedDictionary class]]);
  1213. XCTAssertNil(((GPBAutocreatedDictionary *)message.a.strToStr)->_autocreator);
  1214. }
  1215. }
  1216. - (void)testAutocreatedMapRemoveAllValues {
  1217. // Calling removeAll on autocreated map should not cause it to be visible.
  1218. TestRecursiveMessageWithRepeatedField *message =
  1219. [TestRecursiveMessageWithRepeatedField message];
  1220. [message.a.iToI removeAll];
  1221. XCTAssertFalse([message hasA]);
  1222. [message.a.strToStr removeAllObjects];
  1223. XCTAssertFalse([message hasA]);
  1224. }
  1225. - (void)testExtensionAccessors {
  1226. TestAllExtensions *message = [TestAllExtensions message];
  1227. [self setAllExtensions:message repeatedCount:kGPBDefaultRepeatCount];
  1228. [self assertAllExtensionsSet:message repeatedCount:kGPBDefaultRepeatCount];
  1229. }
  1230. - (void)testExtensionRepeatedSetters {
  1231. TestAllExtensions *message = [TestAllExtensions message];
  1232. [self setAllExtensions:message repeatedCount:kGPBDefaultRepeatCount];
  1233. [self modifyRepeatedExtensions:message];
  1234. [self assertRepeatedExtensionsModified:message
  1235. repeatedCount:kGPBDefaultRepeatCount];
  1236. }
  1237. - (void)testExtensionDefaults {
  1238. [self assertExtensionsClear:[TestAllExtensions message]];
  1239. }
  1240. - (void)testExtensionIsEquals {
  1241. TestAllExtensions *message = [TestAllExtensions message];
  1242. [self setAllExtensions:message repeatedCount:kGPBDefaultRepeatCount];
  1243. [self modifyRepeatedExtensions:message];
  1244. TestAllExtensions *message2 = [TestAllExtensions message];
  1245. [self setAllExtensions:message2 repeatedCount:kGPBDefaultRepeatCount];
  1246. XCTAssertFalse([message isEqual:message2]);
  1247. message2 = [TestAllExtensions message];
  1248. [self setAllExtensions:message2 repeatedCount:kGPBDefaultRepeatCount];
  1249. [self modifyRepeatedExtensions:message2];
  1250. XCTAssertEqualObjects(message, message2);
  1251. }
  1252. - (void)testExtensionsMergeFrom {
  1253. TestAllExtensions *message = [TestAllExtensions message];
  1254. [self setAllExtensions:message repeatedCount:kGPBDefaultRepeatCount];
  1255. [self modifyRepeatedExtensions:message];
  1256. message = [TestAllExtensions message];
  1257. [self setAllExtensions:message repeatedCount:kGPBDefaultRepeatCount];
  1258. TestAllExtensions *message2 = [TestAllExtensions message];
  1259. [self modifyRepeatedExtensions:message2];
  1260. [message2 mergeFrom:message];
  1261. XCTAssertEqualObjects(message, message2);
  1262. }
  1263. - (void)testDefaultingExtensionMessages {
  1264. TestAllExtensions *message = [TestAllExtensions message];
  1265. // Initially they should all not have values.
  1266. XCTAssertFalse([message hasExtension:[UnittestRoot optionalGroupExtension]]);
  1267. XCTAssertFalse([message hasExtension:[UnittestRoot optionalGroupExtension]]);
  1268. XCTAssertFalse(
  1269. [message hasExtension:[UnittestRoot optionalNestedMessageExtension]]);
  1270. XCTAssertFalse(
  1271. [message hasExtension:[UnittestRoot optionalForeignMessageExtension]]);
  1272. XCTAssertFalse(
  1273. [message hasExtension:[UnittestRoot optionalImportMessageExtension]]);
  1274. XCTAssertFalse([message
  1275. hasExtension:[UnittestRoot optionalPublicImportMessageExtension]]);
  1276. XCTAssertFalse(
  1277. [message hasExtension:[UnittestRoot optionalLazyMessageExtension]]);
  1278. // They should auto create something when fetched.
  1279. TestAllTypes_OptionalGroup *optionalGroup =
  1280. [message getExtension:[UnittestRoot optionalGroupExtension]];
  1281. TestAllTypes_NestedMessage *optionalNestedMessage =
  1282. [message getExtension:[UnittestRoot optionalNestedMessageExtension]];
  1283. ForeignMessage *optionalForeignMessage =
  1284. [message getExtension:[UnittestRoot optionalForeignMessageExtension]];
  1285. ImportMessage *optionalImportMessage =
  1286. [message getExtension:[UnittestRoot optionalImportMessageExtension]];
  1287. PublicImportMessage *optionalPublicImportMessage = [message
  1288. getExtension:[UnittestRoot optionalPublicImportMessageExtension]];
  1289. TestAllTypes_NestedMessage *optionalLazyMessage =
  1290. [message getExtension:[UnittestRoot optionalLazyMessageExtension]];
  1291. XCTAssertNotNil(optionalGroup);
  1292. XCTAssertNotNil(optionalNestedMessage);
  1293. XCTAssertNotNil(optionalForeignMessage);
  1294. XCTAssertNotNil(optionalImportMessage);
  1295. XCTAssertNotNil(optionalPublicImportMessage);
  1296. XCTAssertNotNil(optionalLazyMessage);
  1297. // Although it auto-created empty messages, it should not show that it has
  1298. // them.
  1299. XCTAssertFalse([message hasExtension:[UnittestRoot optionalGroupExtension]]);
  1300. XCTAssertFalse([message hasExtension:[UnittestRoot optionalGroupExtension]]);
  1301. XCTAssertFalse([message hasExtension:[UnittestRoot optionalNestedMessageExtension]]);
  1302. XCTAssertFalse([message hasExtension:[UnittestRoot optionalForeignMessageExtension]]);
  1303. XCTAssertFalse([message hasExtension:[UnittestRoot optionalImportMessageExtension]]);
  1304. XCTAssertFalse([message hasExtension:[UnittestRoot optionalPublicImportMessageExtension]]);
  1305. XCTAssertFalse([message hasExtension:[UnittestRoot optionalLazyMessageExtension]]);
  1306. // And they set that value back in to the message since the value created was
  1307. // mutable (so a second fetch should give the same object).
  1308. XCTAssertEqual([message getExtension:[UnittestRoot optionalGroupExtension]],
  1309. optionalGroup);
  1310. XCTAssertEqual(
  1311. [message getExtension:[UnittestRoot optionalNestedMessageExtension]],
  1312. optionalNestedMessage);
  1313. XCTAssertEqual(
  1314. [message getExtension:[UnittestRoot optionalForeignMessageExtension]],
  1315. optionalForeignMessage);
  1316. XCTAssertEqual(
  1317. [message getExtension:[UnittestRoot optionalImportMessageExtension]],
  1318. optionalImportMessage);
  1319. XCTAssertEqual(
  1320. [message getExtension:[UnittestRoot optionalPublicImportMessageExtension]],
  1321. optionalPublicImportMessage);
  1322. XCTAssertEqual(
  1323. [message getExtension:[UnittestRoot optionalLazyMessageExtension]],
  1324. optionalLazyMessage);
  1325. // And the default objects for a second message should be distinct (again,
  1326. // since they are mutable, each needs their own copy).
  1327. TestAllExtensions *message2 = [TestAllExtensions message];
  1328. // Intentionally doing a pointer comparison.
  1329. XCTAssertNotEqual(
  1330. [message2 getExtension:[UnittestRoot optionalGroupExtension]],
  1331. optionalGroup);
  1332. XCTAssertNotEqual(
  1333. [message2 getExtension:[UnittestRoot optionalNestedMessageExtension]],
  1334. optionalNestedMessage);
  1335. XCTAssertNotEqual(
  1336. [message2 getExtension:[UnittestRoot optionalForeignMessageExtension]],
  1337. optionalForeignMessage);
  1338. XCTAssertNotEqual(
  1339. [message2 getExtension:[UnittestRoot optionalImportMessageExtension]],
  1340. optionalImportMessage);
  1341. XCTAssertNotEqual(
  1342. [message2 getExtension:[UnittestRoot optionalPublicImportMessageExtension]],
  1343. optionalPublicImportMessage);
  1344. XCTAssertNotEqual(
  1345. [message2 getExtension:[UnittestRoot optionalLazyMessageExtension]],
  1346. optionalLazyMessage);
  1347. // Clear values, and on next access you get back new submessages.
  1348. [message setExtension:[UnittestRoot optionalGroupExtension] value:nil];
  1349. [message setExtension:[UnittestRoot optionalGroupExtension] value:nil];
  1350. [message setExtension:[UnittestRoot optionalNestedMessageExtension]
  1351. value:nil];
  1352. [message setExtension:[UnittestRoot optionalForeignMessageExtension]
  1353. value:nil];
  1354. [message setExtension:[UnittestRoot optionalImportMessageExtension]
  1355. value:nil];
  1356. [message setExtension:[UnittestRoot optionalPublicImportMessageExtension]
  1357. value:nil];
  1358. [message setExtension:[UnittestRoot optionalLazyMessageExtension] value:nil];
  1359. XCTAssertFalse([message hasExtension:[UnittestRoot optionalGroupExtension]]);
  1360. XCTAssertFalse([message hasExtension:[UnittestRoot optionalGroupExtension]]);
  1361. XCTAssertFalse(
  1362. [message hasExtension:[UnittestRoot optionalNestedMessageExtension]]);
  1363. XCTAssertFalse(
  1364. [message hasExtension:[UnittestRoot optionalForeignMessageExtension]]);
  1365. XCTAssertFalse(
  1366. [message hasExtension:[UnittestRoot optionalImportMessageExtension]]);
  1367. XCTAssertFalse([message
  1368. hasExtension:[UnittestRoot optionalPublicImportMessageExtension]]);
  1369. XCTAssertFalse(
  1370. [message hasExtension:[UnittestRoot optionalLazyMessageExtension]]);
  1371. XCTAssertEqual([message getExtension:[UnittestRoot optionalGroupExtension]],
  1372. optionalGroup);
  1373. XCTAssertEqual(
  1374. [message getExtension:[UnittestRoot optionalNestedMessageExtension]],
  1375. optionalNestedMessage);
  1376. XCTAssertEqual(
  1377. [message getExtension:[UnittestRoot optionalForeignMessageExtension]],
  1378. optionalForeignMessage);
  1379. XCTAssertEqual(
  1380. [message getExtension:[UnittestRoot optionalImportMessageExtension]],
  1381. optionalImportMessage);
  1382. XCTAssertEqual(
  1383. [message
  1384. getExtension:[UnittestRoot optionalPublicImportMessageExtension]],
  1385. optionalPublicImportMessage);
  1386. XCTAssertEqual(
  1387. [message getExtension:[UnittestRoot optionalLazyMessageExtension]],
  1388. optionalLazyMessage);
  1389. }
  1390. - (void)testMultiplePointersToAutocreatedExtension {
  1391. // 2 objects point to the same auto-created extension. One should "has" it.
  1392. // The other should not.
  1393. TestAllExtensions *message = [TestAllExtensions message];
  1394. TestAllExtensions *message2 = [TestAllExtensions message];
  1395. GPBExtensionDescriptor *extension = [UnittestRoot optionalGroupExtension];
  1396. [message setExtension:extension value:[message2 getExtension:extension]];
  1397. XCTAssertEqual([message getExtension:extension],
  1398. [message2 getExtension:extension]);
  1399. XCTAssertFalse([message2 hasExtension:extension]);
  1400. XCTAssertTrue([message hasExtension:extension]);
  1401. TestAllTypes_OptionalGroup *extensionValue =
  1402. [message2 getExtension:extension];
  1403. extensionValue.a = 1;
  1404. XCTAssertTrue([message2 hasExtension:extension]);
  1405. XCTAssertTrue([message hasExtension:extension]);
  1406. }
  1407. - (void)testCopyWithAutocreatedExtension {
  1408. // Mutable copy shouldn't copy autocreated extensions.
  1409. TestAllExtensions *message = [TestAllExtensions message];
  1410. GPBExtensionDescriptor *optionalGroupExtension =
  1411. [UnittestRoot optionalGroupExtension];
  1412. GPBExtensionDescriptor *optionalNestedMessageExtesion =
  1413. [UnittestRoot optionalNestedMessageExtension];
  1414. TestAllTypes_OptionalGroup *optionalGroup =
  1415. [message getExtension:optionalGroupExtension];
  1416. optionalGroup.a = 42;
  1417. XCTAssertNotNil(optionalGroup);
  1418. XCTAssertNotNil([message getExtension:optionalNestedMessageExtesion]);
  1419. XCTAssertTrue([message hasExtension:optionalGroupExtension]);
  1420. XCTAssertFalse([message hasExtension:optionalNestedMessageExtesion]);
  1421. TestAllExtensions *message2 = [[message copy] autorelease];
  1422. // message2 should end up with its own copy of the optional group.
  1423. XCTAssertTrue([message2 hasExtension:optionalGroupExtension]);
  1424. XCTAssertEqualObjects([message getExtension:optionalGroupExtension],
  1425. [message2 getExtension:optionalGroupExtension]);
  1426. // Intentionally doing a pointer comparison.
  1427. XCTAssertNotEqual([message getExtension:optionalGroupExtension],
  1428. [message2 getExtension:optionalGroupExtension]);
  1429. XCTAssertFalse([message2 hasExtension:optionalNestedMessageExtesion]);
  1430. // Intentionally doing a pointer comparison (auto creation should be
  1431. // different)
  1432. XCTAssertNotEqual([message getExtension:optionalNestedMessageExtesion],
  1433. [message2 getExtension:optionalNestedMessageExtesion]);
  1434. }
  1435. - (void)testClearMessageAutocreatedExtension {
  1436. // Call clear should cause it to recreate its autocreated extensions.
  1437. TestAllExtensions *message = [TestAllExtensions message];
  1438. GPBExtensionDescriptor *optionalGroupExtension =
  1439. [UnittestRoot optionalGroupExtension];
  1440. TestAllTypes_OptionalGroup *optionalGroup =
  1441. [[message getExtension:optionalGroupExtension] retain];
  1442. [message clear];
  1443. TestAllTypes_OptionalGroup *optionalGroupNew =
  1444. [message getExtension:optionalGroupExtension];
  1445. // Intentionally doing a pointer comparison.
  1446. XCTAssertNotEqual(optionalGroup, optionalGroupNew);
  1447. [optionalGroup release];
  1448. }
  1449. - (void)testRetainAutocreatedExtension {
  1450. // Should be able to retain autocreated extension while the creator is
  1451. // dealloced.
  1452. TestAllExtensions *message = [TestAllExtensions message];
  1453. GPBExtensionDescriptor *optionalGroupExtension =
  1454. [UnittestRoot optionalGroupExtension];
  1455. @autoreleasepool {
  1456. TestAllExtensions *message2 = [TestAllExtensions message];
  1457. [message setExtension:optionalGroupExtension
  1458. value:[message2 getExtension:optionalGroupExtension]];
  1459. XCTAssertTrue(GPBWasMessageAutocreatedBy(
  1460. [message getExtension:optionalGroupExtension], message2));
  1461. }
  1462. XCTAssertFalse(GPBWasMessageAutocreatedBy(
  1463. [message getExtension:optionalGroupExtension], message));
  1464. }
  1465. - (void)testClearAutocreatedExtension {
  1466. // Clearing autocreated extension should NOT cause it to lose its creator.
  1467. TestAllExtensions *message = [TestAllExtensions message];
  1468. GPBExtensionDescriptor *optionalGroupExtension =
  1469. [UnittestRoot optionalGroupExtension];
  1470. TestAllTypes_OptionalGroup *optionalGroup =
  1471. [[message getExtension:optionalGroupExtension] retain];
  1472. [message clearExtension:optionalGroupExtension];
  1473. TestAllTypes_OptionalGroup *optionalGroupNew =
  1474. [message getExtension:optionalGroupExtension];
  1475. XCTAssertEqual(optionalGroup, optionalGroupNew);
  1476. XCTAssertFalse([message hasExtension:optionalGroupExtension]);
  1477. [optionalGroup release];
  1478. // Clearing autocreated extension should not cause its creator to become
  1479. // visible
  1480. GPBExtensionDescriptor *recursiveExtension =
  1481. [UnittestObjcRoot recursiveExtension];
  1482. TestAllExtensions *message_lvl2 = [message getExtension:recursiveExtension];
  1483. TestAllExtensions *message_lvl3 =
  1484. [message_lvl2 getExtension:recursiveExtension];
  1485. [message_lvl3 clearExtension:recursiveExtension];
  1486. XCTAssertFalse([message hasExtension:recursiveExtension]);
  1487. }
  1488. - (void)testSetAutocreatedExtensionBecomesVisible {
  1489. // Setting an extension should cause the extension to appear to its creator.
  1490. // Test this several levels deep.
  1491. TestAllExtensions *message = [TestAllExtensions message];
  1492. GPBExtensionDescriptor *recursiveExtension =
  1493. [UnittestObjcRoot recursiveExtension];
  1494. TestAllExtensions *message_lvl2 = [message getExtension:recursiveExtension];
  1495. TestAllExtensions *message_lvl3 =
  1496. [message_lvl2 getExtension:recursiveExtension];
  1497. TestAllExtensions *message_lvl4 =
  1498. [message_lvl3 getExtension:recursiveExtension];
  1499. XCTAssertFalse([message hasExtension:recursiveExtension]);
  1500. XCTAssertFalse([message_lvl2 hasExtension:recursiveExtension]);
  1501. XCTAssertFalse([message_lvl3 hasExtension:recursiveExtension]);
  1502. XCTAssertFalse([message_lvl4 hasExtension:recursiveExtension]);
  1503. [message_lvl4 setExtension:[UnittestRoot optionalInt32Extension] value:@(1)];
  1504. XCTAssertTrue([message hasExtension:recursiveExtension]);
  1505. XCTAssertTrue([message_lvl2 hasExtension:recursiveExtension]);
  1506. XCTAssertTrue([message_lvl3 hasExtension:recursiveExtension]);
  1507. XCTAssertFalse([message_lvl4 hasExtension:recursiveExtension]);
  1508. XCTAssertFalse(GPBWasMessageAutocreatedBy(message_lvl4, message_lvl3));
  1509. XCTAssertFalse(GPBWasMessageAutocreatedBy(message_lvl3, message_lvl2));
  1510. XCTAssertFalse(GPBWasMessageAutocreatedBy(message_lvl2, message));
  1511. }
  1512. - (void)testSetAutocreatedExtensionToSelf {
  1513. // Setting extension to itself should cause it to become visible.
  1514. TestAllExtensions *message = [TestAllExtensions message];
  1515. GPBExtensionDescriptor *optionalGroupExtension =
  1516. [UnittestRoot optionalGroupExtension];
  1517. XCTAssertNotNil([message getExtension:optionalGroupExtension]);
  1518. XCTAssertFalse([message hasExtension:optionalGroupExtension]);
  1519. [message setExtension:optionalGroupExtension
  1520. value:[message getExtension:optionalGroupExtension]];
  1521. XCTAssertTrue([message hasExtension:optionalGroupExtension]);
  1522. }
  1523. - (void)testAutocreatedExtensionMemoryLeaks {
  1524. GPBExtensionDescriptor *recursiveExtension =
  1525. [UnittestObjcRoot recursiveExtension];
  1526. // Test for memory leaks with autocreated extensions.
  1527. TestAllExtensions *message;
  1528. TestAllExtensions *message_lvl2;
  1529. TestAllExtensions *message_lvl3;
  1530. TestAllExtensions *message_lvl4;
  1531. @autoreleasepool {
  1532. message = [[TestAllExtensions alloc] init];
  1533. message_lvl2 = [[message getExtension:recursiveExtension] retain];
  1534. message_lvl3 = [[message_lvl2 getExtension:recursiveExtension] retain];
  1535. message_lvl4 = [[message_lvl3 getExtension:recursiveExtension] retain];
  1536. [message_lvl2 setExtension:[UnittestRoot optionalInt32Extension]
  1537. value:@(1)];
  1538. }
  1539. XCTAssertEqual(message.retainCount, (NSUInteger)1);
  1540. @autoreleasepool {
  1541. [message release];
  1542. }
  1543. XCTAssertEqual(message_lvl2.retainCount, (NSUInteger)1);
  1544. @autoreleasepool {
  1545. [message_lvl2 release];
  1546. }
  1547. XCTAssertEqual(message_lvl3.retainCount, (NSUInteger)1);
  1548. @autoreleasepool {
  1549. [message_lvl3 release];
  1550. }
  1551. XCTAssertEqual(message_lvl4.retainCount, (NSUInteger)1);
  1552. [message_lvl4 release];
  1553. }
  1554. - (void)testSetExtensionWithAutocreatedValue {
  1555. GPBExtensionDescriptor *recursiveExtension =
  1556. [UnittestObjcRoot recursiveExtension];
  1557. TestAllExtensions *message;
  1558. @autoreleasepool {
  1559. message = [[TestAllExtensions alloc] init];
  1560. [message getExtension:recursiveExtension];
  1561. }
  1562. // This statements checks that the extension value isn't accidentally
  1563. // dealloced when removing it from the autocreated map.
  1564. [message setExtension:recursiveExtension
  1565. value:[message getExtension:recursiveExtension]];
  1566. XCTAssertTrue([message hasExtension:recursiveExtension]);
  1567. [message release];
  1568. }
  1569. - (void)testRecursion {
  1570. TestRecursiveMessage *message = [TestRecursiveMessage message];
  1571. XCTAssertNotNil(message.a);
  1572. XCTAssertNotNil(message.a.a);
  1573. XCTAssertEqual(message.a.a.i, 0);
  1574. }
  1575. - (void)testGenerateAndParseUnknownMessage {
  1576. GPBUnknownFieldSet *unknowns =
  1577. [[[GPBUnknownFieldSet alloc] init] autorelease];
  1578. [unknowns mergeVarintField:123 value:456];
  1579. GPBMessage *message = [GPBMessage message];
  1580. [message setUnknownFields:unknowns];
  1581. NSData *data = [message data];
  1582. GPBMessage *message2 =
  1583. [GPBMessage parseFromData:data extensionRegistry:nil error:NULL];
  1584. XCTAssertEqualObjects(message, message2);
  1585. }
  1586. - (void)testDelimitedWriteAndParseMultipleMessages {
  1587. GPBUnknownFieldSet *unknowns1 =
  1588. [[[GPBUnknownFieldSet alloc] init] autorelease];
  1589. [unknowns1 mergeVarintField:123 value:456];
  1590. GPBMessage *message1 = [GPBMessage message];
  1591. [message1 setUnknownFields:unknowns1];
  1592. GPBUnknownFieldSet *unknowns2 =
  1593. [[[GPBUnknownFieldSet alloc] init] autorelease];
  1594. [unknowns2 mergeVarintField:789 value:987];
  1595. [unknowns2 mergeVarintField:654 value:321];
  1596. GPBMessage *message2 = [GPBMessage message];
  1597. [message2 setUnknownFields:unknowns2];
  1598. NSMutableData *delimitedData = [NSMutableData data];
  1599. [delimitedData appendData:[message1 delimitedData]];
  1600. [delimitedData appendData:[message2 delimitedData]];
  1601. GPBCodedInputStream *input =
  1602. [GPBCodedInputStream streamWithData:delimitedData];
  1603. GPBMessage *message3 = [GPBMessage parseDelimitedFromCodedInputStream:input
  1604. extensionRegistry:nil
  1605. error:NULL];
  1606. GPBMessage *message4 = [GPBMessage parseDelimitedFromCodedInputStream:input
  1607. extensionRegistry:nil
  1608. error:NULL];
  1609. XCTAssertEqualObjects(message1, message3);
  1610. XCTAssertEqualObjects(message2, message4);
  1611. }
  1612. - (void)testDuplicateEnums {
  1613. XCTAssertEqual(TestEnumWithDupValue_Foo1, TestEnumWithDupValue_Foo2);
  1614. }
  1615. - (void)testWeirdDefaults {
  1616. ObjcWeirdDefaults *message = [ObjcWeirdDefaults message];
  1617. GPBDescriptor *descriptor = [[message class] descriptor];
  1618. GPBFieldDescriptor *fieldDesc = [descriptor fieldWithName:@"foo"];
  1619. XCTAssertNotNil(fieldDesc);
  1620. XCTAssertTrue(fieldDesc.hasDefaultValue);
  1621. XCTAssertFalse(message.hasFoo);
  1622. XCTAssertEqualObjects(message.foo, @"");
  1623. fieldDesc = [descriptor fieldWithName:@"bar"];
  1624. XCTAssertNotNil(fieldDesc);
  1625. XCTAssertTrue(fieldDesc.hasDefaultValue);
  1626. XCTAssertFalse(message.hasBar);
  1627. XCTAssertEqualObjects(message.bar, GPBEmptyNSData());
  1628. }
  1629. - (void)testEnumDescriptorFromExtensionDescriptor {
  1630. GPBExtensionDescriptor *extDescriptor =
  1631. [UnittestRoot optionalForeignEnumExtension];
  1632. XCTAssertEqual(extDescriptor.dataType, GPBDataTypeEnum);
  1633. GPBEnumDescriptor *enumDescriptor = extDescriptor.enumDescriptor;
  1634. GPBEnumDescriptor *expectedDescriptor = ForeignEnum_EnumDescriptor();
  1635. XCTAssertEqualObjects(enumDescriptor, expectedDescriptor);
  1636. }
  1637. - (void)testPropertyNaming {
  1638. // objectivec_helpers.cc has some special handing to get proper all caps
  1639. // for a few cases to meet objc developer expectations.
  1640. //
  1641. // This "test" confirms that the expected names are generated, otherwise the
  1642. // test itself will fail to compile.
  1643. ObjCPropertyNaming *msg = [ObjCPropertyNaming message];
  1644. // On their own, at the end, in the middle.
  1645. msg.URL = @"good";
  1646. msg.thumbnailURL = @"good";
  1647. msg.URLFoo = @"good";
  1648. msg.someURLBlah = @"good";
  1649. msg.HTTP = @"good";
  1650. msg.HTTPS = @"good";
  1651. // No caps since it was "urls".
  1652. [msg.urlsArray addObject:@"good"];
  1653. }
  1654. - (void)testEnumNaming {
  1655. // objectivec_helpers.cc has some interesting cases to deal with in
  1656. // EnumValueName/EnumValueShortName. Confirm that things generated as
  1657. // expected.
  1658. // This block just has to compile to confirm we got the expected types/names.
  1659. // The *_IsValidValue() calls are just there to keep the projects warnings
  1660. // flags happy by providing use of the variables/values.
  1661. Foo aFoo = Foo_SerializedSize;
  1662. Foo_IsValidValue(aFoo);
  1663. aFoo = Foo_Size;
  1664. Foo_IsValidValue(aFoo);
  1665. Category_Enum aCat = Category_Enum_Red;
  1666. Category_Enum_IsValidValue(aCat);
  1667. Time aTime = Time_Base;
  1668. Time_IsValidValue(aTime);
  1669. aTime = Time_SomethingElse;
  1670. Time_IsValidValue(aTime);
  1671. // This block confirms the names in the decriptors is what we wanted.
  1672. GPBEnumDescriptor *descriptor;
  1673. NSString *valueName;
  1674. descriptor = Foo_EnumDescriptor();
  1675. XCTAssertNotNil(descriptor);
  1676. XCTAssertEqualObjects(@"Foo", descriptor.name);
  1677. valueName = [descriptor enumNameForValue:Foo_SerializedSize];
  1678. XCTAssertEqualObjects(@"Foo_SerializedSize", valueName);
  1679. valueName = [descriptor enumNameForValue:Foo_Size];
  1680. XCTAssertEqualObjects(@"Foo_Size", valueName);
  1681. descriptor = Category_Enum_EnumDescriptor();
  1682. XCTAssertNotNil(descriptor);
  1683. XCTAssertEqualObjects(@"Category_Enum", descriptor.name);
  1684. valueName = [descriptor enumNameForValue:Category_Enum_Red];
  1685. XCTAssertEqualObjects(@"Category_Enum_Red", valueName);
  1686. descriptor = Time_EnumDescriptor();
  1687. XCTAssertNotNil(descriptor);
  1688. XCTAssertEqualObjects(@"Time", descriptor.name);
  1689. valueName = [descriptor enumNameForValue:Time_Base];
  1690. XCTAssertEqualObjects(@"Time_Base", valueName);
  1691. valueName = [descriptor enumNameForValue:Time_SomethingElse];
  1692. XCTAssertEqualObjects(@"Time_SomethingElse", valueName);
  1693. }
  1694. - (void)testNegativeEnums {
  1695. EnumTestMsg *msg = [EnumTestMsg message];
  1696. // Defaults
  1697. XCTAssertEqual(msg.foo, EnumTestMsg_MyEnum_Zero);
  1698. XCTAssertEqual(msg.bar, EnumTestMsg_MyEnum_One);
  1699. XCTAssertEqual(msg.baz, EnumTestMsg_MyEnum_NegOne);
  1700. // Bounce to wire and back.
  1701. NSData *data = [msg data];
  1702. XCTAssertNotNil(data);
  1703. EnumTestMsg *msgPrime = [EnumTestMsg parseFromData:data error:NULL];
  1704. XCTAssertEqualObjects(msgPrime, msg);
  1705. XCTAssertEqual(msgPrime.foo, EnumTestMsg_MyEnum_Zero);
  1706. XCTAssertEqual(msgPrime.bar, EnumTestMsg_MyEnum_One);
  1707. XCTAssertEqual(msgPrime.baz, EnumTestMsg_MyEnum_NegOne);
  1708. // Other values
  1709. msg.bar = EnumTestMsg_MyEnum_Two;
  1710. msg.baz = EnumTestMsg_MyEnum_NegTwo;
  1711. XCTAssertEqual(msg.bar, EnumTestMsg_MyEnum_Two);
  1712. XCTAssertEqual(msg.baz, EnumTestMsg_MyEnum_NegTwo);
  1713. // Bounce to wire and back.
  1714. data = [msg data];
  1715. XCTAssertNotNil(data);
  1716. msgPrime = [EnumTestMsg parseFromData:data error:NULL];
  1717. XCTAssertEqualObjects(msgPrime, msg);
  1718. XCTAssertEqual(msgPrime.foo, EnumTestMsg_MyEnum_Zero);
  1719. XCTAssertEqual(msgPrime.bar, EnumTestMsg_MyEnum_Two);
  1720. XCTAssertEqual(msgPrime.baz, EnumTestMsg_MyEnum_NegTwo);
  1721. // Repeated field (shouldn't ever be an issue since developer has to use the
  1722. // right GPBArray methods themselves).
  1723. msg.mumbleArray = [GPBEnumArray
  1724. arrayWithValidationFunction:EnumTestMsg_MyEnum_IsValidValue];
  1725. [msg.mumbleArray addValue:EnumTestMsg_MyEnum_Zero];
  1726. [msg.mumbleArray addValue:EnumTestMsg_MyEnum_One];
  1727. [msg.mumbleArray addValue:EnumTestMsg_MyEnum_Two];
  1728. [msg.mumbleArray addValue:EnumTestMsg_MyEnum_NegOne];
  1729. [msg.mumbleArray addValue:EnumTestMsg_MyEnum_NegTwo];
  1730. XCTAssertEqual([msg.mumbleArray valueAtIndex:0], EnumTestMsg_MyEnum_Zero);
  1731. XCTAssertEqual([msg.mumbleArray valueAtIndex:1], EnumTestMsg_MyEnum_One);
  1732. XCTAssertEqual([msg.mumbleArray valueAtIndex:2], EnumTestMsg_MyEnum_Two);
  1733. XCTAssertEqual([msg.mumbleArray valueAtIndex:3], EnumTestMsg_MyEnum_NegOne);
  1734. XCTAssertEqual([msg.mumbleArray valueAtIndex:4], EnumTestMsg_MyEnum_NegTwo);
  1735. // Bounce to wire and back.
  1736. data = [msg data];
  1737. XCTAssertNotNil(data);
  1738. msgPrime = [EnumTestMsg parseFromData:data error:NULL];
  1739. XCTAssertEqualObjects(msgPrime, msg);
  1740. XCTAssertEqual([msgPrime.mumbleArray valueAtIndex:0],
  1741. EnumTestMsg_MyEnum_Zero);
  1742. XCTAssertEqual([msgPrime.mumbleArray valueAtIndex:1], EnumTestMsg_MyEnum_One);
  1743. XCTAssertEqual([msgPrime.mumbleArray valueAtIndex:2], EnumTestMsg_MyEnum_Two);
  1744. XCTAssertEqual([msgPrime.mumbleArray valueAtIndex:3],
  1745. EnumTestMsg_MyEnum_NegOne);
  1746. XCTAssertEqual([msgPrime.mumbleArray valueAtIndex:4],
  1747. EnumTestMsg_MyEnum_NegTwo);
  1748. }
  1749. - (void)testReservedWordNaming {
  1750. // objectivec_helpers.cc has some special handing to make sure that
  1751. // some "reserved" objc names get renamed in a way so they
  1752. // don't conflict.
  1753. //
  1754. // This "test" confirms that the expected names are generated,
  1755. // otherwise the test itself will fail to compile.
  1756. self_Class *msg = [self_Class message];
  1757. // Some ObjC/C/C++ keywords.
  1758. msg.className_p = msg.hasClassName_p;
  1759. msg.cmd = msg.hasCmd;
  1760. msg.nullable_p = msg.hasNullable_p;
  1761. msg.typeof_p = msg.hasTypeof_p;
  1762. msg.instancetype_p = msg.hasInstancetype_p;
  1763. msg.nil_p = msg.hasNil_p;
  1764. msg.instancetype_p = msg.hasInstancetype_p;
  1765. msg.public_p = msg.hasPublic_p;
  1766. // Some that would override NSObject methods
  1767. msg.camltype = msg.hasCamltype;
  1768. msg.isNsdictionary = msg.hasIsNsdictionary;
  1769. msg.dealloc_p = msg.hasDealloc_p;
  1770. msg.zone_p = msg.hasZone_p;
  1771. msg.accessibilityLabel_p = msg.hasAccessibilityLabel_p;
  1772. // Some that we shouldn't need to handle.
  1773. msg.atomic = msg.hasAtomic;
  1774. msg.nonatomic = msg.hasNonatomic;
  1775. msg.strong = msg.hasStrong;
  1776. msg.nullResettable = msg.hasNullResettable;
  1777. // Some that would override GPBMessage methods
  1778. msg.clear_p = msg.hasClear_p;
  1779. msg.data_p = msg.hasData_p;
  1780. // Some MacTypes
  1781. msg.fixed = msg.hasFixed;
  1782. msg.style = msg.hasStyle;
  1783. // Some C Identifiers
  1784. msg.generic = msg.hasGeneric;
  1785. msg.block = msg.hasBlock;
  1786. }
  1787. - (void)testOneBasedEnumHolder {
  1788. // Test case for https://github.com/protocolbuffers/protobuf/issues/1453
  1789. // Message with no explicit defaults, but a non zero default for an enum.
  1790. MessageWithOneBasedEnum *enumMsg = [MessageWithOneBasedEnum message];
  1791. XCTAssertEqual(enumMsg.enumField, MessageWithOneBasedEnum_OneBasedEnum_One);
  1792. }
  1793. - (void)testBoolOffsetUsage {
  1794. // Bools use storage within has_bits; this test ensures that this is honored
  1795. // in all places where things should crash or fail based on reading out of
  1796. // field storage instead.
  1797. BoolOnlyMessage *msg1 = [BoolOnlyMessage message];
  1798. BoolOnlyMessage *msg2 = [BoolOnlyMessage message];
  1799. msg1.boolField1 = YES;
  1800. msg2.boolField1 = YES;
  1801. msg1.boolField3 = YES;
  1802. msg2.boolField3 = YES;
  1803. msg1.boolField5 = YES;
  1804. msg2.boolField5 = YES;
  1805. msg1.boolField7 = YES;
  1806. msg2.boolField7 = YES;
  1807. msg1.boolField9 = YES;
  1808. msg2.boolField9 = YES;
  1809. msg1.boolField11 = YES;
  1810. msg2.boolField11 = YES;
  1811. msg1.boolField13 = YES;
  1812. msg2.boolField13 = YES;
  1813. msg1.boolField15 = YES;
  1814. msg2.boolField15 = YES;
  1815. msg1.boolField17 = YES;
  1816. msg2.boolField17 = YES;
  1817. msg1.boolField19 = YES;
  1818. msg2.boolField19 = YES;
  1819. msg1.boolField21 = YES;
  1820. msg2.boolField21 = YES;
  1821. msg1.boolField23 = YES;
  1822. msg2.boolField23 = YES;
  1823. msg1.boolField25 = YES;
  1824. msg2.boolField25 = YES;
  1825. msg1.boolField27 = YES;
  1826. msg2.boolField27 = YES;
  1827. msg1.boolField29 = YES;
  1828. msg2.boolField29 = YES;
  1829. msg1.boolField31 = YES;
  1830. msg2.boolField31 = YES;
  1831. msg1.boolField32 = YES;
  1832. msg2.boolField32 = YES;
  1833. XCTAssertTrue(msg1 != msg2); // Different pointers.
  1834. XCTAssertEqual([msg1 hash], [msg2 hash]);
  1835. XCTAssertEqualObjects(msg1, msg2);
  1836. BoolOnlyMessage *msg1Prime = [[msg1 copy] autorelease];
  1837. XCTAssertTrue(msg1Prime != msg1); // Different pointers.
  1838. XCTAssertEqual([msg1 hash], [msg1Prime hash]);
  1839. XCTAssertEqualObjects(msg1, msg1Prime);
  1840. // Field set in one, but not the other means they don't match (even if
  1841. // set to default value).
  1842. msg1Prime.boolField2 = NO;
  1843. XCTAssertNotEqualObjects(msg1Prime, msg1);
  1844. // And when set to different values.
  1845. msg1.boolField2 = YES;
  1846. XCTAssertNotEqualObjects(msg1Prime, msg1);
  1847. // And then they match again.
  1848. msg1.boolField2 = NO;
  1849. XCTAssertEqualObjects(msg1Prime, msg1);
  1850. XCTAssertEqual([msg1 hash], [msg1Prime hash]);
  1851. }
  1852. - (void)testCopyingMapFields {
  1853. TestMessageOfMaps *msg = [TestMessageOfMaps message];
  1854. msg.strToStr[@"foo"] = @"bar";
  1855. [msg.strToInt setInt32:1 forKey:@"mumble"];
  1856. [msg.intToStr setObject:@"wee" forKey:42];
  1857. [msg.intToInt setInt32:123 forKey:321];
  1858. [msg.strToBool setBool:YES forKey:@"one"];
  1859. [msg.boolToStr setObject:@"something" forKey:YES];
  1860. [msg.boolToBool setBool:YES forKey:NO];
  1861. [msg.intToBool setBool:YES forKey:13];
  1862. [msg.boolToInt setInt32:111 forKey:NO];
  1863. TestAllTypes *subMsg1 = [TestAllTypes message];
  1864. subMsg1.optionalInt32 = 1;
  1865. TestAllTypes *subMsg2 = [TestAllTypes message];
  1866. subMsg1.optionalInt32 = 2;
  1867. TestAllTypes *subMsg3 = [TestAllTypes message];
  1868. subMsg1.optionalInt32 = 3;
  1869. msg.strToMsg[@"baz"] = subMsg1;
  1870. [msg.intToMsg setObject:subMsg2 forKey:222];
  1871. [msg.boolToMsg setObject:subMsg3 forKey:YES];
  1872. TestMessageOfMaps *msg2 = [[msg copy] autorelease];
  1873. XCTAssertNotNil(msg2);
  1874. XCTAssertEqualObjects(msg2, msg);
  1875. XCTAssertTrue(msg2 != msg); // ptr compare
  1876. XCTAssertTrue(msg.strToStr != msg2.strToStr); // ptr compare
  1877. XCTAssertTrue(msg.intToStr != msg2.intToStr); // ptr compare
  1878. XCTAssertTrue(msg.intToInt != msg2.intToInt); // ptr compare
  1879. XCTAssertTrue(msg.strToBool != msg2.strToBool); // ptr compare
  1880. XCTAssertTrue(msg.boolToStr != msg2.boolToStr); // ptr compare
  1881. XCTAssertTrue(msg.boolToBool != msg2.boolToBool); // ptr compare
  1882. XCTAssertTrue(msg.intToBool != msg2.intToBool); // ptr compare
  1883. XCTAssertTrue(msg.boolToInt != msg2.boolToInt); // ptr compare
  1884. XCTAssertTrue(msg.strToMsg != msg2.strToMsg); // ptr compare
  1885. XCTAssertTrue(msg.intToMsg != msg2.intToMsg); // ptr compare
  1886. XCTAssertTrue(msg.boolToMsg != msg2.boolToMsg); // ptr compare
  1887. XCTAssertTrue(msg.strToMsg[@"baz"] != msg2.strToMsg[@"baz"]); // ptr compare
  1888. XCTAssertEqualObjects(msg.strToMsg[@"baz"], msg2.strToMsg[@"baz"]);
  1889. XCTAssertTrue([msg.intToMsg objectForKey:222] != [msg2.intToMsg objectForKey:222]); // ptr compare
  1890. XCTAssertEqualObjects([msg.intToMsg objectForKey:222], [msg2.intToMsg objectForKey:222]);
  1891. XCTAssertTrue([msg.boolToMsg objectForKey:YES] != [msg2.boolToMsg objectForKey:YES]); // ptr compare
  1892. XCTAssertEqualObjects([msg.boolToMsg objectForKey:YES], [msg2.boolToMsg objectForKey:YES]);
  1893. }
  1894. - (void)testPrefixedNames {
  1895. // The fact that this compiles is sufficient as a test.
  1896. // The assertions are just there to avoid "not-used" warnings.
  1897. // Verify that enum types and values get the prefix.
  1898. GPBTESTTestObjcProtoPrefixEnum value = GPBTESTTestObjcProtoPrefixEnum_Value;
  1899. XCTAssertNotEqual(value, 0);
  1900. // Verify that roots get the prefix.
  1901. GPBTESTUnittestObjcOptionsRoot *root = nil;
  1902. XCTAssertNil(root);
  1903. // Verify that messages that don't already have the prefix get a prefix.
  1904. GPBTESTTestObjcProtoPrefixMessage *prefixedMessage = nil;
  1905. XCTAssertNil(prefixedMessage);
  1906. // Verify that messages that already have a prefix aren't prefixed twice.
  1907. GPBTESTTestHasAPrefixMessage *alreadyPrefixedMessage = nil;
  1908. XCTAssertNil(alreadyPrefixedMessage);
  1909. // Verify that enums that already have a prefix aren't prefixed twice.
  1910. GPBTESTTestHasAPrefixEnum prefixedValue = GPBTESTTestHasAPrefixEnum_ValueB;
  1911. XCTAssertNotEqual(prefixedValue, 0);
  1912. // Verify that classes named the same as prefixes are prefixed.
  1913. GPBTESTGPBTEST *prefixMessage = nil;
  1914. XCTAssertNil(prefixMessage);
  1915. // Verify that classes that have the prefix followed by a lowercase
  1916. // letter DO get the prefix.
  1917. GPBTESTGPBTESTshouldGetAPrefixMessage *shouldGetAPrefixMessage = nil;
  1918. XCTAssertNil(shouldGetAPrefixMessage);
  1919. }
  1920. @end