<?php
require_once 'CiviTest/CiviUnitTestCase.php';
require_once 'CiviTest/Contact.php';

/**
 * Class CRM_Contact_BAO_ContactType_RelationshipTest
 */
class CRM_Contact_BAO_ContactType_RelationshipTest extends CiviUnitTestCase {

  public function setUp() {
    parent::setUp();

    //create contact subtypes
    $params = array(
      'label' => 'indivi_student',
      'name' => 'indivi_student',
      // Individual
      'parent_id' => 1,
      'is_active' => 1,
    );
    $result = CRM_Contact_BAO_ContactType::add($params);
    $this->student = $params['name'];

    $params = array(
      'label' => 'indivi_parent',
      'name' => 'indivi_parent',
      // Individual
      'parent_id' => 1,
      'is_active' => 1,
    );
    $result = CRM_Contact_BAO_ContactType::add($params);
    $this->parent = $params['name'];

    $params = array(
      'label' => 'org_sponsor',
      'name' => 'org_sponsor',
      // Organization
      'parent_id' => 3,
      'is_active' => 1,
    );
    $result = CRM_Contact_BAO_ContactType::add($params);
    $this->sponsor = $params['name'];

    //create contacts
    $params = array(
      'first_name' => 'Anne',
      'last_name' => 'Grant',
      'contact_type' => 'Individual',
    );
    $this->individual = Contact::create($params);

    $params = array(
      'first_name' => 'Bill',
      'last_name' => 'Adams',
      'contact_type' => 'Individual',
      'contact_sub_type' => $this->student,
    );
    $this->indivi_student = Contact::create($params);

    $params = array(
      'first_name' => 'Alen',
      'last_name' => 'Adams',
      'contact_type' => 'Individual',
      'contact_sub_type' => $this->parent,
    );
    $this->indivi_parent = Contact::create($params);

    $params = array(
      'organization_name' => 'Compumentor',
      'contact_type' => 'Organization',
    );
    $this->organization = Contact::create($params);

    $params = array(
      'organization_name' => 'Conservation Corp',
      'contact_type' => 'Organization',
      'contact_sub_type' => $this->sponsor,
    );
    $this->organization_sponsor = Contact::create($params);
  }

  public function tearDown() {
    $this->quickCleanup(array('civicrm_contact'));

    $query = "
DELETE FROM civicrm_contact_type
      WHERE name IN ('{$this->student}','{$this->parent}','{$this->sponsor}');
    ";
    CRM_Core_DAO::executeQuery($query);
  }

  /**
   * Methods create relationshipType with valid data.
   * success expected
   */
  public function testRelationshipTypeAddIndiviParent() {
    //check Individual to Parent RelationshipType
    $params = array(
      'name_a_b' => 'indivToparent',
      'name_b_a' => 'parentToindiv',
      'contact_type_a' => 'Individual',
      'contact_type_b' => 'Individual',
      'contact_sub_type_b' => $this->parent,
    );
    $ids = array();
    $result = CRM_Contact_BAO_RelationshipType::add($params, $ids);
    $this->assertEquals($result->name_a_b, 'indivToparent');
    $this->assertEquals($result->contact_type_a, 'Individual');
    $this->assertEquals($result->contact_type_b, 'Individual');
    $this->assertEquals($result->contact_sub_type_b, $this->parent);
    $this->relationshipTypeDelete($result->id);
  }

  public function testRelationshipTypeAddSponcorIndivi() {
    //check Sponcor to Individual RelationshipType
    $params = array(
      'name_a_b' => 'SponsorToIndiv',
      'name_b_a' => 'IndivToSponsor',
      'contact_type_a' => 'Organization',
      'contact_sub_type_a' => $this->sponsor,
      'contact_type_b' => 'Individual',
    );
    $ids = array();
    $result = CRM_Contact_BAO_RelationshipType::add($params, $ids);
    $this->assertEquals($result->name_a_b, 'SponsorToIndiv');
    $this->assertEquals($result->contact_type_a, 'Organization');
    $this->assertEquals($result->contact_sub_type_a, $this->sponsor);
    $this->assertEquals($result->contact_type_b, 'Individual');
    $this->relationshipTypeDelete($result->id);
  }

