chrome/browser/chromeos/contacts/contact_map_unittest.cc

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "chrome/browser/chromeos/contacts/contact_map.h"
   6
   7 #include "base/memory/scoped_ptr.h"
   8 #include "base/memory/scoped_vector.h"
   9 #include "base/time.h"
  10 #include "chrome/browser/chromeos/contacts/contact.pb.h"
  11 #include "chrome/browser/chromeos/contacts/contact_test_util.h"
  12 #include "testing/gtest/include/gtest/gtest.h"
  13
  14 namespace contacts {
  15 namespace test {
  16
  17 TEST(ContactMapTest, Merge) {
  18   ContactMap map;
  19   EXPECT_TRUE(map.empty());
  20   EXPECT_EQ(0U, map.size());
  21
  22   // Create a contact.
  23   const std::string kContactId1 = "contact_id_1";
  24   scoped_ptr<Contact> contact1(new Contact);
  25   InitContact(kContactId1, "1", false, contact1.get());
  26
  27   // Merge it into the map and check that it's stored.
  28   scoped_ptr<ScopedVector<Contact> > contacts_to_merge(
  29       new ScopedVector<Contact>);
  30   contacts_to_merge->push_back(new Contact(*contact1));
  31   map.Merge(contacts_to_merge.Pass(), ContactMap::KEEP_DELETED_CONTACTS);
  32   EXPECT_FALSE(map.empty());
  33   EXPECT_EQ(1U, map.size());
  34   ASSERT_TRUE(map.Find(kContactId1));
  35   EXPECT_EQ(ContactMapToString(map), VarContactsToString(1, contact1.get()));
  36
  37   // Create a second, deleted contact.
  38   const std::string kContactId2 = "contact_id_2";
  39   scoped_ptr<Contact> contact2(new Contact);
  40   InitContact(kContactId2, "2", true, contact2.get());
  41
  42   // Merge it into the map.  Since we request keeping deleted contacts, the
  43   // contact should be saved.
  44   contacts_to_merge.reset(new ScopedVector<Contact>);
  45   contacts_to_merge->push_back(new Contact(*contact2));
  46   map.Merge(contacts_to_merge.Pass(), ContactMap::KEEP_DELETED_CONTACTS);
  47   EXPECT_EQ(2U, map.size());
  48   ASSERT_TRUE(map.Find(kContactId2));
  49   EXPECT_EQ(ContactMapToString(map),
  50             VarContactsToString(2, contact1.get(), contact2.get()));
  51
  52   // Update the first contact's update time and merge it into the map.
  53   contact1->set_update_time(contact1->update_time() + 20);
  54   contacts_to_merge.reset(new ScopedVector<Contact>);
  55   contacts_to_merge->push_back(new Contact(*contact1));
  56   map.Merge(contacts_to_merge.Pass(), ContactMap::KEEP_DELETED_CONTACTS);
  57   EXPECT_EQ(ContactMapToString(map),
  58             VarContactsToString(2, contact1.get(), contact2.get()));
  59
  60   // Create another deleted contact.
  61   const std::string kContactId3 = "contact_id_3";
  62   scoped_ptr<Contact> contact3(new Contact);
  63   InitContact(kContactId3, "3", true, contact3.get());
  64
  65   // Merge it into the map with DROP_DELETED_CONTACTS.  The contact shouldn't be
  66   // saved.
  67   contacts_to_merge.reset(new ScopedVector<Contact>);
  68   contacts_to_merge->push_back(new Contact(*contact3));
  69   map.Merge(contacts_to_merge.Pass(), ContactMap::DROP_DELETED_CONTACTS);
  70   EXPECT_EQ(ContactMapToString(map),
  71             VarContactsToString(2, contact1.get(), contact2.get()));
  72
  73   // Mark the first contact as being deleted and merge it with
  74   // DROP_DELETED_CONTACTS.  The previous version of the contact should also be
  75   // removed.
  76   contact1->set_deleted(true);
  77   contacts_to_merge.reset(new ScopedVector<Contact>);
  78   contacts_to_merge->push_back(new Contact(*contact1));
  79   map.Merge(contacts_to_merge.Pass(), ContactMap::DROP_DELETED_CONTACTS);
  80   EXPECT_EQ(ContactMapToString(map), VarContactsToString(1, contact2.get()));
  81
  82   map.Clear();
  83   EXPECT_TRUE(map.empty());
  84   EXPECT_EQ(0U, map.size());
  85 }
  86
  87 TEST(ContactMapTest, Erase) {
  88   ContactMap map;
  89   const std::string kContactId = "contact_id";
  90   scoped_ptr<Contact> contact(new Contact);
  91   InitContact(kContactId, "1", false, contact.get());
  92
  93   scoped_ptr<ScopedVector<Contact> > contacts_to_merge(
  94       new ScopedVector<Contact>);
  95   contacts_to_merge->push_back(new Contact(*contact));
  96   map.Merge(contacts_to_merge.Pass(), ContactMap::KEEP_DELETED_CONTACTS);
  97   EXPECT_TRUE(map.Find(kContactId));
  98
  99   map.Erase(kContactId);
 100   EXPECT_FALSE(map.Find(kContactId));
 101   EXPECT_TRUE(map.empty());
 102 }
 103
 104 }  // namespace test
 105 }  // namespace contacts