LinkTests.m 7.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238
  1. ////////////////////////////////////////////////////////////////////////////
  2. //
  3. // Copyright 2014 Realm Inc.
  4. //
  5. // Licensed under the Apache License, Version 2.0 (the "License");
  6. // you may not use this file except in compliance with the License.
  7. // You may obtain a copy of the License at
  8. //
  9. // http://www.apache.org/licenses/LICENSE-2.0
  10. //
  11. // Unless required by applicable law or agreed to in writing, software
  12. // distributed under the License is distributed on an "AS IS" BASIS,
  13. // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14. // See the License for the specific language governing permissions and
  15. // limitations under the License.
  16. //
  17. ////////////////////////////////////////////////////////////////////////////
  18. #import "RLMTestCase.h"
  19. #import "RLMRealm_Dynamic.h"
  20. RLM_ARRAY_TYPE(CircularArrayObject)
  21. @interface CircularArrayObject : RLMObject
  22. @property RLM_GENERIC_ARRAY(CircularArrayObject) *array;
  23. @end
  24. @implementation CircularArrayObject
  25. @end
  26. @interface LinkTests : RLMTestCase
  27. @end
  28. @implementation LinkTests
  29. - (void)makeDogWithName:(NSString *)name owner:(NSString *)ownerName {
  30. RLMRealm *realm = [self realmWithTestPath];
  31. OwnerObject *owner = [[OwnerObject alloc] init];
  32. owner.name = ownerName;
  33. owner.dog = [[DogObject alloc] init];
  34. owner.dog.dogName = name;
  35. owner.dog.age = 0;
  36. [realm beginWriteTransaction];
  37. [realm addObject:owner];
  38. [realm commitWriteTransaction];
  39. }
  40. - (void)testBasicLink
  41. {
  42. [self makeDogWithName:@"Harvie" owner:@"Tim"];
  43. RLMRealm *realm = [self realmWithTestPath];
  44. RLMResults *owners = [OwnerObject objectsInRealm:realm withPredicate:nil];
  45. RLMResults *dogs = [DogObject objectsInRealm:realm withPredicate:nil];
  46. XCTAssertEqual(owners.count, 1U);
  47. XCTAssertEqual(dogs.count, 1U);
  48. XCTAssertEqualObjects([owners[0] name], @"Tim", @"Tim is named Tim");
  49. XCTAssertEqualObjects([dogs[0] dogName], @"Harvie", @"Harvie is named Harvie");
  50. OwnerObject *tim = owners[0];
  51. XCTAssertEqualObjects(tim.dog.dogName, @"Harvie", @"Tim's dog should be Harvie");
  52. }
  53. -(void)testBasicLinkWithNil
  54. {
  55. RLMRealm *realm = [self realmWithTestPath];
  56. OwnerObject *owner = [[OwnerObject alloc] init];
  57. owner.name = @"Tim";
  58. owner.dog = nil;
  59. [realm beginWriteTransaction];
  60. [realm addObject:owner];
  61. [realm commitWriteTransaction];
  62. RLMResults *owners = [OwnerObject objectsInRealm:realm withPredicate:nil];
  63. RLMResults *dogs = [DogObject objectsInRealm:realm withPredicate:nil];
  64. XCTAssertEqual(owners.count, 1U);
  65. XCTAssertEqual(dogs.count, 0U);
  66. XCTAssertEqualObjects([owners[0] name], @"Tim", @"Tim is named Tim");
  67. OwnerObject *tim = owners[0];
  68. XCTAssertEqualObjects(tim.dog, nil, @"Tim does not have a dog");
  69. }
  70. - (void)testMultipleOwnerLink
  71. {
  72. [self makeDogWithName:@"Harvie" owner:@"Tim"];
  73. RLMRealm *realm = [self realmWithTestPath];
  74. XCTAssertEqual([OwnerObject allObjectsInRealm:realm].count, 1U);
  75. XCTAssertEqual([DogObject allObjectsInRealm:realm].count, 1U);
  76. [realm beginWriteTransaction];
  77. OwnerObject *fiel = [OwnerObject createInRealm:realm withValue:@[@"Fiel", [NSNull null]]];
  78. fiel.dog = [DogObject allObjectsInRealm:realm].firstObject;
  79. [realm commitWriteTransaction];
  80. XCTAssertEqual([OwnerObject objectsInRealm:realm withPredicate:nil].count, 2U);
  81. XCTAssertEqual([DogObject objectsInRealm:realm withPredicate:nil].count, 1U);
  82. }
  83. - (void)testLinkRemoval
  84. {
  85. [self makeDogWithName:@"Harvie" owner:@"Tim"];
  86. RLMRealm *realm = [self realmWithTestPath];
  87. XCTAssertEqual([OwnerObject objectsInRealm:realm withPredicate:nil].count, 1U);
  88. XCTAssertEqual([DogObject objectsInRealm:realm withPredicate:nil].count, 1U);
  89. DogObject *dog = [DogObject allObjectsInRealm:realm].firstObject;
  90. OwnerObject *owner = [OwnerObject allObjectsInRealm:realm].firstObject;
  91. [realm beginWriteTransaction];
  92. [realm deleteObject:dog];
  93. [realm commitWriteTransaction];
  94. XCTAssertNil(owner.dog, @"Dog should be nullified when deleted");
  95. XCTAssertThrows(dog.dogName, @"Dog object should be invalid after being deleted from the realm");
  96. // refresh owner and check
  97. owner = [OwnerObject allObjectsInRealm:realm].firstObject;
  98. XCTAssertNotNil(owner, @"Should have 1 owner");
  99. XCTAssertNil(owner.dog, @"Dog should be nullified when deleted");
  100. XCTAssertEqual([DogObject objectsInRealm:realm withPredicate:nil].count, 0U);
  101. }
  102. - (void)testInvalidLinks
  103. {
  104. RLMRealm *realm = [self realmWithTestPath];
  105. LinkToAllTypesObject *linkObject = [[LinkToAllTypesObject alloc] init];
  106. linkObject.allTypesCol = [[AllTypesObject alloc] init];
  107. [realm beginWriteTransaction];
  108. XCTAssertThrows([realm addObject:linkObject], @"dateCol not set on linked object");
  109. StringObject *to = [StringObject createInRealm:realm withValue:@[@"testObject"]];
  110. NSArray *args = @[@"Tim", to];
  111. XCTAssertThrows([OwnerObject createInRealm:realm withValue:args], @"Inserting wrong object type should throw");
  112. [realm commitWriteTransaction];
  113. }
  114. - (void)testLinkTooManyRelationships
  115. {
  116. RLMRealm *realm = [self realmWithTestPath];
  117. OwnerObject *owner = [[OwnerObject alloc] init];
  118. owner.name = @"Tim";
  119. owner.dog = [[DogObject alloc] init];
  120. owner.dog.dogName = @"Harvie";
  121. [realm beginWriteTransaction];
  122. [realm addObject:owner];
  123. [realm commitWriteTransaction];
  124. XCTAssertThrows([OwnerObject objectsInRealm:realm where:@"dog.dogName.first = 'Fifo'"], @"3 levels of relationship");
  125. }
  126. - (void)testBidirectionalRelationship {
  127. RLMRealm *realm = [RLMRealm defaultRealm];
  128. CircleObject *obj0 = [[CircleObject alloc] initWithValue:@[@"a", NSNull.null]];
  129. CircleObject *obj1 = [[CircleObject alloc] initWithValue:@[@"b", obj0]];
  130. obj0.next = obj1;
  131. [realm beginWriteTransaction];
  132. [realm addObject:obj0];
  133. [realm addObject:obj1];
  134. [realm commitWriteTransaction];
  135. RLMResults *results = [CircleObject allObjects];
  136. XCTAssertEqualObjects(@"a", [results[0] data]);
  137. XCTAssertEqualObjects(@"b", [results[1] data]);
  138. }
  139. - (void)testAddingCircularReferenceDoesNotLeakSourceObjects {
  140. CircleObject __weak *weakObj0, __weak *weakObj1;
  141. @autoreleasepool {
  142. CircleObject *obj0 = [[CircleObject alloc] initWithValue:@[@"a", NSNull.null]];
  143. CircleObject *obj1 = [[CircleObject alloc] initWithValue:@[@"b", obj0]];
  144. obj0.next = obj1;
  145. weakObj0 = obj0;
  146. weakObj1 = obj1;
  147. RLMRealm *realm = [RLMRealm defaultRealm];
  148. [realm beginWriteTransaction];
  149. [realm addObject:obj0];
  150. obj0.next = nil;
  151. obj1.next = nil;
  152. [realm commitWriteTransaction];
  153. }
  154. XCTAssertNil(weakObj0);
  155. XCTAssertNil(weakObj1);
  156. }
  157. - (void)testAddingCircularReferenceInArrayDoesNotLeakSourceObjects {
  158. CircularArrayObject __weak *weakObj0, __weak *weakObj1;
  159. @autoreleasepool {
  160. CircularArrayObject *obj0 = [[CircularArrayObject alloc] initWithValue:@[@[]]];
  161. CircularArrayObject *obj1 = [[CircularArrayObject alloc] initWithValue:@[@[obj0]]];
  162. [obj0.array addObject:obj1];
  163. weakObj0 = obj0;
  164. weakObj1 = obj1;
  165. RLMRealm *realm = [RLMRealm defaultRealm];
  166. [realm beginWriteTransaction];
  167. [realm addObject:obj0];
  168. [realm commitWriteTransaction];
  169. }
  170. XCTAssertNil(weakObj0);
  171. XCTAssertNil(weakObj1);
  172. }
  173. - (void)testCircularLinks {
  174. RLMRealm *realm = [self realmWithTestPath];
  175. CircleObject *obj = [[CircleObject alloc] init];
  176. obj.data = @"a";
  177. obj.next = obj;
  178. [realm beginWriteTransaction];
  179. [realm addObject:obj];
  180. obj.next.data = @"b";
  181. [realm commitWriteTransaction];
  182. XCTAssertEqual(1U, [CircleObject allObjectsInRealm:realm].count);
  183. CircleObject *obj1 = [CircleObject allObjectsInRealm:realm].firstObject;
  184. XCTAssertEqualObjects(obj1.data, @"b", @"data should be 'b'");
  185. XCTAssertEqualObjects(obj1.data, obj.next.data, @"objects should be equal");
  186. }
  187. @end