  public function testRelationshipTypeAddStudentSponcor() {
    //check Student to Sponcer RelationshipType
    $params = array(
      'name_a_b' => 'StudentToSponser',
      'name_b_a' => 'SponsorToStudent',
      'contact_type_a' => 'Individual',
      'contact_sub_type_a' => $this->student,
      'contact_type_b' => 'Organization',
      'contact_sub_type_b' => $this->sponsor,
    );
    $ids = array();
    $result = CRM_Contact_BAO_RelationshipType::add($params, $ids);
    $this->assertEquals($result->name_a_b, 'StudentToSponser');
    $this->assertEquals($result->contact_type_a, 'Individual');
    $this->assertEquals($result->contact_sub_type_a, $this->student);
    $this->assertEquals($result->contact_type_b, 'Organization');
    $this->assertEquals($result->contact_sub_type_b, $this->sponsor);
    $this->relationshipTypeDelete($result->id);
  }

  /**
   * Methods create relationshipe within same contact type with invalid Relationships.
   */
  public function testRelationshipCreateInvalidWithinSameType() {
    //check for Individual to Parent
    $relTypeParams = array(
      'name_a_b' => 'indivToparent',
      'name_b_a' => 'parentToindiv',
      'contact_type_a' => 'Individual',
      'contact_type_b' => 'Individual',
      'contact_sub_type_b' => $this->parent,
    );
    $relTypeIds = array();
    $relType = CRM_Contact_BAO_RelationshipType::add($relTypeParams, $relTypeIds);
    $params = array(
      'relationship_type_id' => $relType->id . '_a_b',
      'contact_check' => array($this->indivi_student => 1),
    );
    $ids = array('contact' => $this->individual);

    list($valid, $invalid, $duplicate, $saved, $relationshipIds) = CRM_Contact_BAO_Relationship::legacyCreateMultiple($params, $ids);

    $this->assertEquals($invalid, 1, 'In line ' . __LINE__);
    $this->assertEquals(empty($relationshipIds), TRUE, 'In line ' . __LINE__);
    $this->relationshipTypeDelete($relType->id);
  }

  /**
   * Methods create relationshipe within diff contact type with invalid Relationships.
   */
  public function testRelCreateInvalidWithinDiffTypeSpocorIndivi() {
    //check for Sponcer to Individual
    $relTypeParams = array(
      'name_a_b' => 'SponsorToIndiv',
      'name_b_a' => 'IndivToSponsor',
      'contact_type_a' => 'Organization',
      'contact_sub_type_a' => $this->sponsor,
      'contact_type_b' => 'Individual',
    );
    $relTypeIds = array();
    $relType = CRM_Contact_BAO_RelationshipType::add($relTypeParams, $relTypeIds);
    $params = array(
      'relationship_type_id' => $relType->id . '_a_b',
      'contact_check' => array($this->individual => 1),
    );
    $ids = array('contact' => $this->indivi_parent);

    list($valid, $invalid, $duplicate, $saved, $relationshipIds) = CRM_Contact_BAO_Relationship::legacyCreateMultiple($params, $ids);

    $this->assertEquals($invalid, 1, 'In line ' . __LINE__);
    $this->assertEquals(empty($relationshipIds), TRUE, 'In line ' . __LINE__);
    $this->relationshipTypeDelete($relType->id);
  }

  public function testRelCreateInvalidWithinDiffTypeStudentSponcor() {
    //check for Student to Sponcer
    $relTypeParams = array(
      'name_a_b' => 'StudentToSponser',
      'name_b_a' => 'SponsorToStudent',
      'contact_type_a' => 'Individual',
      'contact_sub_type_a' => $this->student,
      'contact_type_b' => 'Organization',
      'contact_sub_type_b' => 'Sponser',
    );
    $relTypeIds = array();
    $relType = CRM_Contact_BAO_RelationshipType::add($relTypeParams, $relTypeIds);
    $params = array(
      'relationship_type_id' => $relType->id . '_a_b',
      'contact_check' => array($this->individual => 1),
    );
    $ids = array('contact' => $this->indivi_parent);

    list($valid, $invalid, $duplicate, $saved, $relationshipIds) = CRM_Contact_BAO_Relationship::legacyCreateMultiple($params, $ids);

    $this->assertEquals($invalid, 1, 'In line ' . __LINE__);
    $this->assertEquals(empty($relationshipIds), TRUE, 'In line ' . __LINE__);
    $this->relationshipTypeDelete($relType->id);
  }

  /**
   * Methods create relationshipe within same contact type with valid data.
   * success expected
   */
  public function testRelationshipCreateWithinSameType() {
    //check for Individual to Parent
    $relTypeParams = array(
      'name_a_b' => 'indivToparent',
      'name_b_a' => 'parentToindiv',
      'contact_type_a' => 'Individual',
      'contact_type_b' => 'Individual',
      'contact_sub_type_b' => $this->parent,
    );
    $relTypeIds = array();
    $relType = CRM_Contact_BAO_RelationshipType::add($relTypeParams, $relTypeIds);
    $params = array(
      'relationship_type_id' => $relType->id . '_a_b',
      'is_active' => 1,
      'contact_check' => array($this->indivi_parent => $this->indivi_parent),
    );
    $ids = array('contact' => $this->individual);
    list($valid, $invalid, $duplicate, $saved, $relationshipIds) = CRM_Contact_BAO_Relationship::legacyCreateMultiple($params, $ids);

    $this->assertEquals($valid, 1, 'In line ' . __LINE__);
    $this->assertEquals(empty($relationshipIds), FALSE, 'In line ' . __LINE__);
    $this->relationshipTypeDelete($relType->id);
  }

  /**
   * Methods create relationshipe within different contact type with valid data.
   * success expected
   */
  public function testRelCreateWithinDiffTypeSponsorIndivi() {
    //check for Sponcer to Individual
    $relTypeParams = array(
      'name_a_b' => 'SponsorToIndiv',
      'name_b_a' => 'IndivToSponsor',
      'contact_type_a' => 'Organization',
      'contact_sub_type_a' => $this->sponsor,
      'contact_type_b' => 'Individual',
    );
    $relTypeIds = array();
    $relType = CRM_Contact_BAO_RelationshipType::add($relTypeParams, $relTypeIds);
    $params = array(
      'relationship_type_id' => $relType->id . '_a_b',
      'is_active' => 1,
      'contact_check' => array($this->indivi_student => 1),
    );
    $ids = array('contact' => $this->organization_sponsor);
    list($valid, $invalid, $duplicate, $saved, $relationshipIds) = CRM_Contact_BAO_Relationship::legacyCreateMultiple($params, $ids);

    $this->assertEquals($valid, 1, 'In line ' . __LINE__);
    $this->assertEquals(empty($relationshipIds), FALSE, 'In line ' . __LINE__);
    $this->relationshipTypeDelete($relType->id);
  }

  public function testRelCreateWithinDiffTypeStudentSponsor() {
    //check for Student to Sponcer
    $relTypeParams = array(
      'name_a_b' => 'StudentToSponsor',
      'name_b_a' => 'SponsorToStudent',
      'contact_type_a' => 'Individual',
      'contact_sub_type_a' => $this->student,
      'contact_type_b' => 'Organization',
      'contact_sub_type_b' => $this->sponsor,
    );
    $relTypeIds = array();
    $relType = CRM_Contact_BAO_RelationshipType::add($relTypeParams, $relTypeIds);
    $params = array(
      'relationship_type_id' => $relType->id . '_a_b',
      'is_active' => 1,
      'contact_check' => array($this->organization_sponsor => 1),
    );
    $ids = array('contact' => $this->indivi_student);
    list($valid, $invalid, $duplicate, $saved, $relationshipIds) = CRM_Contact_BAO_Relationship::legacyCreateMultiple($params, $ids);

    $this->assertEquals($valid, 1, 'In line ' . __LINE__);
    $this->assertEquals(empty($relationshipIds), FALSE, 'In line ' . __LINE__);
    $this->relationshipTypeDelete($relType->id);
  }

}