Merge pull request #104 from xr1337/master

Unload when a new post processor is added.
appsquickly · Nov 19, 2013 · 56fc367 · 56fc367
2 parents 59c8472 + 7b485cc
commit 56fc367
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 0 deletions.
diff --git a/Source/Factory/TyphoonComponentFactory.m b/Source/Factory/TyphoonComponentFactory.m
@@ -186,6 +186,9 @@ - (void)attachPostProcessor:(id <TyphoonComponentFactoryPostProcessor>)postProce
 {
     LogTrace(@"Attaching post processor: %@", postProcessor);
     [_postProcessors addObject:postProcessor];
+    if ([self isLoaded]) {
+        [self unload];
+    }
 }
 
 - (void)injectProperties:(id)instance

diff --git a/Tests/Factory/Config/Patcher/TyphoonPatcherTests.m b/Tests/Factory/Config/Patcher/TyphoonPatcherTests.m
@@ -52,4 +52,30 @@ - (void)test_allows_patching_out_a_component_with_a_mock
 
 }
 
+- (void)test_allows_patching_out_a_loaded_component_with_a_mock
+{
+    MiddleAgesAssembly* assembly = [MiddleAgesAssembly assembly];
+    TyphoonComponentFactory* factory = [TyphoonBlockComponentFactory factoryWithAssembly:assembly];
+    [factory componentForKey:@"knight"];
+    //at this point, patching a loaded factory will not pass.
+
+    TyphoonPatcher* patcher = [[TyphoonPatcher alloc] init];
+    [patcher patchDefinition:[assembly knight] withObject:^id
+     {
+         Knight* mockKnight = mock([Knight class]);
+         [given([mockKnight favoriteDamsels]) willReturn:@[
+                                                           @"Mary",
+                                                           @"Janezzz"
+                                                           ]];
+
+         return mockKnight;
+     }];
+
+    [factory attachPostProcessor:patcher];
+
+    Knight* knight = [factory componentForKey:@"knight"];
+    assertThatBool([knight favoriteDamsels].count > 0, is(equalToBool(YES)));
+    LogDebug(@"Damsels: %@", [knight favoriteDamsels]);
+
+}
 @end