1
|
|
|
<?php |
2
|
|
|
|
3
|
|
|
use SilverStripe\Elastica\ReindexTask; |
4
|
|
|
use SilverStripe\Elastica\ElasticaService; |
5
|
|
|
|
6
|
|
|
|
7
|
|
|
class ElasticsearchFunctionalTestBase extends FunctionalTest { |
8
|
|
|
public static $ignoreFixtureFileFor = array(); |
9
|
|
|
|
10
|
|
|
protected $extraDataObjects = array( |
11
|
|
|
'SearchableTestPage','FlickrPhotoTO','FlickrAuthorTO','FlickrSetTO','FlickrTagTO', |
12
|
|
|
'SearchableTestFatherPage','SearchableTestGrandFatherPage' |
13
|
|
|
); |
14
|
|
|
|
15
|
|
|
|
16
|
|
|
public function setUpOnce() { |
17
|
|
|
$config = Config::inst(); |
18
|
|
|
$config->remove('Injector', 'SilverStripe\Elastica\ElasticaService'); |
19
|
|
|
$constructor = array('constructor' => array('%$Elastica\Client', 'elastica_ss_module_test')); |
20
|
|
|
$config->update('Injector', 'SilverStripe\Elastica\ElasticaService', $constructor); |
21
|
|
|
parent::setUpOnce(); |
22
|
|
|
} |
23
|
|
|
|
24
|
|
|
|
25
|
|
|
public function setUp() { |
|
|
|
|
26
|
|
|
$cache = SS_Cache::factory('elasticsearch'); |
27
|
|
|
$cache->clean(Zend_Cache::CLEANING_MODE_ALL); |
28
|
|
|
SS_Cache::set_cache_lifetime('elasticsearch', 3600, 1000); |
29
|
|
|
|
30
|
|
|
// this needs to be called in order to create the list of searchable |
31
|
|
|
// classes and fields that are available. Simulates part of a build |
32
|
|
|
$classes = array('SearchableTestPage','SiteTree','Page','FlickrPhotoTO','FlickrSetTO', |
33
|
|
|
'FlickrTagTO', 'FlickrAuthorTO'); |
34
|
|
|
$this->requireDefaultRecordsFrom = $classes; |
35
|
|
|
|
36
|
|
|
// add Searchable extension where appropriate |
37
|
|
|
FlickrSetTO::add_extension('SilverStripe\Elastica\Searchable'); |
38
|
|
|
FlickrPhotoTO::add_extension('SilverStripe\Elastica\Searchable'); |
39
|
|
|
FlickrTagTO::add_extension('SilverStripe\Elastica\Searchable'); |
40
|
|
|
FlickrAuthorTO::add_extension('SilverStripe\Elastica\Searchable'); |
41
|
|
|
SearchableTestPage::add_extension('SilverStripe\Elastica\Searchable'); |
42
|
|
|
|
43
|
|
|
// clear the index |
44
|
|
|
$this->service = Injector::inst()->create('SilverStripe\Elastica\ElasticaService'); |
45
|
|
|
$this->service->setTestMode(true); |
46
|
|
|
|
47
|
|
|
// A previous test may have deleted the index and then failed, so check for this |
48
|
|
|
if (!$this->service->getIndex()->exists()) { |
49
|
|
|
$this->service->getIndex()->create(); |
50
|
|
|
} |
51
|
|
|
$this->service->reset(); |
52
|
|
|
|
53
|
|
|
// FIXME - use request getVar instead? |
54
|
|
|
$_GET['progress'] = 20; |
55
|
|
|
// load fixtures |
56
|
|
|
|
57
|
|
|
$orig_fixture_file = static::$fixture_file; |
58
|
|
|
|
59
|
|
View Code Duplication |
foreach (static::$ignoreFixtureFileFor as $testPattern) { |
|
|
|
|
60
|
|
|
$pattern = '/'.$testPattern.'/'; |
61
|
|
|
if (preg_match($pattern, $this->getName())) { |
|
|
|
|
62
|
|
|
static::$fixture_file = null; |
63
|
|
|
} |
64
|
|
|
} |
65
|
|
|
|
66
|
|
|
parent::setUp(); |
67
|
|
|
static::$fixture_file = $orig_fixture_file; |
68
|
|
|
|
69
|
|
|
$this->publishSiteTree(); |
70
|
|
|
|
71
|
|
|
$this->service->reset(); |
72
|
|
|
|
73
|
|
|
// index loaded fixtures |
74
|
|
|
$task = new ReindexTask($this->service); |
75
|
|
|
// null request is fine as no parameters used |
76
|
|
|
|
77
|
|
|
$task->run(null); |
78
|
|
|
|
79
|
|
|
} |
80
|
|
|
|
81
|
|
|
|
82
|
|
|
private function publishSiteTree() { |
83
|
|
|
foreach (SiteTree::get()->getIterator() as $page) { |
84
|
|
|
// temporarily disable Elasticsearch indexing, it will be done in a batch |
85
|
|
|
$page->IndexingOff = true; |
86
|
|
|
|
87
|
|
|
$page->publish('Stage','Live'); |
88
|
|
|
} |
89
|
|
|
} |
90
|
|
|
|
91
|
|
|
|
92
|
|
|
//---- The HTML for search results is too long to check for, so instead check just the starting text ---- |
93
|
|
|
|
94
|
|
|
/** |
95
|
|
|
*Assert that the indexth matching css node has a prefix as expected |
96
|
|
|
* |
97
|
|
|
* Note: characters are stripped from the content; make sure that your assertions take this into account. |
98
|
|
|
* |
99
|
|
|
* @param string $selector A basic CSS selector, e.g. 'li.jobs h3' |
100
|
|
|
* @param array|string $expectedMatches The content of at least one of the matched tags |
|
|
|
|
101
|
|
|
* @throws PHPUnit_Framework_AssertionFailedError |
102
|
|
|
* @return boolean |
103
|
|
|
*/ |
104
|
|
|
public function assertSelectorStartsWithOrEquals($selector, $index, $expectedPrefix) { |
105
|
|
|
$items = $this->cssParser()->getBySelector($selector); |
106
|
|
|
|
107
|
|
|
$ctr = 0; |
108
|
|
|
foreach ($items as $item) { |
109
|
|
|
$text = strip_tags($item); |
110
|
|
|
$escaped = str_replace("'", "\'", $text); |
|
|
|
|
111
|
|
|
$ctr++; |
112
|
|
|
} |
113
|
|
|
|
114
|
|
|
$ctr = 0; |
|
|
|
|
115
|
|
|
$item = strip_tags($items[$index]); |
116
|
|
|
|
117
|
|
|
|
118
|
|
|
$errorMessage = "Failed to assert that '$item' started with '$expectedPrefix'"; |
119
|
|
|
$this->assertStringStartsWith($expectedPrefix, $item, $errorMessage); |
|
|
|
|
120
|
|
|
|
121
|
|
|
return true; |
122
|
|
|
} |
123
|
|
|
|
124
|
|
|
|
125
|
|
|
/* |
126
|
|
|
Check all the nodes matching the selector for attribute name = expected value |
127
|
|
|
*/ |
128
|
|
|
public function assertAttributeHasExactValue($selector, $attributeName, $expectedValue) { |
|
|
|
|
129
|
|
|
$items = $this->cssParser()->getBySelector($selector); |
130
|
|
|
foreach ($items as $item) { |
131
|
|
|
$this->assertEquals($expectedValue, $item['value']); |
|
|
|
|
132
|
|
|
} |
133
|
|
|
} |
134
|
|
|
|
135
|
|
|
|
136
|
|
|
public function assertAttributesHaveExactValues($selector, $expectedValues) { |
137
|
|
|
$attributeNames = array_keys($expectedValues); |
138
|
|
|
$items = $this->cssParser()->getBySelector($selector); |
139
|
|
|
foreach ($items as $item) { |
140
|
|
|
$actualValues = array(); |
141
|
|
|
foreach ($attributeNames as $attributeName) { |
142
|
|
|
$actualValues[$attributeName] = (string)$item[$attributeName]; |
143
|
|
|
} |
144
|
|
|
$this->assertEquals($expectedValues, $actualValues); |
|
|
|
|
145
|
|
|
} |
146
|
|
|
} |
147
|
|
|
|
148
|
|
|
|
149
|
|
|
public function assertNumberOfNodes($selector, $expectedAmount) { |
150
|
|
|
$items = $this->cssParser()->getBySelector($selector); |
151
|
|
|
foreach ($items as $item) { |
152
|
|
|
$text = strip_tags($item); |
|
|
|
|
153
|
|
|
} |
154
|
|
|
|
155
|
|
|
$ct = sizeof($items); |
156
|
|
|
$this->assertEquals($expectedAmount, $ct); |
|
|
|
|
157
|
|
|
} |
158
|
|
|
|
159
|
|
|
|
160
|
|
|
/* Collect an array of all the <ClassName>_<ID> search results, used for checking pagination */ |
161
|
|
|
public function collateSearchResults() { |
162
|
|
|
$items = $this->cssParser()->getBySelector('div.searchResults .searchResult'); |
163
|
|
|
$result = array(); |
164
|
|
|
foreach ($items as $item) { |
165
|
|
|
$attr = $item->attributes()->id; |
166
|
|
|
array_push($result, $attr.""); |
167
|
|
|
} |
168
|
|
|
|
169
|
|
|
return $result; |
170
|
|
|
} |
171
|
|
|
|
172
|
|
|
} |
173
|
|
|
|
Instead of super-globals, we recommend to explicitly inject the dependencies of your class. This makes your code less dependent on global state and it becomes generally more testable: