Home | History | Annotate | Download | only in contacts
      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_test_util.h"
      6 
      7 #include <algorithm>
      8 #include <vector>
      9 
     10 #include "base/bind.h"
     11 #include "base/logging.h"
     12 #include "base/strings/string_number_conversions.h"
     13 #include "base/strings/string_util.h"
     14 #include "base/time/time.h"
     15 #include "chrome/browser/chromeos/contacts/contact_map.h"
     16 #include "third_party/skia/include/core/SkBitmap.h"
     17 #include "third_party/skia/include/core/SkCanvas.h"
     18 #include "ui/gfx/codec/png_codec.h"
     19 #include "ui/gfx/size.h"
     20 
     21 namespace contacts {
     22 namespace test {
     23 
     24 namespace {
     25 
     26 // Invokes |stringify_callback| on each item in |items| and prepends |prefix|,
     27 // and then sorts the resulting strings and joins them using |join_char|.
     28 template<class T>
     29 std::string StringifyField(
     30     const ::google::protobuf::RepeatedPtrField<T>& items,
     31     base::Callback<std::string(const T&)> stringify_callback,
     32     const std::string& prefix,
     33     char join_char) {
     34   std::vector<std::string> strings;
     35   for (int i = 0; i < items.size(); ++i)
     36     strings.push_back(prefix + stringify_callback.Run(items.Get(i)));
     37   std::sort(strings.begin(), strings.end());
     38   return JoinString(strings, join_char);
     39 }
     40 
     41 std::string EmailAddressToString(const Contact_EmailAddress& email) {
     42   return email.address() + "," +
     43          base::IntToString(email.type().relation()) + "," +
     44          email.type().label() + "," +
     45          base::IntToString(email.primary());
     46 }
     47 
     48 std::string PhoneNumberToString(const Contact_PhoneNumber& phone) {
     49   return phone.number() + "," +
     50          base::IntToString(phone.type().relation()) + "," +
     51          phone.type().label() + "," +
     52          base::IntToString(phone.primary());
     53 }
     54 
     55 std::string PostalAddressToString(const Contact_PostalAddress& postal) {
     56   return postal.address() + "," +
     57          base::IntToString(postal.type().relation()) + "," +
     58          postal.type().label() + "," +
     59          base::IntToString(postal.primary());
     60 }
     61 
     62 std::string InstantMessagingAddressToString(
     63     const Contact_InstantMessagingAddress& im) {
     64   return im.address() + "," +
     65          base::IntToString(im.protocol()) + "," +
     66          base::IntToString(im.type().relation()) + "," +
     67          im.type().label() + "," +
     68          base::IntToString(im.primary());
     69 }
     70 
     71 }  // namespace
     72 
     73 std::string ContactToString(const Contact& contact) {
     74   std::string result =
     75       contact.contact_id() + "," +
     76       base::Int64ToString(contact.update_time()) + "," +
     77       base::IntToString(contact.deleted()) + "," +
     78       contact.full_name() + "," +
     79       contact.given_name() + "," +
     80       contact.additional_name() + "," +
     81       contact.family_name() + "," +
     82       contact.name_prefix() + "," +
     83       contact.name_suffix();
     84 
     85   SkBitmap bitmap;
     86   if (contact.has_raw_untrusted_photo()) {
     87     // Testing code just uses PNG for now.  If that changes, use ImageDecoder
     88     // here instead.
     89     CHECK(gfx::PNGCodec::Decode(reinterpret_cast<const unsigned char*>(
     90                                     contact.raw_untrusted_photo().data()),
     91                                 contact.raw_untrusted_photo().size(),
     92                                 &bitmap));
     93   }
     94   result += "," + base::IntToString(bitmap.width()) + "x" +
     95             base::IntToString(bitmap.height());
     96 
     97   result += " " + StringifyField(contact.email_addresses(),
     98                                  base::Bind(EmailAddressToString),
     99                                  "email=", ' ');
    100   result += " " + StringifyField(contact.phone_numbers(),
    101                                  base::Bind(PhoneNumberToString),
    102                                  "phone=", ' ');
    103   result += " " + StringifyField(contact.postal_addresses(),
    104                                  base::Bind(PostalAddressToString),
    105                                  "postal=", ' ');
    106   result += " " + StringifyField(contact.instant_messaging_addresses(),
    107                                  base::Bind(InstantMessagingAddressToString),
    108                                  "im=", ' ');
    109 
    110   return result;
    111 }
    112 
    113 std::string ContactsToString(const ContactPointers& contacts) {
    114   std::vector<std::string> contact_strings;
    115   for (size_t i = 0; i < contacts.size(); ++i)
    116     contact_strings.push_back(ContactToString(*contacts[i]));
    117   std::sort(contact_strings.begin(), contact_strings.end());
    118   return JoinString(contact_strings, '\n');
    119 }
    120 
    121 std::string ContactsToString(const ScopedVector<Contact>& contacts) {
    122   ContactPointers pointers;
    123   for (size_t i = 0; i < contacts.size(); ++i)
    124     pointers.push_back(contacts[i]);
    125   return ContactsToString(pointers);
    126 }
    127 
    128 std::string VarContactsToString(int num_contacts, ...) {
    129   ContactPointers contacts;
    130   va_list list;
    131   va_start(list, num_contacts);
    132   for (int i = 0; i < num_contacts; ++i)
    133     contacts.push_back(va_arg(list, const Contact*));
    134   va_end(list);
    135   return ContactsToString(contacts);
    136 }
    137 
    138 std::string ContactMapToString(const ContactMap& contact_map) {
    139   ContactPointers contacts;
    140   for (ContactMap::const_iterator it = contact_map.begin();
    141        it != contact_map.end(); ++it) {
    142     contacts.push_back(it->second);
    143   }
    144   return ContactsToString(contacts);
    145 }
    146 
    147 void CopyContacts(const ContactPointers& source,
    148                   ScopedVector<Contact>* dest) {
    149   DCHECK(dest);
    150   dest->clear();
    151   for (size_t i = 0; i < source.size(); ++i)
    152     dest->push_back(new Contact(*source[i]));
    153 }
    154 
    155 void CopyContacts(const ScopedVector<Contact>& source,
    156                   ScopedVector<Contact>* dest) {
    157   ContactPointers pointers;
    158   for (size_t i = 0; i < source.size(); ++i)
    159     pointers.push_back(source[i]);
    160   CopyContacts(pointers, dest);
    161 }
    162 
    163 void InitContact(const std::string& contact_id,
    164                  const std::string& name_suffix,
    165                  bool deleted,
    166                  Contact* contact) {
    167   DCHECK(contact);
    168   contact->Clear();
    169   contact->set_contact_id(contact_id);
    170   contact->set_update_time(base::Time::Now().ToInternalValue());
    171   contact->set_deleted(deleted);
    172   contact->set_full_name("full_name_" + name_suffix);
    173   contact->set_given_name("given_name_" + name_suffix);
    174   contact->set_additional_name("additional_name_" + name_suffix);
    175   contact->set_family_name("family_name_" + name_suffix);
    176   contact->set_name_prefix("name_prefix_" + name_suffix);
    177   contact->set_name_suffix("name_suffix_" + name_suffix);
    178 }
    179 
    180 void AddEmailAddress(const std::string& address,
    181                      Contact_AddressType_Relation relation,
    182                      const std::string& label,
    183                      bool primary,
    184                      Contact* contact) {
    185   DCHECK(contact);
    186   Contact::EmailAddress* email = contact->add_email_addresses();
    187   email->set_address(address);
    188   email->mutable_type()->set_relation(relation);
    189   email->mutable_type()->set_label(label);
    190   email->set_primary(primary);
    191 }
    192 
    193 void AddPhoneNumber(const std::string& number,
    194                     Contact_AddressType_Relation relation,
    195                     const std::string& label,
    196                     bool primary,
    197                     Contact* contact) {
    198   DCHECK(contact);
    199   Contact::PhoneNumber* phone = contact->add_phone_numbers();
    200   phone->set_number(number);
    201   phone->mutable_type()->set_relation(relation);
    202   phone->mutable_type()->set_label(label);
    203   phone->set_primary(primary);
    204 }
    205 
    206 void AddPostalAddress(const std::string& address,
    207                       Contact_AddressType_Relation relation,
    208                       const std::string& label,
    209                       bool primary,
    210                       Contact* contact) {
    211   DCHECK(contact);
    212   Contact::PostalAddress* postal = contact->add_postal_addresses();
    213   postal->set_address(address);
    214   postal->mutable_type()->set_relation(relation);
    215   postal->mutable_type()->set_label(label);
    216   postal->set_primary(primary);
    217 }
    218 
    219 void AddInstantMessagingAddress(
    220     const std::string& address,
    221     Contact_InstantMessagingAddress_Protocol protocol,
    222     Contact_AddressType_Relation relation,
    223     const std::string& label,
    224     bool primary,
    225     Contact* contact) {
    226   DCHECK(contact);
    227   Contact::InstantMessagingAddress* im =
    228       contact->add_instant_messaging_addresses();
    229   im->set_address(address);
    230   im->set_protocol(protocol);
    231   im->mutable_type()->set_relation(relation);
    232   im->mutable_type()->set_label(label);
    233   im->set_primary(primary);
    234 }
    235 
    236 void SetPhoto(const gfx::Size& size, Contact* contact) {
    237   DCHECK(contact);
    238   if (size.IsEmpty()) {
    239     contact->clear_raw_untrusted_photo();
    240     return;
    241   }
    242 
    243   SkBitmap bitmap;
    244   bitmap.setConfig(SkBitmap::kARGB_8888_Config, size.width(), size.height());
    245   bitmap.allocPixels();
    246   SkCanvas canvas(bitmap);
    247   canvas.clear(SK_ColorBLACK);
    248 
    249   std::vector<unsigned char> png_photo;
    250   CHECK(gfx::PNGCodec::EncodeBGRASkBitmap(bitmap, false, &png_photo));
    251   contact->set_raw_untrusted_photo(&png_photo[0], png_photo.size());
    252 }
    253 
    254 }  // namespace test
    255 }  // namespace contacts
    256