solr_base_query.test in Apache Solr Search 7

File

tests/solr_base_query.test

<?php

/**
 * Unit tests for query object methods.
 */
class SolrBaseQueryTests extends DrupalUnitTestCase {
  public static function getInfo() {
    return array(
      'name' => 'SolrBaseQuery Unit tests',
      'description' => 'Unit Tests for queries.',
      'group' => 'ApacheSolr',
    );
  }
  function setUp() {
    parent::setUp();
    require_once dirname(dirname(realpath(__FILE__))) . '/apachesolr.module';
    require_once dirname(dirname(realpath(__FILE__))) . '/apachesolr.interface.inc';
    require_once dirname(dirname(realpath(__FILE__))) . '/Solr_Base_Query.php';
    require_once dirname(dirname(realpath(__FILE__))) . '/Drupal_Apache_Solr_Service.php';
    require_once dirname(dirname(realpath(__FILE__))) . '/tests/Dummy_Solr.php';
  }

  /**
   * Helper function to simulate the auto loading and other non-needed functions
   * that otherwise require a database
   * @see apachesolr_drupal_query().
   * @return SolrBaseQuery
   */
  private function _apachesolr_drupal_query($name, $params = array(), $solrsort = '', $base_path = '', $solr = NULL) {
    if (empty($solr)) {
      $solr = new DummySolr(NULL);
    }
    return new SolrBaseQuery($name, $solr, $params, $solrsort, $base_path);
  }
  private function _apachesolr_drupal_subquery($operator = 'OR') {
    return new SolrFilterSubQuery($operator);
  }

