sync: add more integration tests for wifi_sync

chrome/browser/sync/test/integration/wifi_credentials_helper.cc

 // Copyright 2014 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.
 
 #include "chrome/browser/sync/test/integration/wifi_credentials_helper.h"
 
 #include "base/logging.h"
+#include "base/memory/scoped_ptr.h"
 #include "base/strings/string_number_conversions.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/sync/test/integration/sync_datatype_helper.h"
 #include "chrome/browser/sync/test/integration/sync_test.h"
 #include "components/wifi_sync/wifi_credential.h"
-#include "components/wifi_sync/wifi_security_class.h"
+#include "components/wifi_sync/wifi_credential_syncable_service.h"
+#include "components/wifi_sync/wifi_credential_syncable_service_factory.h"
 
 #if defined(OS_CHROMEOS)
 #include "chrome/browser/sync/test/integration/wifi_credentials_helper_chromeos.h"
 #endif
 
 using wifi_sync::WifiCredential;
+using wifi_sync::WifiCredentialSyncableService;
+using wifi_sync::WifiCredentialSyncableServiceFactory;
+using wifi_sync::WifiSecurityClass;
 using sync_datatype_helper::test;
 
 using WifiCredentialSet = wifi_sync::WifiCredential::CredentialSet;
 
 namespace wifi_credentials_helper {
 
 namespace {

[stevenjb, 2015/01/28 21:29:43]

WS

[mukesh agrawal, 2015/01/28 22:55:11]

Done.

+bool SetupClientForProfile(Profile* profile) {
+#if defined(OS_CHROMEOS)
+  return SetupClientForProfileChromeOs(profile);
+#else
+  NOTREACHED();
+  return false;
+#endif
+}
+
+WifiCredentialSyncableService* GetServiceForBrowserContext(
+    content::BrowserContext* context) {
+  return WifiCredentialSyncableServiceFactory::GetForBrowserContext(
+      context);
+}
+
+WifiCredentialSyncableService* GetServiceForProfile(int profile_index) {
+  return GetServiceForBrowserContext(test()->GetProfile(profile_index));
+}
+
+void AddWifiCredentialToProfile(
+    Profile* profile, const WifiCredential& credential) {
+#if defined(OS_CHROMEOS)
+  AddWifiCredentialToProfileChromeOs(profile, credential);
+#else
+  NOTREACHED();
+#endif
+}
 
 WifiCredentialSet GetWifiCredentialsForProfile(const Profile* profile) {
 #if defined(OS_CHROMEOS)
   return GetWifiCredentialsForProfileChromeOs(profile);
 #else
-  NOTIMPLEMENTED();
+  NOTREACHED();
   return WifiCredential::MakeSet();
 #endif
 }
 
 bool CredentialsMatch(const WifiCredentialSet& a_credentials,
                       const WifiCredentialSet& b_credentials) {
   if (a_credentials.size() != b_credentials.size()) {
     LOG(ERROR) << "CredentialSets a and b do not match in size: "
                << a_credentials.size()
                << " vs " << b_credentials.size() << " respectively.";
     return false;
   }
 
-  for (const auto &credential : a_credentials) {
+  for (const WifiCredential& credential : a_credentials) {
     if (b_credentials.find(credential) == b_credentials.end()) {
       LOG(ERROR)
           << "Network from a not found in b. "
           << "SSID (hex): "
           << base::HexEncode(credential.ssid().data(),
                              credential.ssid().size()).c_str()
           << " SecurityClass: " << credential.security_class()
           << " Passphrase: " << credential.passphrase();
       return false;
     }
   }
 
   return true;
 }
 
 }  // namespace
 
+void SetUp() {
+#if defined(OS_CHROMEOS)
+  return SetUpChromeOs();

[stevenjb, 2015/01/28 21:29:43]

This is a little confusing since it is not defined here. I would put the
chromeos specific code in the namespace wifi_credentials_helper::chromeos and
explicitly call wifi_credentials_helper::chromeos::SetuUpChromeOS (and I would
go ahead and use the redundant naming to be extra clear).

[mukesh agrawal, 2015/01/28 22:55:11]

Done.

+#else
+  NOTREACHED();
+#endif
+}
+
+bool SetupClients() {
+  if (!SetupClientForProfile(test()->verifier()))
+    return false;
+
+  for (int i = 0; i < test()->num_clients(); ++i) {
+    if (!SetupClientForProfile(test()->GetProfile(i)))
+      return false;
+  }
+
+  return true;
+}
+
 bool VerifierIsEmpty() {
   return GetWifiCredentialsForProfile(test()->verifier()).empty();
 }
 
 bool ProfileMatchesVerifier(int profile_index) {
   WifiCredentialSet verifier_credentials =
       GetWifiCredentialsForProfile(test()->verifier());
   WifiCredentialSet other_credentials =
       GetWifiCredentialsForProfile(test()->GetProfile(profile_index));
   return CredentialsMatch(verifier_credentials, other_credentials);
@@ skipping 11 matching lines @@
     WifiCredentialSet other_profile_credentials =
         GetWifiCredentialsForProfile(test()->GetProfile(i));
     if (!CredentialsMatch(profile0_credentials, other_profile_credentials)) {
       LOG(ERROR) << "Profile " << i << " " << "does not match with profile 0.";
       return false;
     }
   }
   return true;
 }
 
+WifiCredential MakeWifiCredential(const std::string& ssid,
+                                  WifiSecurityClass security_class,
+                                  const std::string& passphrase) {
+  scoped_ptr<WifiCredential> credential =
+      WifiCredential::Create(
+          WifiCredential::MakeSsidBytesForTest(ssid),
+          security_class,
+          passphrase);
+  CHECK(credential);
+  return *credential;

[stevenjb, 2015/01/28 21:29:43]

The implict copy here is confusing. I would have MakeWifiCredential return a
scoped_ptr and do the copy in the caller.

[mukesh agrawal, 2015/01/28 22:55:11]

Done.

+}
+
+void AddWifiCredential(int profile_index,
+                       const std::string& sync_id,
+                       const WifiCredential& credential) {
+  AddWifiCredentialToProfile(test()->GetProfile(profile_index), credential);
+  if (test()->use_verifier())
+    AddWifiCredentialToProfile(test()->verifier(), credential);
+
+  // TODO(quiche): Remove this, once we have plumbing to route
+  // NetworkConfigurationObserver events to
+  // WifiCredentialSyncableService instances.
+  GetServiceForProfile(profile_index)
+      ->AddToSyncedNetworks(sync_id, credential);
+}
+
 }  // namespace wifi_credentials_helper