15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2012 The Chromium Authors. All rights reserved.
25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be
35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file.
45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/memory/scoped_vector.h"
6868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)#include "base/strings/stringprintf.h"
75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/browser/sessions/session_service.h"
85d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)#include "chrome/browser/sync/test/integration/profile_sync_service_harness.h"
95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/browser/sync/test/integration/sessions_helper.h"
105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/browser/sync/test/integration/sync_test.h"
115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
12a02191e04bc25c4935f804f2c080ae28663d096dBen Murdochusing sessions_helper::AwaitCheckForeignSessionsAgainst;
135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)using sessions_helper::CheckInitialState;
145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)using sessions_helper::OpenTabAndGetLocalWindows;
155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)using sessions_helper::ScopedWindowMap;
165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)using sessions_helper::SessionWindowMap;
175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)using sessions_helper::SyncedSessionVector;
185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class MultipleClientSessionsSyncTest : public SyncTest {
205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public:
215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  MultipleClientSessionsSyncTest() : SyncTest(MULTIPLE_CLIENT) {}
225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  virtual ~MultipleClientSessionsSyncTest() {}
235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) private:
255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DISALLOW_COPY_AND_ASSIGN(MultipleClientSessionsSyncTest);
265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)};
275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Timeout on Windows, see http://crbug.com/99819
295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#if defined(OS_WIN)
305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#define MAYBE_AllChanged DISABLED_AllChanged
315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#define MAYBE_EncryptedAndChanged DISABLED_EncryptedAndChanged
325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#else
335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#define MAYBE_AllChanged AllChanged
345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#define MAYBE_EncryptedAndChanged EncryptedAndChanged
355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif
365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)IN_PROC_BROWSER_TEST_F(MultipleClientSessionsSyncTest, MAYBE_AllChanged) {
385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  ASSERT_TRUE(SetupSync()) << "SetupSync() failed.";
395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  std::vector<ScopedWindowMap> client_windows(num_clients());
405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  for (int i = 0; i < num_clients(); ++i) {
425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    ASSERT_TRUE(CheckInitialState(i));
435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // Open tabs on all clients and retain window information.
465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  for (int i = 0; i < num_clients(); ++i) {
475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    SessionWindowMap windows;
485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    ASSERT_TRUE(OpenTabAndGetLocalWindows(
492a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)        i, GURL(base::StringPrintf("http://127.0.0.1/bubba%i", i)), &windows));
505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    client_windows[i].Reset(&windows);
515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // Get foreign session data from all clients and check it against all
545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // client_windows.
555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  for (int i = 0; i < num_clients(); ++i) {
56a02191e04bc25c4935f804f2c080ae28663d096dBen Murdoch    ASSERT_TRUE(AwaitCheckForeignSessionsAgainst(i, client_windows));
575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)IN_PROC_BROWSER_TEST_F(MultipleClientSessionsSyncTest,
615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                       MAYBE_EncryptedAndChanged) {
625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  ASSERT_TRUE(SetupSync()) << "SetupSync() failed.";
635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  std::vector<ScopedWindowMap> client_windows(num_clients());
645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  for (int i = 0; i < num_clients(); ++i) {
665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    ASSERT_TRUE(CheckInitialState(i));
675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // Enable encryption on client 0, should propagate to all other clients.
705d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)  ASSERT_TRUE(EnableEncryption(0));
715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // Wait for sync.
735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // TODO(zea): Fix sync completion detection so we don't need this. For now,
745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // the profile sync service harness detects completion before all encryption
755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // changes are propagated.
765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  ASSERT_TRUE(GetClient(0)->AwaitGroupSyncCycleCompletion(clients()));
775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // Open tabs on all clients and retain window information.
795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  for (int i = 0; i < num_clients(); ++i) {
805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    SessionWindowMap windows;
815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    ASSERT_TRUE(OpenTabAndGetLocalWindows(
822a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)        i, GURL(base::StringPrintf("http://127.0.0.1/bubba%i", i)), &windows));
835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    client_windows[i].Reset(&windows);
845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // Get foreign session data from all clients and check it against all
875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // client_windows.
885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  for (int i = 0; i < num_clients(); ++i) {
89a02191e04bc25c4935f804f2c080ae28663d096dBen Murdoch    ASSERT_TRUE(AwaitCheckForeignSessionsAgainst(i, client_windows));
905d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)    ASSERT_TRUE(IsEncryptionComplete(i));
915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
93