<?php
/*
 *  File for the TestMailing class
 *
 *  (PHP 5)
 *
 *   @package   CiviCRM
 *
 *   This file is part of CiviCRM
 *
 *   CiviCRM is free software; you can redistribute it and/or
 *   modify it under the terms of the GNU Affero General Public License
 *   as published by the Free Software Foundation; either version 3 of
 *   the License, or (at your option) any later version.
 *
 *   CiviCRM is distributed in the hope that it will be useful,
 *   but WITHOUT ANY WARRANTY; without even the implied warranty of
 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *   GNU Affero General Public License for more details.
 *
 *   You should have received a copy of the GNU Affero General Public
 *   License along with this program.  If not, see
 *   <http://www.gnu.org/licenses/>.
 */

/**
 *  Test APIv3 civicrm_mailingab_* functions
 *
 * @package   CiviCRM
 */
class api_v3_MailingABTest extends CiviUnitTestCase {
  protected $_mailingID_A;
  protected $_mailingID_B;
  protected $_mailingID_C;
  protected $_params;
  protected $_apiversion = 3;
  protected $_entity = 'MailingAB';
  protected $_groupID;

  public function setUp() {
    parent::setUp();
    $this->useTransaction(TRUE);
    $this->createLoggedInUser();
    $this->_mailingID_A = $this->createMailing();
    $this->_mailingID_B = $this->createMailing();
    $this->_mailingID_C = $this->createMailing();
    $this->_groupID = $this->groupCreate();

    $this->_params = array(
      'mailing_id_a' => $this->_mailingID_A,
      'mailing_id_b' => $this->_mailingID_B,
      'mailing_id_c' => $this->_mailingID_C,
      'testing_criteria' => 'subject',
      'winner_criteria' => 'open',
      'declare_winning_time' => '+2 days',
      'group_percentage' => 10,
    );
  }

  /**
   * Test civicrm_mailing_create.
   */
  public function testMailingABCreateSuccess() {
    $result = $this->callAPIAndDocument($this->_entity, 'create', $this->_params, __FUNCTION__, __FILE__);
    $this->assertTrue(is_numeric($result['id']), "In line " . __LINE__);
    $this->assertEquals($this->_params['group_percentage'], $result['values'][$result['id']]['group_percentage']);
  }

  /**
   * Test civicrm_mailing_delete.
   */
  public function testMailerDeleteSuccess() {
    $result = $this->callAPISuccess($this->_entity, 'create', $this->_params);

    $this->assertDBQuery(1, "SELECT count(*) FROM civicrm_mailing_abtest WHERE id = %1", array(
      1 => array($result['id'], 'Integer'),
    ));
    $this->assertDBQuery(3, "SELECT count(*) FROM civicrm_mailing WHERE id IN (%1,%2,%3)", array(
      1 => array($this->_mailingID_A, 'Integer'),
      2 => array($this->_mailingID_B, 'Integer'),
      3 => array($this->_mailingID_C, 'Integer'),
    ));

    $this->callAPISuccess($this->_entity, 'delete', array('id' => $result['id']));

    $this->assertDBQuery(0, "SELECT count(*) FROM civicrm_mailing_abtest WHERE id = %1", array(
      1 => array($result['id'], 'Integer'),
    ));
    $this->assertDBQuery(0, "SELECT count(*) FROM civicrm_mailing WHERE id IN (%1,%2,%3)", array(
      1 => array($this->_mailingID_A, 'Integer'),
      2 => array($this->_mailingID_B, 'Integer'),
      3 => array($this->_mailingID_C, 'Integer'),
    ));
  }