  /**
   * Test ordering of parsed filter positions.
   *
   * Regression test for http://drupal.org/node/891962
   */
  function testParseFilters() {
    $fq = array(
      'tid:3',
      'sort_label:hello',
      'tid:11',
      'tid:1',
      'tid:12',
      'label:hello',
    );

    // Setup dummy Solr object.
    $query = $this
      ->_apachesolr_drupal_query("apachesolr_tests", array(
      'q' => 'mykeys',
      'fq' => $fq,
    ), 'sort_label asc', 'search/test');

    // Check setSolrSort().
    $this
      ->assertEqual(array(
      'solrsort' => 'sort_label asc',
    ), $query
      ->getSolrsortUrlQuery());
    $query
      ->setSolrsort('sort_name', 'desc');
    $this
      ->assertEqual(array(
      'solrsort' => 'sort_name desc',
    ), $query
      ->getSolrsortUrlQuery());
    $query
      ->setSolrsort('score', 'desc');
    $this
      ->assertEqual(array(), $query
      ->getSolrsortUrlQuery());

    // We cannot set a sort if the sort was not made available.
    $query
      ->setSolrsort('geodist()', 'asc');
    $this
      ->assertNotEqual(array(
      'solrsort' => 'geodist() asc',
    ), $query
      ->getSolrsortUrlQuery(), t('Sort was not applied as it was not available'));

    // Set the sort after it was made available.
    $query
      ->setAvailableSort('geodist()', array(
      'title' => 'Distance',
      'default' => 'asc',
    ));
    $query
      ->setSolrsort('geodist()', 'asc');
    $this
      ->assertEqual(array(
      'solrsort' => 'geodist() asc',
    ), $query
      ->getSolrsortUrlQuery());

    // Check escaping of the setSolrSort functinality
    // Check getPath() functionality
    $this
      ->assertEqual('search/test/mykeys', $query
      ->getPath());
    $this
      ->assertEqual('search/test/newkeys', $query
      ->getPath('newkeys'));

    // Check hasFilter functionality
    $this
      ->assertFalse($query
      ->hasFilter('label', 'Jello'), "hasFilter('label', 'Jello') is FALSE");
    $this
      ->assertTrue($query
      ->hasFilter('label', 'hello'), "hasFilter('label', 'hello') is TRUE");
    $this
      ->assertTrue($query
      ->hasFilter('label', 'hello', FALSE), "hasFilter('label', 'hello', FALSE) is TRUE");
    $this
      ->assertFalse($query
      ->hasFilter('label', 'hello', TRUE), "hasFilter('label', 'hello', TRUE) is FALSE");
    $filters = $query
      ->getFilters();
    $this
      ->assertEqual(count($filters), 6, count($filters) . ' filters found, expected 6 filters');

    // Check positions of filters
    foreach ($fq as $idx => $filter) {
      $this
        ->assertEqual($filter, $query
        ->makeFilterQuery($filters[$idx]));
    }

    // Check that the query string is re-assembled correctly
    $this
      ->assertEqual($fq, $query
      ->getParam('fq'));
    $this
      ->assertEqual('mykeys', $query
      ->getParam('q'));
    $query
      ->removeFilter('tid', '11');
    $filters = $query
      ->getFilters();
    $this
      ->assertEqual(count($filters), 5, count($filters) . ' filters found, expected 5 filters');
    $this
      ->assertEqual(array(
      'tid:3',
      'sort_label:hello',
      'tid:1',
      'tid:12',
      'label:hello',
    ), $query
      ->getParam('fq'));
    $query
      ->removeFilter('tid');
    $filters = $query
      ->getFilters();
    $this
      ->assertEqual(count($filters), 2, count($filters) . ' filters found, expected 2 filters');
    $this
      ->assertEqual(array(
      'sort_label:hello',
      'label:hello',
    ), $query
      ->getParam('fq'));
    $subquery = $this
      ->_apachesolr_drupal_subquery();
    $subquery
      ->addFilter('access__all', 0);
    $subquery
      ->addFilter('hash', 'randomhash');
    $query
      ->addFilterSubQuery($subquery);
    $this
      ->assertEqual(count($query
      ->getParam('fq')), 3, count($query
      ->getParam('fq')) . ' fq params found, expected 3 after adding subquery');
    $this
      ->assertEqual(array(
      'sort_label:hello',
      'label:hello',
      '(access__all:0 OR hash:randomhash' . ')',
    ), $query
      ->getParam('fq'));
  }
  function testAddParams() {
    $examples = array();
    $examples['{!cache=false}inStock:true'] = array(
      '#local' => 'cache=false',
      '#exclude' => FALSE,
      '#name' => 'inStock',
      '#value' => 'true',
    );
    $examples['{!frange l=1 u=4 cache=false}sqrt(popularity)'] = array(
      '#local' => 'frange l=1 u=4 cache=false',
      '#exclude' => FALSE,
      '#name' => NULL,
      '#value' => 'sqrt(popularity)',
    );
    $examples['{!cache=false cost=5}inStock:true'] = array(
      '#local' => 'cache=false cost=5',
      '#exclude' => FALSE,
      '#name' => 'inStock',
      '#value' => 'true',
    );
    $examples['{!tag=impala}model:Impala'] = array(
      '#local' => 'tag=impala',
      '#exclude' => FALSE,
      '#name' => 'model',
      '#value' => 'Impala',
    );
    $examples['{!anything that appears to be local}'] = array(
      '#local' => 'anything that appears to be local',
      '#exclude' => FALSE,
      '#name' => NULL,
      '#value' => NULL,
    );
    $examples['bundle:(article OR page)'] = array(
      '#local' => NULL,
      '#exclude' => FALSE,
      '#name' => 'bundle',
      '#value' => '(article OR page)',
    );
    $examples['-bundle:(article OR page)'] = array(
      '#local' => NULL,
      '#exclude' => TRUE,
      '#name' => 'bundle',
      '#value' => '(article OR page)',
    );
    $examples['-{!anything that appears to be local}'] = array(
      '#local' => 'anything that appears to be local',
      '#exclude' => TRUE,
      '#name' => NULL,
      '#value' => NULL,
    );
    $examples['title:"double words"'] = array(
      '#local' => NULL,
      '#exclude' => FALSE,
      '#name' => 'title',
      '#value' => '"double words"',
    );
    $examples['field_date:[1970-12-31T23:59:59Z TO NOW]'] = array(
      '#local' => NULL,
      '#exclude' => FALSE,
      '#name' => 'field_date',
      '#value' => '[1970-12-31T23:59:59Z TO NOW]',
    );
    $query = $this
      ->_apachesolr_drupal_query("apachesolr_tests");
    foreach ($examples as $fq => $example) {
      $name = !empty($example['#name']) ? $example['#name'] : '_QUERY_';
      $value = !empty($example['#value']) ? $example['#value'] : '_VALUE_';
      $filter = $name . ':' . $value;

      // Check if filter is seen as a valid one
      $message = t('Filter (@fq) is Valid', array(
        '@fq' => $filter,
      ));
      $this
        ->assertTrue($query
        ->validFilterValue($filter), $message);
      $query
        ->addParam('fq', $fq);

      // Check if filter was added
      $message = t('hasFilter(@name, @value) is True', array(
        '@name' => $example['#name'],
        '@value' => $example['#value'],
      ));
      $this
        ->assertTrue($query
        ->hasFilter($example['#name'], $example['#value'], $example['#exclude']), $message);
      $filters = $query
        ->getFilters();
      $filter = reset($filters);
      $message = t('The filter "@fq" was added with all the given properties', array(
        '@fq' => $fq,
      ));
      $this
        ->assertTrue(!array_diff($example, $filter), $message);
      $query
        ->removeFilter($example['#name']);
      $message = t('The filter "@fq" was correctly removed', array(
        '@fq' => $fq,
      ));
      $this
        ->assertFalse($query
        ->hasFilter($example['#name'], $example['#value'], $example['#exclude']), $message);

      // Since we cannot remove filters without names yet we have to clear the whole fq array
      $query
        ->removeParam('fq');

      // Check the ones without the name also
    }

    // Check for invalid combinations
    $bad_examples['wrong name:"double words"'] = array(
      '#local' => NULL,
      '#exclude' => FALSE,
      '#name' => 'wrong name',
      '#value' => '"double words"',
    );
    $bad_examples['field_date:[1970-12-31 TO NOW]'] = array(
      '#local' => NULL,
      '#exclude' => FALSE,
      '#name' => 'field_date',
      '#value' => '[1970-12-31 TO NOW]',
    );
    $bad_examples['bundle:((article OR page)]'] = array(
      '#local' => NULL,
      '#exclude' => FALSE,
      '#name' => 'bundle',
      '#value' => '((article OR page)]',
    );
    foreach ($bad_examples as $fq => $example) {
      $name = !empty($example['#name']) ? $example['#name'] : '_QUERY_';
      $value = !empty($example['#value']) ? $example['#value'] : '_VALUE_';
      $filter = $name . ':' . $value;

      // Check if filter is seen as a valid one
      $message = t('Filter (@fq) is not Valid', array(
        '@fq' => $filter,
      ));
      $this
        ->assertFalse($query
        ->validFilterValue($filter), $message);
    }

    // Check parameter normalization.
    $query
      ->addParam('spellcheck', TRUE);
    $this
      ->assertTrue($query
      ->getParam('spellcheck') === 'true', "TRUE normalized to string 'true'");
    $query
      ->replaceParam('spellcheck', FALSE);
    $this
      ->assertTrue($query
      ->getParam('spellcheck') === 'false', "FALSE normalized to string 'false'");
    $query
      ->addParam('fl', array(
      ' x ',
      TRUE,
    ));
    $this
      ->assertTrue($query
      ->getParam('fl') === array(
      'x',
      'true',
    ), "Array of params all normalized");
  }

}

Classes

Name	Description
SolrBaseQueryTests	Unit tests for query object methods.