<?php

/**
 * @file
 * Ubercart product kit tests
 */

class UbercartProductKitTestCase extends UbercartTestHelper {

  public static function getInfo() {
    return array(
      'name' => 'Product kits',
      'description' => 'Ensure that the product kit functions properly.',
      'group' => 'Ubercart',
    );
  }

  /**
   * Overrides DrupalWebTestCase::setUp().
   */
  public function setUp() {
    parent::setUp(array('uc_product_kit'), array('create product_kit content', 'edit any product_kit content'));
  }

  public function testProductKitNodeForm() {
    $this->drupalLogin($this->adminUser);

    // Allow the default quantity to be set.
    variable_set('uc_product_add_to_cart_qty', TRUE);

    // Create some test products.
    $products = array();
    for ($i = 0; $i < 3; $i++) {
      $products[$i] = $this->createProduct();
    }

    // Test the product kit fields.
    $this->drupalGet('node/add/product-kit');
    foreach (array('mutable', 'products[]', 'ordering') as $field) {
      $this->assertFieldByName($field);
    }

    // Test creation of a basic kit.
    $body_key = 'body[und][0][value]';
    $edit = array(
      'title' => $this->randomName(32),
      $body_key => $this->randomName(64),
      'products[]' => array(
        $products[0]->nid,
        $products[1]->nid,
        $products[2]->nid,
      ),
      'default_qty' => mt_rand(2, 100),
    );
    $this->drupalPost('node/add/product-kit', $edit, 'Save');
    $this->assertText(t('Product kit @title has been created.', array('@title' => $edit['title'])));
    $this->assertText($edit[$body_key], 'Product kit body found.');
    $this->assertText('1 × ' . $products[0]->title, 'Product 1 title found.');
    $this->assertText('1 × ' . $products[1]->title, 'Product 2 title found.');
    $this->assertText('1 × ' . $products[2]->title, 'Product 3 title found.');
    $total = $products[0]->sell_price + $products[1]->sell_price + $products[2]->sell_price;
    $this->assertText(uc_currency_format($total), 'Product kit total found.');
    $this->assertFieldByName('qty', $edit['default_qty']);
  }

  public function testProductKitDiscounts() {
    $this->drupalLogin($this->adminUser);

    // Create some test products and a kit.
    $products = array();
    for ($i = 0; $i < 3; $i++) {
      $products[$i] = $this->createProduct();
    }
    $kit = $this->drupalCreateNode(array(
      'type' => 'product_kit',
      'title' => $this->randomName(32),
      'products' => array(
        $products[0]->nid,
        $products[1]->nid,
        $products[2]->nid,
      ),
      'mutable' => UC_PRODUCT_KIT_UNMUTABLE_NO_LIST,
      'default_qty' => 1,
      'ordering' => 0,
    ));

    // Test the product kit extra fields available to configure discounts.
    $this->drupalGet('node/' . $kit->nid . '/edit');
    $this->assertFieldByName('kit_total');
    foreach ($products as $product) {
      $this->assertFieldByName('items[' . $product->nid . '][qty]');
      $this->assertFieldByName('items[' . $product->nid . '][ordering]');
      $this->assertFieldByName('items[' . $product->nid . '][discount]');
    }

    // Set some discounts.
    $discounts = array(
      mt_rand(-100, 100),
      mt_rand(-100, 100),
      mt_rand(-100, 100),
    );
    $edit = array(
      'items[' . $products[0]->nid . '][discount]' => $discounts[0],
      'items[' . $products[1]->nid . '][discount]' => $discounts[1],
      'items[' . $products[2]->nid . '][discount]' => $discounts[2],
    );
    $this->drupalPost('node/' . $kit->nid . '/edit', $edit, 'Save');

    // Check the discounted total.
    $total = $products[0]->sell_price + $products[1]->sell_price + $products[2]->sell_price;
    $total += array_sum($discounts);
    $this->assertText(uc_currency_format($total), 'Discounted product kit total found.');

    // Check the discounts on the edit page.
    $this->drupalGet('node/' . $kit->nid . '/edit');
    $this->assertText('Currently, the total sell price is ' . uc_currency_format($total), 'Discounted product kit total found.');
    $this->assertFieldByName('items[' . $products[0]->nid . '][discount]', $discounts[0]);
    $this->assertFieldByName('items[' . $products[1]->nid . '][discount]', $discounts[1]);
    $this->assertFieldByName('items[' . $products[2]->nid . '][discount]', $discounts[2]);

    // Set the kit total.
    $total = 2 * ($products[0]->sell_price + $products[1]->sell_price + $products[2]->sell_price);
    $this->drupalPost('node/' . $kit->nid . '/edit', array('kit_total' => $total), 'Save');

    // Check the fixed total.
    $this->assertText(uc_currency_format($total), 'Fixed product kit total found.');

    // Check the discounts on the edit page.
    $this->drupalGet('node/' . $kit->nid . '/edit');
    $this->assertFieldByName('kit_total', $total);
    $this->assertFieldByName('items[' . $products[0]->nid . '][discount]', $products[0]->sell_price);
    $this->assertFieldByName('items[' . $products[1]->nid . '][discount]', $products[1]->sell_price);
    $this->assertFieldByName('items[' . $products[2]->nid . '][discount]', $products[2]->sell_price);
  }