  /**
   * @return array
   */
  public function groupPctProvider() {
    $cases = array(); // array(int $totalSize, int $groupPct, int $expectedCountA, $expectedCountB, $expectedCountC)
    $cases[] = array(400, 7, 28, 28, 344);
    $cases[] = array(100, 10, 10, 10, 80);
    $cases[] = array(50, 20, 10, 10, 30);
    $cases[] = array(50, 10, 5, 5, 40);
    $cases[] = array(3, 10, 1, 1, 1);
    $cases[] = array(2, 10, 1, 1, 0);
    $cases[] = array(1, 10, 1, 0, 0);
    return $cases;
  }

  /**
   * Create a test and ensure that all three mailings (A/B/C) wind up with the correct
   * number of recipients.
   *
   * @param $totalGroupContacts
   * @param $groupPct
   * @param $expectedCountA
   * @param $expectedCountB
   * @param $expectedCountC
   * @dataProvider groupPctProvider
   */
  public function testDistribution($totalGroupContacts, $groupPct, $expectedCountA, $expectedCountB, $expectedCountC) {

    $result = $this->groupContactCreate($this->_groupID, $totalGroupContacts);
    $this->assertEquals($totalGroupContacts, $result['added'], "in line " . __LINE__);

    $params = $this->_params;
    $params['group_percentage'] = $groupPct;
    $result = $this->callAPISuccess($this->_entity, 'create', $params);

    $this->callAPISuccess('Mailing', 'create', array(
      'id' => $this->_mailingID_A,
      'groups' => array('include' => array($this->_groupID)),
    ));
    $this->assertJobCounts(0, 0, 0);

    $this->callAPISuccess('MailingAB', 'submit', array(
      'id' => $result['id'],
      'status' => 'Testing',
      'scheduled_date' => date('YmdHis'),
      'approval_date' => date('YmdHis'),
    ));
    $this->assertRecipientCounts($expectedCountA, $expectedCountB, $expectedCountC);
    $this->assertJobCounts(1, 1, 0);

    $this->callAPISuccess('MailingAB', 'submit', array(
      'id' => $result['id'],
      'status' => 'Final',
      'scheduled_date' => date('YmdHis'),
      'approval_date' => date('YmdHis'),
    ));
    $this->assertRecipientCounts($expectedCountA, $expectedCountB, $expectedCountC);
    $this->assertJobCounts(1, 1, 1);
  }

  /**
   * @param $expectedCountA
   * @param $expectedCountB
   * @param $expectedCountC
   */
  protected function assertRecipientCounts($expectedCountA, $expectedCountB, $expectedCountC) {
    $countA = $this->callAPISuccess('MailingRecipients', 'getcount', array('mailing_id' => $this->_mailingID_A));
    $countB = $this->callAPISuccess('MailingRecipients', 'getcount', array('mailing_id' => $this->_mailingID_B));
    $countC = $this->callAPISuccess('MailingRecipients', 'getcount', array('mailing_id' => $this->_mailingID_C));
    $this->assertEquals($expectedCountA, $countA, "check mailing recipients A in line " . __LINE__);
    $this->assertEquals($expectedCountB, $countB, "check mailing recipients B in line " . __LINE__);
    $this->assertEquals($expectedCountC, $countC, "check mailing recipients C in line " . __LINE__);
  }

  /**
   * @param $expectedA
   * @param $expectedB
   * @param $expectedC
   */
  protected function assertJobCounts($expectedA, $expectedB, $expectedC) {
    $this->assertDBQuery($expectedA, 'SELECT count(*) FROM civicrm_mailing_job WHERE mailing_id = %1', array(
      1 => array(
        $this->_mailingID_A,
        'Integer',
      ),
    ));
    $this->assertDBQuery($expectedB, 'SELECT count(*) FROM civicrm_mailing_job WHERE mailing_id = %1', array(
      1 => array(
        $this->_mailingID_B,
        'Integer',
      ),
    ));
    $this->assertDBQuery($expectedC, 'SELECT count(*) FROM civicrm_mailing_job WHERE mailing_id = %1', array(
      1 => array(
        $this->_mailingID_C,
        'Integer',
      ),
    ));
  }

}