Source of file SolrReindexTest.php
Size: 11,500 Bytes - Last Modified: 2021-12-24T06:44:08+00:00
/var/www/docs.ssmods.com/process/src/tests/SolrReindexTest.php
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288 | <?php namespace SilverStripe\FullTextSearch\Tests; use SilverStripe\Core\Config\Config; use SilverStripe\Core\Injector\Injector; use SilverStripe\Dev\SapphireTest; use SilverStripe\FullTextSearch\Search\FullTextSearch; use SilverStripe\FullTextSearch\Search\Variants\SearchVariant; use SilverStripe\FullTextSearch\Solr\Reindex\Handlers\SolrReindexHandler; use SilverStripe\FullTextSearch\Solr\Services\Solr4Service; use SilverStripe\FullTextSearch\Solr\Services\SolrService; use SilverStripe\FullTextSearch\Solr\Tasks\Solr_Reindex; use SilverStripe\FullTextSearch\Tests\SolrReindexTest\SolrReindexTest_Index; use SilverStripe\FullTextSearch\Tests\SolrReindexTest\SolrReindexTest_Item; use SilverStripe\FullTextSearch\Tests\SolrReindexTest\SolrReindexTest_RecordingLogger; use SilverStripe\FullTextSearch\Tests\SolrReindexTest\SolrReindexTest_TestHandler; use SilverStripe\FullTextSearch\Tests\SolrReindexTest\SolrReindexTest_Variant; class SolrReindexTest extends SapphireTest { protected $usesDatabase = true; protected static $extra_dataobjects = array( SolrReindexTest_Item::class ); /** * Forced index for testing * * @var SolrReindexTest_Index */ protected $index = null; /** * Mock service * * @var SolrService */ protected $service = null; protected function setUp() { parent::setUp(); // Set test handler for reindex Config::modify()->set(Injector::class, SolrReindexHandler::class, array( 'class' => SolrReindexTest_TestHandler::class )); Injector::inst()->registerService(new SolrReindexTest_TestHandler(), SolrReindexHandler::class); // Set test variant SolrReindexTest_Variant::enable(); // Set index list $this->service = $this->getServiceMock(); $this->index = singleton(SolrReindexTest_Index::class); $this->index->setService($this->service); FullTextSearch::force_index_list($this->index); } /** * Populate database with dummy dataset * * @param int $number Number of records to create in each variant */ protected function createDummyData($number) { // Note that we don't create any records in variant = 2, to represent a variant // that should be cleared without any re-indexes performed foreach ([0, 1] as $variant) { for ($i = 1; $i <= $number; $i++) { $item = new SolrReindexTest_Item(); $item->Variant = $variant; $item->Title = "Item $variant / $i"; $item->write(); } } } /** * Mock service * * @return SolrService */ protected function getServiceMock() { $serviceMock = $this->getMockBuilder(Solr4Service::class) ->setMethods(['deleteByQuery', 'addDocument']); return $serviceMock->getMock(); } protected function tearDown() { FullTextSearch::force_index_list(); SolrReindexTest_Variant::disable(); parent::tearDown(); } /** * Get the reindex handler * * @return SolrReindexHandler */ protected function getHandler() { return Injector::inst()->get(SolrReindexHandler::class); } /** * Ensure the test variant is up and running properly */ public function testVariant() { // State defaults to 0 $variant = SearchVariant::current_state(); $this->assertEquals( array( SolrReindexTest_Variant::class => "0" ), $variant ); // All states enumerated $allStates = iterator_to_array(SearchVariant::reindex_states()); $this->assertEquals( array( array( SolrReindexTest_Variant::class => "0" ), array( SolrReindexTest_Variant::class => "1" ), array( SolrReindexTest_Variant::class => "2" ) ), $allStates ); // Check correct items created and that filtering on variant works $this->createDummyData(120); SolrReindexTest_Variant::set_current(2); $this->assertEquals(0, SolrReindexTest_Item::get()->count()); SolrReindexTest_Variant::set_current(1); $this->assertEquals(120, SolrReindexTest_Item::get()->count()); SolrReindexTest_Variant::set_current(0); $this->assertEquals(120, SolrReindexTest_Item::get()->count()); SolrReindexTest_Variant::disable(); $this->assertEquals(240, SolrReindexTest_Item::get()->count()); } /** * Given the invocation of a new re-index with a given set of data, ensure that the necessary * list of groups are created and segmented for each state * * Test should work fine with any variants (versioned, subsites, etc) specified */ public function testReindexSegmentsGroups() { $this->service->method('deleteByQuery') ->withConsecutive( ['-(ClassHierarchy:' . SolrReindexTest_Item::class . ')'], ['+(ClassHierarchy:' . SolrReindexTest_Item::class . ') +(_testvariant:"2")'] ); $this->createDummyData(120); // Initiate re-index $logger = new SolrReindexTest_RecordingLogger(); $this->getHandler()->runReindex($logger, 21, Solr_Reindex::class); // Test that invalid classes are removed $this->assertContains( 'Clearing obsolete classes from ' . str_replace('\\', '-', SolrReindexTest_Index::class), $logger->getMessages() ); // Test that valid classes in invalid variants are removed $this->assertContains( 'Clearing all records of type ' . SolrReindexTest_Item::class . ' in the current state: {' . json_encode(SolrReindexTest_Variant::class) . ':"2"}', $logger->getMessages() ); // 120x2 grouped into groups of 21 results in 12 groups $this->assertEquals(12, $logger->countMessages('Called processGroup with ')); $this->assertEquals(6, $logger->countMessages('{' . json_encode(SolrReindexTest_Variant::class) . ':"0"}')); $this->assertEquals(6, $logger->countMessages('{' . json_encode(SolrReindexTest_Variant::class) . ':"1"}')); // Given that there are two variants, there should be two group ids of each number $this->assertEquals(2, $logger->countMessages(' ' . SolrReindexTest_Item::class . ', group 0 of 6')); $this->assertEquals(2, $logger->countMessages(' ' . SolrReindexTest_Item::class . ', group 1 of 6')); $this->assertEquals(2, $logger->countMessages(' ' . SolrReindexTest_Item::class . ', group 2 of 6')); $this->assertEquals(2, $logger->countMessages(' ' . SolrReindexTest_Item::class . ', group 3 of 6')); $this->assertEquals(2, $logger->countMessages(' ' . SolrReindexTest_Item::class . ', group 4 of 6')); $this->assertEquals(2, $logger->countMessages(' ' . SolrReindexTest_Item::class . ', group 5 of 6')); // Check various group sizes $logger->clear(); $this->getHandler()->runReindex($logger, 120, 'Solr_Reindex'); $this->assertEquals(2, $logger->countMessages('Called processGroup with ')); $logger->clear(); $this->getHandler()->runReindex($logger, 119, 'Solr_Reindex'); $this->assertEquals(4, $logger->countMessages('Called processGroup with ')); $logger->clear(); $this->getHandler()->runReindex($logger, 121, 'Solr_Reindex'); $this->assertEquals(2, $logger->countMessages('Called processGroup with ')); $logger->clear(); $this->getHandler()->runReindex($logger, 2, 'Solr_Reindex'); $this->assertEquals(120, $logger->countMessages('Called processGroup with ')); } /** * Test index processing on individual groups */ public function testRunGroup() { $this->service->method('deleteByQuery') ->with('+(ClassHierarchy:' . SolrReindexTest_Item::class . ') +_query_:"{!frange l=2 u=2}mod(ID, 6)" +(_testvariant:"1")'); $this->createDummyData(120); $logger = new SolrReindexTest_RecordingLogger(); // Initiate re-index of third group (index 2 of 6) $state = array(SolrReindexTest_Variant::class => '1'); $this->getHandler()->runGroup($logger, $this->index, $state, SolrReindexTest_Item::class, 6, 2); $idMessage = $logger->filterMessages('Updated '); $this->assertNotEmpty(preg_match('/^Updated (?<ids>[,\d]+)/i', $idMessage[0], $matches)); $ids = array_unique(explode(',', $matches['ids'])); // Test successful $this->assertNotEmpty($logger->getMessages('Adding ' . SolrReindexTest_Item::class)); $this->assertNotEmpty($logger->getMessages('Done')); // Test that items in this variant / group are re-indexed // 120 divided into 6 groups should be 20 at least (max 21) $this->assertEquals(21, count($ids), 'Group size is about 20', 1); foreach ($ids as $id) { // Each id should be % 6 == 2 $this->assertEquals(2, $id % 6, "ID $id Should match pattern ID % 6 = 2"); } } /** * Test that running all groups covers the entire range of dataobject IDs */ public function testRunAllGroups() { $this->service->method('deleteByQuery') ->withConsecutive( ['+(ClassHierarchy:' . SolrReindexTest_Item::class . ') +_query_:"{!frange l=0 u=0}mod(ID, 6)" +(_testvariant:"1")'], ['+(ClassHierarchy:' . SolrReindexTest_Item::class . ') +_query_:"{!frange l=1 u=1}mod(ID, 6)" +(_testvariant:"1")'], ['+(ClassHierarchy:' . SolrReindexTest_Item::class . ') +_query_:"{!frange l=2 u=2}mod(ID, 6)" +(_testvariant:"1")'], ['+(ClassHierarchy:' . SolrReindexTest_Item::class . ') +_query_:"{!frange l=3 u=3}mod(ID, 6)" +(_testvariant:"1")'], ['+(ClassHierarchy:' . SolrReindexTest_Item::class . ') +_query_:"{!frange l=4 u=4}mod(ID, 6)" +(_testvariant:"1")'], ['+(ClassHierarchy:' . SolrReindexTest_Item::class . ') +_query_:"{!frange l=5 u=5}mod(ID, 6)" +(_testvariant:"1")'], ['+(ClassHierarchy:' . SolrReindexTest_Item::class . ') +_query_:"{!frange l=6 u=6}mod(ID, 6)" +(_testvariant:"1")'] ); $this->createDummyData(120); $logger = new SolrReindexTest_RecordingLogger(); // Test that running all groups covers the complete set of ids $state = array(SolrReindexTest_Variant::class => '1'); for ($i = 0; $i < 6; $i++) { // See testReindexSegmentsGroups for test that each of these states is invoked during a full reindex $this ->getHandler() ->runGroup($logger, $this->index, $state, SolrReindexTest_Item::class, 6, $i); } // Count all ids updated $ids = array(); foreach ($logger->filterMessages('Updated ') as $message) { $this->assertNotEmpty(preg_match('/^Updated (?<ids>[,\d]+)/', $message, $matches)); $ids = array_unique(array_merge($ids, explode(',', $matches['ids']))); } // Check ids $this->assertEquals(120, count($ids)); } } |