tripal/tests/tripal_chado/fields/ChadoFieldGetValuesListTest.php

<?php
namespace Tests\tripal_chado\fields;

use StatonLab\TripalTestSuite\DBTransaction;
use StatonLab\TripalTestSuite\TripalTestCase;

/**
 * Test ChadoField->getValueList() Method.
 */
class ChadoFieldGetValuesListTest extends TripalTestCase {
  // Uncomment to auto start and rollback db transactions per test method.
  use DBTransaction;

  // Stores a list of fields to be tested including their storage method and instance info.
  private $field_list = NULL;

  /**
   * Test for fields based on columns in the base table.
   *
   * @group fields
   * @group getValueList
   */
  public function testBaseTableColumns() {
    include_once(drupal_get_path('tripal_chado', 'module') . '/includes/TripalFields/ChadoField.inc');

    // Retrieve a list of fields to test.
    // Note: this list is cached to improve performance.
    $fields = $this->retrieveFieldList();

    // Only iterate through fields that store their data in chado and
    // specifically, where the field stores it's data in the base table of the bundle
    // and is not a foreign key.
    foreach ($fields['field_chado_storage']['base'] as $key => $info) {
        $field_name = $info['field_name'];

        // Construct the Field instance we want the values for.
        // Specifying "ChadoField" here ensures we are only testing our 
        // implementation of getValueList() and not the custom version for any
        // given field.
        // YOU SHOULD TEST CUSTOM FIELD IMPLEMENTATIONS SEPARATELY.
        $instance = new \ChadoField($info['field_info'], $info['instance_info']);

        // Retrieve the values.
        // $values will be an array containing the distinct set of values for this field instance.
        $values = $instance->getValueList(array('limit' => 5));

        // Ensure we have values returned!
        $this->assertTrue(
          is_array($values), 
          t(
            'No values returned for @field_name (bundle: @bundle_name, bundle base table: @bundle_base_table, chado table: @chado_table, chado column: @chado_column).',
            array(
              '@field_name' => $field_name, 
              '@bundle_name' => $info['bundle_name'], 
              '@bundle_base_table' => $info['bundle_base_table'], 
              '@chado_table' => $info['instance_info']['settings']['chado_table'], 
              '@chado_column' => $info['instance_info']['settings']['chado_column'],
            )
          )
        );

      // Ensure there are no more then 5 as specified in the limit above.
      $this->assertLessThanOrEqual(5, sizeof($values),
        t('Returned too many results for @field_name.', array('@field_name' => $field_name)));

    }
  }

  /**
   * Test for fields based on columns in the base table that are also foreign keys.
   *
   * @group fields
   * @group getValueList
   */
  public function testBaseTableForeignKey() {
    include_once(drupal_get_path('tripal_chado', 'module') . '/includes/TripalFields/ChadoField.inc');

    // Retrieve a list of fields to test.
    // Note: this list is cached to improve performance.
    $fields = $this->retrieveFieldList();

    // Only iterate through fields that store their data in chado and
    // specifically, where the field stores it's data in the base table of the bundle
    // and IS a foreign key.
    foreach ($fields['field_chado_storage']['foreign key'] as $key => $info) {
        $field_name = $info['field_name'];
        
        // Construct the Field instance we want the values for.
        // Specifying "ChadoField" here ensures we are only testing our 
        // implementation of getValueList() and not the custom version for any
        // given field.
        // YOU SHOULD TEST CUSTOM FIELD IMPLEMENTATIONS SEPARATELY.
        $instance = new \ChadoField($info['field_info'], $info['instance_info']);
        
        // Retrieve the values using defaults.
        // $values will be an array containing the distinct set of values for this field instance.
        $values = $instance->getValueList(array('limit' => 5));
        
        // Ensure we have values returned!
        $this->assertTrue(
          is_array($values),
          t(
            'No values returned for @field_name with no label string set (bundle: @bundle_name, bundle base table: @bundle_base_table, chado table: @chado_table, chado column: @chado_column).',
            array(
              '@field_name' => $field_name, 
              '@bundle_name' => $info['bundle_name'], 
              '@bundle_base_table' => $info['bundle_base_table'], 
              '@chado_table' => $info['instance_info']['settings']['chado_table'], 
              '@chado_column' => $info['instance_info']['settings']['chado_column'],
            ) 
          )   
        );    
      
      // Ensure there are no more then 5 as specified in the limit above.
      $this->assertLessThanOrEqual(5, sizeof($values),
        t('Returned too many results for @field_name.', array('@field_name' => $field_name)));

      // @todo Ensure it works with a label string set.

    }
  }

  /**
   * Returns a list of Fields sorted by their backend, etc. for use in tests.
   */
  private function retrieveFieldList() {
    if ($this->field_list === NULL) {

      $this->field_list = array();

      $bundles = field_info_instances('TripalEntity');
      foreach($bundles as $bundle_name => $fields) {

        $bundle = tripal_load_bundle_entity(array('name'=> $bundle_name));

        foreach ($fields as $field_name => $instance_info) {
          $bundle_base_table = $base_schema = NULL;
          
          // Load the field info.
          $field_info = field_info_field($field_name);

          $storage = $field_info['storage']['type'];

          // If this field stores it's data in chado...
          // Determine the relationship between this field and the bundle base table.
          $rel = NULL;
          if ($storage == 'field_chado_storage') {

            // We need to know the table this field stores it's data in.
            $bundle_base_table = $bundle->data_table;
            // and the schema for that table.
            $base_schema = chado_get_schema($bundle_base_table);
            // and the table this field stores it's data in.
            $field_table = $instance_info['settings']['chado_table'];
            $field_column = $instance_info['settings']['chado_column'];

            // By default we simply assume there is some relationship.
            $rel = 'referring';
            // If the field and bundle store their data in the same table
            // then it's either a "base" or "foreign key" relationship
            // based on the schema.
            if ($bundle_base_table == $field_table) {

              // We assume it's not a foreign key...
              $rel = 'base';
              // and then check the schema to see if we're wrong :-)
              foreach ($base_schema['foreign keys'] as $schema_info) {
                if (isset($schema_info['columns'][ $field_column ])) { $rel = 'foreign key'; }
              }
            }
          }
         
          $info = array(
            'field_name' => $field_name,
            'bundle_name' => $bundle_name,
            'bundle' => $bundle,
            'bundle_base_table' => $bundle_base_table,
            'base_schema' => $base_schema,
            'field_info' => $field_info,
            'instance_info' => $instance_info,
          );

          $key = $bundle_name . '--' . $field_name;

          if ($rel) {
            $this->field_list[$storage][$rel][$key] = $info;
          }
          else {
            $this->field_list[$storage][$key] = $info;
          }

        }
      }
    }

    return $this->field_list;
  }

}