Implement Add and Update for PasswordStoreMac....

chrome/browser/keychain_mock_mac.h

 // Copyright (c) 2009 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #ifndef CHROME_BROWSER_KEYCHAIN_MOCK_MAC_H_
 #define CHROME_BROWSER_KEYCHAIN_MOCK_MAC_H_
 
+#include <set>
 #include <vector>
 
 #include "chrome/browser/keychain_mac.h"
 
 // Mock Keychain wrapper for testing code that interacts with the OS Keychain.
 // The basic idea of this mock is that it has a static array of data, and
 // SecKeychainItemRef values are just indexes into that array (offset by 1 to
 // prevent problems with clients that null-check refs), cast to pointers.
 //
 // Note that "const" is pretty much meaningless for this class; the const-ness
@@ skipping 37 matching lines @@
                                        const void *passwordData,
                                        SecKeychainItemRef *itemRef) const;
   virtual void Free(CFTypeRef ref) const;
 
   // Return the counts of objects returned by Create/Copy functions but never
   // Free'd as they should have been.
   int UnfreedSearchCount() const;
   int UnfreedKeychainItemCount() const;
   int UnfreedAttributeDataCount() const;
 
+  // Returns true if all items added with AddInternetPassword have a creator
+  // code set.
+  bool CreatorCodesSetForAddedItems() const;
+
   struct KeychainTestData {
     const SecAuthenticationType auth_type;
     const char* server;
     const SecProtocolType protocol;
     const char* path;
     const UInt32 port;
     const char* security_domain;
     const char* creation_date;
     const char* username;
     const char* password;
@@ skipping 10 matching lines @@
   // docs don't indicate whether it is or not, so clients should not assume
   // that it will be.
   void SetTestDataString(int item, UInt32 tag, const char* value);
   // Sets the data of the corresponding attribute of the item-th test item to
   // |value|. Assumes that the space has alread been allocated, and the length
   // set.
   void SetTestDataPort(int item, UInt32 value);
   void SetTestDataProtocol(int item, SecProtocolType value);
   void SetTestDataAuthType(int item, SecAuthenticationType value);
   void SetTestDataNegativeItem(int item, Boolean value);
+  void SetTestDataCreator(int item, OSType value);
   // Sets the password data and length for the item-th test item.
   void SetTestDataPasswordBytes(int item, const void* data, size_t length);
   // Sets the password for the item-th test item. As with SetTestDataString,
   // the data will not be null-terminated.
   void SetTestDataPasswordString(int item, const char* value);
 
-  // Returns the index of |tag| in |attribute_list|, or -1 if it's not found.
-  static int IndexForTag(const SecKeychainAttributeList& attribute_list,
-                         UInt32 tag);
+  // Returns the address of the attribute in attribute_list with tag |tag|, or
+  // NULL if the attribute is not found.
+  static SecKeychainAttribute* AttributeWithTag(
+      const SecKeychainAttributeList& attribute_list, UInt32 tag);
 
   static const int kDummySearchRef = 1000;
 
   typedef struct  {
     void* data;
     UInt32 length;
   } KeychainPasswordData;
 
   SecKeychainAttributeList* keychain_attr_list_;
   KeychainPasswordData* keychain_data_;
   unsigned int item_capacity_;
   mutable unsigned int item_count_;
 
   // Tracks the items that should be returned in subsequent calls to
   // SearchCopyNext, based on the last call to SearchCreateFromAttributes.
   // We can't handle multiple active searches, since we don't track the search
   // ref we return, but we don't need to for our mocking.
   mutable std::vector<unsigned int> remaining_search_results_;
 
   // Track copies and releases to make sure they balance. Really these should
   // be maps to track per item, but this should be good enough to catch
   // real mistakes.
   mutable int search_copy_count_;
   mutable int keychain_item_copy_count_;
   mutable int attribute_data_copy_count_;
+
+  // Tracks which items (by index) were added with AddInternetPassword.
+  mutable std::set<unsigned int> added_via_api_;
 };
 
 #endif  // CHROME_BROWSER_KEYCHAIN_MOCK_MAC_H_