  public function testProductKitMutability() {
    $this->drupalLogin($this->adminUser);

    // Create some test products and prepare a kit.
    $products = array();
    for ($i = 0; $i < 3; $i++) {
      $products[$i] = $this->createProduct();
    }
    $kit_data = array(
      'type' => 'product_kit',
      'title' => $this->randomName(32),
      'products' => array(
        $products[0]->nid,
        $products[1]->nid,
        $products[2]->nid,
      ),
      'default_qty' => 1,
      'ordering' => 0,
    );

    // Test kits with no listing.
    $kit_data['mutable'] = UC_PRODUCT_KIT_UNMUTABLE_NO_LIST;
    $kit = $this->drupalCreateNode($kit_data);
    $this->drupalGet('node/' . $kit->nid);
    $this->assertText($kit->title, 'Product kit title found.');
    $this->assertNoText($products[0]->title, 'Product 1 title not shown.');
    $this->assertNoText($products[1]->title, 'Product 2 title not shown.');
    $this->assertNoText($products[2]->title, 'Product 3 title not shown.');

    $this->drupalPost('node/' . $kit->nid, array(), 'Add to cart');
    $this->drupalGet('cart');
    $this->assertText($kit->title, 'Product kit title found.');
    $this->assertNoText($products[0]->title, 'Product 1 title not shown.');
    $this->assertNoText($products[1]->title, 'Product 2 title not shown.');
    $this->assertNoText($products[2]->title, 'Product 3 title not shown.');

    $total = $products[0]->sell_price + $products[1]->sell_price + $products[2]->sell_price;
    $this->assertText('Subtotal: ' . uc_currency_format($total), 'Product kit total found.');

    $qty = mt_rand(2, 10);
    $this->drupalPost(NULL, array('items[2][qty]' => $qty), 'Update cart');
    $this->assertText('Subtotal: ' . uc_currency_format($total * $qty), 'Updated product kit total found.');

    $this->drupalPost(NULL, array(), 'Remove');
    $this->assertText('There are no products in your shopping cart.');

    // Test kits with listing.
    $kit_data['mutable'] = UC_PRODUCT_KIT_UNMUTABLE_WITH_LIST;
    $kit = $this->drupalCreateNode($kit_data);
    $this->drupalGet('node/' . $kit->nid);
    $this->assertText($kit->title, 'Product kit title found.');
    $this->assertText($products[0]->title, 'Product 1 title shown.');
    $this->assertText($products[1]->title, 'Product 2 title shown.');
    $this->assertText($products[2]->title, 'Product 3 title shown.');

    $this->drupalPost('node/' . $kit->nid, array(), 'Add to cart');
    $this->drupalGet('cart');
    $this->assertText($kit->title, 'Product kit title found.');
    $this->assertText($products[0]->title, 'Product 1 title shown.');
    $this->assertText($products[1]->title, 'Product 2 title shown.');
    $this->assertText($products[2]->title, 'Product 3 title shown.');

    $total = $products[0]->sell_price + $products[1]->sell_price + $products[2]->sell_price;
    $this->assertText('Subtotal: ' . uc_currency_format($total), 'Product kit total found.');

    $qty = mt_rand(2, 10);
    $this->drupalPost(NULL, array('items[2][qty]' => $qty), 'Update cart');
    $this->assertText('Subtotal: ' . uc_currency_format($total * $qty), 'Updated product kit total found.');

    $this->drupalPost(NULL, array(), 'Remove');
    $this->assertText('There are no products in your shopping cart.');

    // Test mutable kits.
    $kit_data['mutable'] = UC_PRODUCT_KIT_MUTABLE;
    $kit = $this->drupalCreateNode($kit_data);
    $this->drupalGet('node/' . $kit->nid);
    $this->assertText($kit->title, 'Product kit title found.');
    $this->assertText($products[0]->title, 'Product 1 title shown.');
    $this->assertText($products[1]->title, 'Product 2 title shown.');
    $this->assertText($products[2]->title, 'Product 3 title shown.');

    $this->drupalPost('node/' . $kit->nid, array(), 'Add to cart');
    $this->drupalGet('cart');
    $this->assertNoText($kit->title, 'Product kit title not shown.');
    $this->assertText($products[0]->title, 'Product 1 title shown.');
    $this->assertText($products[1]->title, 'Product 2 title shown.');
    $this->assertText($products[2]->title, 'Product 3 title shown.');

    $total = $products[0]->sell_price + $products[1]->sell_price + $products[2]->sell_price;
    $this->assertText('Subtotal: ' . uc_currency_format($total), 'Product kit total found.');

    $qty = array(mt_rand(2, 10), mt_rand(2, 10), mt_rand(2, 10));
    $edit = array(
      'items[0][qty]' => $qty[0],
      'items[1][qty]' => $qty[1],
      'items[2][qty]' => $qty[2],
    );
    $this->drupalPost(NULL, $edit, 'Update cart');
    $total = $products[0]->sell_price * $qty[0];
    $total += $products[1]->sell_price * $qty[1];
    $total += $products[2]->sell_price * $qty[2];
    $this->assertText('Subtotal: ' . uc_currency_format($total), 'Updated product kit total found.');

    $this->drupalPost(NULL, array(), 'Remove');
    $this->drupalPost(NULL, array(), 'Remove');
    $this->drupalPost(NULL, array(), 'Remove');
    $this->assertText('There are no products in your shopping cart.');
  }

}