X-Git-Url: https://git.saurik.com/apple/security.git/blobdiff_plain/b04fe171f0375ecd5d8a24747ca1dff85720a0ca..6b200bc335dc93c5516ccb52f14bd896d8c7fad7:/OSX/sec/securityd/Regressions/secd-52-account-changed.c?ds=sidebyside

diff --git a/OSX/sec/securityd/Regressions/secd-52-account-changed.c b/OSX/sec/securityd/Regressions/secd-52-account-changed.c
index ea8db431..791558f9 100644
--- a/OSX/sec/securityd/Regressions/secd-52-account-changed.c
+++ b/OSX/sec/securityd/Regressions/secd-52-account-changed.c
@@ -109,9 +109,20 @@ static void tests(void)
 
     accounts_agree_internal("bob&alice pair", bob_account, alice_account, false);
     accounts_agree_internal("bob&carol pair", bob_account, carol_account, false);
+    CFDictionaryRef alice_devstate = SOSTestSaveStaticAccountState(alice_account);
+    CFDictionaryRef bob_devstate = SOSTestSaveStaticAccountState(bob_account);
+    CFDictionaryRef carol_devstate = SOSTestSaveStaticAccountState(carol_account);
 
     /* ==================== Three Accounts in circle =============================================*/
     InjectChangeToMulti(changes, CFSTR("^AccountChanged"), CFSTR("none"), alice_account, bob_account, carol_account, NULL);
+    
+    SOSTestRestoreAccountState(alice_account, alice_devstate);
+    SOSTestRestoreAccountState(bob_account, bob_devstate);
+    SOSTestRestoreAccountState(carol_account, carol_devstate);
+    
+    CFReleaseNull(alice_devstate);
+    CFReleaseNull(bob_devstate);
+    CFReleaseNull(carol_devstate);
 
     SOSAccountEnsureFactoryCirclesTest(alice_account, CFSTR("Alice"));
     SOSAccountEnsureFactoryCirclesTest(bob_account, CFSTR("Bob"));
@@ -184,21 +195,12 @@ static void tests(void)
     accounts_agree_internal("bob&carol pair", bob_account, carol_account, false);
     
     CFReleaseSafe(cfpassword2);
-    
-    CFReleaseNull(alice_account);
-    CFReleaseNull(bob_account);
-    CFReleaseNull(carol_account);
-
-    SOSUnregisterAllTransportMessages();
-    SOSUnregisterAllTransportCircles();
-    SOSUnregisterAllTransportKeyParameters();
-    CFArrayRemoveAllValues(key_transports);
-    CFArrayRemoveAllValues(circle_transports);
-    CFArrayRemoveAllValues(message_transports);
     CFReleaseNull(changes);
     CFReleaseNull(alice_account);
     CFReleaseNull(bob_account);
     CFReleaseNull(carol_account);
+    SOSTestCleanup();
+
 }