You are here

class LdapTestFunctions in Lightweight Directory Access Protocol (LDAP) 8.2

Same name and namespace in other branches
  1. 7.2 ldap_test/LdapTestFunctions.class.php \LdapTestFunctions
  2. 7 ldap_servers/tests/LdapTestFunctions.class.php \LdapTestFunctions

Hierarchy

Expanded class hierarchy of LdapTestFunctions

File

ldap_test/LdapTestFunctions.class.php, line 16

View source
class LdapTestFunctions {
  public $data = array();
  public $ldapData = array();

  // data in ldap array format, but keyed on dn
  public $csvTables = array();
  public $ldapTypeConf;
  function __construct() {
    module_load_include('module', 'ldap_servers');
    $this->data['ldap_servers'] = ldap_test_ldap_servers_data();
    module_load_include('module', 'ldap_user');
    $this->data['ldap_user'] = ldap_test_ldap_user_data();
    module_load_include('module', 'ldap_authentication');
    $this->data['ldap_authorization'] = ldap_test_ldap_authorization_data();
    module_load_include('module', 'ldap_authorization');
    $this->data['ldap_authentication'] = ldap_test_ldap_authentication_data();
  }
  function configureLdapServers($sids, $feetures = FALSE, $feature_name = NULL) {
    foreach ($sids as $i => $sid) {
      $current_sids[$sid] = $sid;
      variable_set('ldap_test_server__' . $sid, $this->data['ldap_servers'][$sid]);
    }
    variable_set('ldap_test_servers', $current_sids);
  }
  function setFakeServerProperty($sid, $prop, $value) {
    $test_data = variable_get('ldap_test_server__' . $sid, array());
    $test_data['properties'][$prop] = $value;
    variable_set('ldap_test_server__' . $sid, $test_data);
  }
  function setFakeServerUserAttribute($sid, $dn, $attr_name, $attr_value, $i = 0) {
    $attr_name = drupal_strtolower($attr_name);
    $test_data = variable_get('ldap_test_server__' . $sid, array());
    $test_data['entries'][$dn][$attr_name][$i] = $attr_value;
    $count_set = (int) isset($test_data['entries'][$dn][$attr_name]['count']);
    $test_data['entries'][$dn][$attr_name]['count'] = count($test_data['entries'][$dn][$attr_name]) - $count_set;

    // don't count the 'count'
    $test_data['ldap'][$dn][$attr_name][$i] = $attr_value;
    $count_set = (int) isset($test_data['ldap'][$dn][$attr_name]['count']);
    $test_data['ldap'][$dn][$attr_name]['count'] = count($test_data['ldap'][$dn][$attr_name]) - $count_set;

    // don't count the 'count'
    variable_set('ldap_test_server__' . $sid, $test_data);
    $ldap_server = ldap_servers_get_servers($sid, NULL, TRUE, TRUE);

    // clear server cache;
  }
  function configureLdapAuthentication($ldap_authentication_test_conf_id, $sids) {
    module_load_include('php', 'ldap_authentication', 'LdapAuthenticationConfAdmin.class');
    $options = $this->data['ldap_authentication'][$ldap_authentication_test_conf_id];
    foreach ($sids as $i => $sid) {
      $options['sids'][$sid] = $sid;
    }
    $ldapServerAdmin = new LdapAuthenticationConfAdmin();
    foreach ($ldapServerAdmin->saveable as $prop_name) {
      if (isset($options[$prop_name])) {
        $ldapServerAdmin->{$prop_name} = $options[$prop_name];
      }
    }
    $ldapServerAdmin
      ->save();
  }
  function configureLdapUser($ldap_user_test_conf_id) {
    module_load_include('php', 'ldap_user', 'LdapUserConfAdmin.class');
    $ldapUserConfAdmin = new LdapUserConfAdmin();
    $options = $this->data['ldap_user'][$ldap_user_test_conf_id];
    foreach ($ldapUserConfAdmin->saveable as $prop_name) {
      if (isset($options[$prop_name])) {
        $ldapUserConfAdmin->{$prop_name} = $options[$prop_name];
      }
    }
    $ldapUserConfAdmin
      ->save();
  }
  function prepConsumerConf($consumer_confs) {

    // create consumer authorization configuration.
    foreach ($consumer_confs as $consumer_type => $consumer_conf) {
      $consumer_obj = ldap_authorization_get_consumer_object($consumer_type);
      $consumer_conf_admin = new LdapAuthorizationConsumerConfAdmin($consumer_obj, TRUE);
      foreach ($consumer_conf as $property_name => $property_value) {
        $consumer_conf_admin->{$property_name} = $property_value;
      }
      foreach ($consumer_conf_admin->mappings as $i => $mapping) {
        $mappings = $consumer_obj
          ->normalizeMappings(array(
          array(
            $mapping['from'],
            $mapping['user_entered'],
          ),
        ), FALSE);
        $consumer_conf_admin->mappings[$i] = $mappings[0];
      }
      $consumer_conf_admin
        ->save();
    }
  }
  function ldapUserIsAuthmapped($username) {
    $authmaps = user_get_authmaps($username);
    return $authmaps && in_array('ldap_user', array_keys($authmaps));
  }
  function drupalLdapUpdateUser($edit = array(), $ldap_authenticated = FALSE, $user) {
    if (count($edit)) {
      $user = user_save($user, $edit);
    }
    if ($ldap_authenticated) {
      user_set_authmaps($user, array(
        'authname_ldap_user' => $user->name,
      ));
    }
    return $user;
  }

