1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
|
// Copyright (c) 2011 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/dictionary_helper.h"
#include <algorithm>
#include "chrome/browser/profiles/profile.h"
#include "chrome/browser/spellchecker/spellcheck_factory.h"
#include "chrome/browser/spellchecker/spellcheck_service.h"
#include "chrome/browser/sync/test/integration/dictionary_load_observer.h"
#include "chrome/browser/sync/test/integration/profile_sync_service_harness.h"
#include "chrome/browser/sync/test/integration/sync_datatype_helper.h"
#include "chrome/browser/sync/test/integration/sync_test.h"
#include "chrome/common/chrome_switches.h"
#include "chrome/common/spellcheck_common.h"
#include "content/public/test/test_utils.h"
class DictionarySyncIntegrationTestHelper {
public:
// Same as SpellcheckCustomDictionary::AddWord/RemoveWord, except does not
// write to disk.
static bool ApplyChange(
SpellcheckCustomDictionary* dictionary,
SpellcheckCustomDictionary::Change& change) {
int result = change.Sanitize(dictionary->GetWords());
dictionary->Apply(change);
dictionary->Notify(change);
dictionary->Sync(change);
return !result;
}
DISALLOW_COPY_AND_ASSIGN(DictionarySyncIntegrationTestHelper);
};
namespace dictionary_helper {
namespace {
SpellcheckCustomDictionary* GetDictionary(int index) {
return SpellcheckServiceFactory::GetForContext(
sync_datatype_helper::test()->GetProfile(index))->GetCustomDictionary();
}
SpellcheckCustomDictionary* GetVerifierDictionary() {
return SpellcheckServiceFactory::GetForContext(
sync_datatype_helper::test()->verifier())->GetCustomDictionary();
}
void LoadDictionary(SpellcheckCustomDictionary* dictionary) {
if (dictionary->IsLoaded())
return;
base::RunLoop run_loop;
DictionaryLoadObserver observer(content::GetQuitTaskForRunLoop(&run_loop));
dictionary->AddObserver(&observer);
dictionary->Load();
content::RunThisRunLoop(&run_loop);
dictionary->RemoveObserver(&observer);
ASSERT_TRUE(dictionary->IsLoaded());
}
} // namespace
void LoadDictionaries() {
for (int i = 0; i < sync_datatype_helper::test()->num_clients(); ++i)
LoadDictionary(GetDictionary(i));
if (sync_datatype_helper::test()->use_verifier())
LoadDictionary(GetVerifierDictionary());
}
size_t GetDictionarySize(int index) {
return GetDictionary(index)->GetWords().size();
}
size_t GetVerifierDictionarySize() {
return GetVerifierDictionary()->GetWords().size();
}
bool DictionariesMatch() {
const chrome::spellcheck_common::WordSet& reference =
sync_datatype_helper::test()->use_verifier()
? GetVerifierDictionary()->GetWords()
: GetDictionary(0)->GetWords();
for (int i = 0; i < sync_datatype_helper::test()->num_clients(); ++i) {
const chrome::spellcheck_common::WordSet& dictionary =
GetDictionary(i)->GetWords();
if (reference.size() != dictionary.size() ||
!std::equal(reference.begin(), reference.end(), dictionary.begin())) {
return false;
}
}
return true;
}
bool DictionaryMatchesVerifier(int index) {
const chrome::spellcheck_common::WordSet& expected =
GetVerifierDictionary()->GetWords();
const chrome::spellcheck_common::WordSet& actual =
GetDictionary(index)->GetWords();
return expected.size() == actual.size() &&
std::equal(expected.begin(), expected.end(), actual.begin());
}
bool AddWord(int index, const std::string& word) {
SpellcheckCustomDictionary::Change dictionary_change;
dictionary_change.AddWord(word);
bool result = DictionarySyncIntegrationTestHelper::ApplyChange(
GetDictionary(index), dictionary_change);
if (sync_datatype_helper::test()->use_verifier()) {
result &= DictionarySyncIntegrationTestHelper::ApplyChange(
GetVerifierDictionary(), dictionary_change);
}
return result;
}
bool RemoveWord(int index, const std::string& word) {
SpellcheckCustomDictionary::Change dictionary_change;
dictionary_change.RemoveWord(word);
bool result = DictionarySyncIntegrationTestHelper::ApplyChange(
GetDictionary(index), dictionary_change);
if (sync_datatype_helper::test()->use_verifier()) {
result &= DictionarySyncIntegrationTestHelper::ApplyChange(
GetVerifierDictionary(), dictionary_change);
}
return result;
}
} // namespace dictionary_helper
|