  // from http://www.midwesternmac.com/blogs/jeff-geerling/programmatically-adding-roles
  public function removeRoleFromUser($user, $role_name) {
    if (is_numeric($user)) {
      $user = user_load($user);
    }
    $key = array_search($role_name, $user->roles);
    if ($key == TRUE) {

      // Get the rid from the roles table.
      $roles = user_roles(TRUE);
      $rid = array_search($role_name, $roles);
      if ($rid != FALSE) {

        // Make a copy of the roles array, without the deleted one.
        $new_roles = array();
        foreach ($user->roles as $id => $name) {
          if ($id != $rid) {
            $new_roles[$id] = $name;
          }
        }
        user_save($user, array(
          'roles' => $new_roles,
        ));
      }
    }
  }
  public function userByNameFlushingCache($name) {
    $user = user_load_by_name($name);
    $users = user_load_multiple(array(
      $user->uid,
    ), array(), TRUE);

    // clear user cache
    $user = $users[$user->uid];
    return $user;
  }

  /**
   * set variable with fake test data
   *
   * @param string $test_ldap_id eg. 'hogwarts'
   * @param string $test_ldap_type e.g. openLdap, openLdapTest1, etc.
   * @parma string $sid where fake data is stored. e.g. 'default',
   */
  public function populateFakeLdapServerData($test_ldap_id, $sid = 'default') {

    // read csvs into key/value array
    // create fake ldap data array
    $clones = empty($this->data['ldap_servers'][$sid]['clones']) ? FALSE : $this->data['ldap_servers'][$sid]['clones'];
    $server_properties = $this->data['ldap_servers'][$sid]['properties'];
    $this
      ->getCsvLdapData($test_ldap_id);
    foreach ($this->csvTables['users'] as $guid => $user) {
      $dn = 'cn=' . $user['cn'] . ',' . $this->csvTables['conf'][$test_ldap_id]['userbasedn'];
      $this->csvTables['users'][$guid]['dn'] = $dn;
      $attributes = $this
        ->generateUserLDAPAttributes($test_ldap_id, $user);
      $this
        ->addLDAPUserToLDAPArraysFromAttributes($user, $sid, $dn, $attributes, $server_properties['ldap_type'], $server_properties['user_attr']);
    }
    if ($clones) {
      $clonable_user = $this->csvTables['users'][101];
      for ($i = 0; $i < $clones; $i++) {
        $user = $clonable_user;
        $cn = "clone" . $i;
        $dn = 'cn=' . $cn . ',' . $this->csvTables['conf'][$test_ldap_id]['userbasedn'];
        $user['cn'] = $cn;
        $user['dn'] = $dn;
        $user['uid'] = 20 + $i;
        $user['guid'] = 120 + $i;
        $user['lname'] = $user['lname'] . "_{$i}";
        $attributes = $this
          ->generateUserLDAPAttributes($test_ldap_id, $user);
        $this
          ->addLDAPUserToLDAPArraysFromAttributes($user, $sid, $dn, $attributes, $server_properties['ldap_type'], $server_properties['user_attr']);
      }
    }
    foreach ($this->csvTables['groups'] as $guid => $group) {
      $dn = 'cn=' . $group['cn'] . ',' . $this->csvTables['conf'][$test_ldap_id]['groupbasedn'];
      $this->csvTables['groups'][$guid]['dn'] = $dn;
      $attributes = array(
        'cn' => array(
          0 => $group['cn'],
          'count' => 1,
        ),
        'gid' => array(
          0 => $group['gid'],
          'count' => 1,
        ),
        'guid' => array(
          0 => $guid,
          'count' => 1,
        ),
      );
      if ($server_properties['groupMembershipsAttr']) {
        $membershipAttr = $server_properties['groupMembershipsAttr'];
        foreach ($this->csvTables['memberships'] as $membership_id => $membership) {
          if ($membership['gid'] == $group['gid']) {
            $member_guid = $membership['member_guid'];
            if (isset($this->csvTables['users'][$member_guid])) {
              $member = $this->csvTables['users'][$member_guid];
            }
            elseif (isset($this->csvTables['groups'][$member_guid])) {
              $member = $this->csvTables['groups'][$member_guid];
            }
            if ($server_properties['groupMembershipsAttrMatchingUserAttr'] == 'dn') {
              $attributes[$server_properties['groupMembershipsAttr']][] = $member['dn'];
            }
            else {
              $attributes[$server_properties['groupMembershipsAttr']][] = $member['attr'][$membershipAttr][0];
            }
          }
        }
        $attributes[$membershipAttr]['count'] = count($attributes[$membershipAttr]);
      }

      // need to figure out if memberOf type attribute is desired and populate it
      $this->data['ldap_servers_by_guid'][$sid][$group['guid']]['attr'] = $attributes;
      $this->data['ldap_servers_by_guid'][$sid][$group['guid']]['dn'] = $dn;
      $this->data['ldap_servers'][$sid]['groups'][$dn]['attr'] = $attributes;
      $this->ldapData['ldap_servers'][$sid][$dn] = $attributes;
    }
    if ($server_properties['groupUserMembershipsAttrExists']) {
      $member_attr = $server_properties['groupUserMembershipsAttr'];
      foreach ($this->csvTables['memberships'] as $gid => $membership) {
        $group_dn = $this->data['ldap_servers_by_guid'][$sid][$membership['group_guid']]['dn'];
        $user_dn = $this->data['ldap_servers_by_guid'][$sid][$membership['member_guid']]['dn'];
        $this->ldapData['ldap_servers'][$sid][$user_dn][$member_attr][] = $group_dn;
        if (isset($this->ldapData['ldap_servers'][$sid][$user_dn][$member_attr]['count'])) {
          unset($this->ldapData['ldap_servers'][$sid][$user_dn][$member_attr]['count']);
        }
        $this->ldapData['ldap_servers'][$sid][$user_dn][$member_attr]['count'] = count($this->ldapData['ldap_servers'][$sid][$user_dn][$member_attr]);
      }
    }
    $this->data['ldap_servers'][$sid]['ldap'] = $this->ldapData['ldap_servers'][$sid];
    $this->data['ldap_servers'][$sid]['csv'] = $this->csvTables;
    variable_set('ldap_test_server__' . $sid, $this->data['ldap_servers'][$sid]);
    $current_sids = variable_get('ldap_test_servers', array());
    $current_sids[] = $sid;
    variable_set('ldap_test_servers', array_unique($current_sids));
  }
  public function generateUserLDAPAttributes($test_ldap_id, $user) {
    $attributes = array(
      'cn' => array(
        0 => $user['cn'],
        'count' => 1,
      ),
      'mail' => array(
        0 => $user['cn'] . '@' . $this->csvTables['conf'][$test_ldap_id]['mailhostname'],
        'count' => 1,
      ),
      'uid' => array(
        0 => $user['uid'],
        'count' => 1,
      ),
      'guid' => array(
        0 => $user['guid'],
        'count' => 1,
      ),
      'sn' => array(
        0 => $user['lname'],
        'count' => 1,
      ),
      'givenname' => array(
        0 => $user['fname'],
        'count' => 1,
      ),
      'house' => array(
        0 => $user['house'],
        'count' => 1,
      ),
      'department' => array(
        0 => $user['department'],
        'count' => 1,
      ),
      'faculty' => array(
        0 => (int) (bool) $user['faculty'],
        'count' => 1,
      ),
      'staff' => array(
        0 => (int) (bool) $user['staff'],
        'count' => 1,
      ),
      'student' => array(
        0 => (int) (bool) $user['student'],
        'count' => 1,
      ),
      'gpa' => array(
        0 => $user['gpa'],
        'count' => 1,
      ),
      'probation' => array(
        0 => (int) (bool) $user['probation'],
        'count' => 1,
      ),
      'password' => array(
        0 => 'goodpwd',
        'count' => 1,
      ),
    );
    return $attributes;
  }
  public function addLDAPUserToLDAPArraysFromAttributes($user, $sid, $dn, $attributes, $ldap_type, $user_attr) {
    if ($ldap_type == 'activedirectory') {
      $attributes[$user_attr] = array(
        0 => $user['cn'],
        'count' => 1,
      );
      $attributes['distinguishedname'] = array(
        0 => $dn,
        'count' => 1,
      );
    }
    elseif ($ldap_type == 'openldap') {
    }
    $this->data['ldap_servers'][$sid]['users'][$dn]['attr'] = $attributes;
    $this->data['ldap_servers_by_guid'][$sid][$user['guid']]['attr'] = $attributes;
    $this->data['ldap_servers_by_guid'][$sid][$user['guid']]['dn'] = $dn;
    $this->ldapData['ldap_servers'][$sid][$dn] = $attributes;
    $this->ldapData['ldap_servers'][$sid][$dn]['count'] = count($attributes);
  }
  public function getCsvLdapData($test_ldap_id) {
    foreach (array(
      'groups',
      'users',
      'memberships',
      'conf',
    ) as $type) {
      $path = drupal_get_path('module', 'ldap_test') . '/test_ldap/' . $test_ldap_id . '/' . $type . '.csv';
      $this->csvTables[$type] = $this
        ->parseCsv($path);
    }
  }
  public function parseCsv($filepath) {
    $row = 1;
    $table = array();
    if (($handle = fopen($filepath, "r")) !== FALSE) {
      while (($data = fgetcsv($handle, 1000, ",")) !== FALSE) {
        if (count($data) > 1) {
          $table[] = $data;
        }
      }
      fclose($handle);
    }
    $table_associative = array();
    $headings = array_shift($table);
    foreach ($table as $i => $row) {
      $row_id = $row[0];
      foreach ($row as $j => $item) {
        $table_associative[$row_id][$headings[$j]] = $item;
      }
    }
    return $table_associative;
  }

}

Members