<?php
// $Id: privatemsg_roles.test,v 1.1.2.6 2010-10-06 19:41:08 berdir Exp $

/**
 * @file
 * This file contains tests for the privatemsg roles module
 */

/**
 * Test cases for the privatemsg_roles module.
 */
class PrivatemsgRolesTestCase extends DrupalWebTestCase {
  public static function getInfo() {
    return array(
      'name' => 'Privatemsg Roles functionality',
      'description' => 'Tests sending messages to roles',
      'group' => 'Privatemsg',
    );
  }

  function setUp() {
    parent::setUp('privatemsg', 'privatemsg_roles', 'privatemsg_filter', 'pm_block_user');
  }

  function testSendMessagetoRoleAPI() {
    $admin = $this->drupalCreateUser(array('read privatemsg', 'write privatemsg', 'write privatemsg to roles'));
    $user1 = $this->drupalCreateUser(array('read privatemsg', 'write privatemsg'));
    $user2 = $this->drupalCreateUser();

    // Add role of user 1 to user 2;
    $edit = array('roles' => $user2->roles + $user1->roles);
    user_save($user2, $edit);

    $recipient = user_role_load($user1->roles[5]);
    $recipient->recipient = $recipient->rid;
    $recipient->type = 'role';
    privatemsg_new_thread(array($recipient), $subject = $this->randomName(10), $body = $this->randomName(50), array('author' => $admin));

    $this->drupalLogin($user2);
    $this->drupalGet('messages');
    $this->assertRaw($subject . '</a> <span class="marker">new</span>', t('Message is displayed as new'));

    $this->clickLink($subject);
    $this->assertText($body, t('Thread starter body displayed.'));
  }

  function testSendMessagetoRoleCron() {
    $admin = $this->drupalCreateUser(array('read privatemsg', 'write privatemsg', 'write privatemsg to roles'));
    $user1 = $this->drupalCreateUser(array('read privatemsg', 'write privatemsg'));

    variable_set('privatemsg_recipient_small_threshold', 20);
    variable_set('privatemgs_cron_recipient_per_run', 20);

    // Create 26 users (more than are allowed to be process directly).
    $users = array();
    for ($i = 0; $i < 25; $i++) {
      $users[$i] = $this->drupalCreateUser();
      // Add role of user 1 to user 2;
      $edit = array('roles' => $users[$i]->roles + $user1->roles);
      user_save($users[$i], $edit);
    }

    $recipient = user_role_load($user1->roles[5]);
    $recipient->recipient = $recipient->rid;
    $recipient->type = 'role';
    privatemsg_new_thread(array($recipient), $subject = $this->randomName(10), $body = $this->randomName(50), array('author' => $admin));

    // Run cron.
    privatemsg_cron();

    // Test a few recipients to see if they recieved the message.
    foreach (array(0, 5, 18) as $uid) {
      $this->drupalLogin($users[$uid]);
      $this->drupalGet('messages');
      $this->assertRaw($subject . '</a> <span class="marker">new</span>', t('Message is displayed as new'));

      $this->clickLink($subject);
      $this->assertText($body, t('Thread starter body displayed.'));
    }

    // Make sure that user 20 has not yet recieved the message.
    $this->drupalLogin($users[20]);
    $this->drupalGet('messages');
    $this->assertNoText($subject, t('Message is not yet displayed for this user'));

    // Run cron again.
    privatemsg_cron();

    // Test that the remaining recipients do now see the message too.
    foreach (array(20, 24) as $uid) {
      $this->drupalLogin($users[$uid]);
      $this->drupalGet('messages');
      $this->assertRaw($subject . '</a> <span class="marker">new</span>', t('Message is displayed as new'));

      $this->clickLink($subject);
      $this->assertText($body, t('Thread starter body displayed.'));
    }
  }

  function testSendMessagetoRoleBatch() {
    $admin = $this->drupalCreateUser(array('read privatemsg', 'write privatemsg', 'write privatemsg to roles'));
    $user1 = $this->drupalCreateUser(array('read privatemsg', 'write privatemsg'));

    variable_set('privatemsg_recipient_small_threshold', 20);
    // Create 25 users (more than are allowed to be process directly).
    $users = array();
    for ($i = 0; $i < 25; $i++) {
      $users[$i] = $this->drupalCreateUser();
      // Add role of user 1 to user 2;
      $edit = array('roles' => $users[$i]->roles + $user1->roles);
      user_save($users[$i], $edit);
    }
    $this->drupalLogin($admin);

    // Send a message to the role shared by all users.
    $edit = array(
      'recipient' => user_role_load($user1->roles[5])->name . '[role]',
      'subject' => $this->randomName(10),
      'body[value]' => $this->randomName(50),
    );
    $this->drupalPost('messages/new', $edit, t('Send message'));
    $this->assertText(t('A message has been sent to @role (role).', array('@role' => user_role_load($user1->roles[5])->name)));

    // Test a few recipients to see if they all recieved the message.
    foreach ($users as $user) {
      $this->drupalLogin($user);
      $this->drupalGet('messages');
      $this->assertRaw($edit['subject'] . '</a> <span class="marker">new</span>', t('Message is displayed as new'));

      $this->clickLink($edit['subject']);
      $this->assertText($edit['body[value]'], t('Thread starter body displayed.'));
    }
  }

  function testPermission() {
    $user1 = $this->drupalCreateUser(array('read privatemsg', 'write privatemsg'));
    $user2 = $this->drupalCreateUser(array('read privatemsg', 'write privatemsg'));
    $this->drupalLogin($user1);

    // Send a message to the role of user 1 and 2.
    $edit = array(
      'recipient' => $user2->roles[5] . '[role]',
      'subject' => $this->randomName(10),
      'body[value]' => $this->randomName(50),
    );
    $this->drupalPost('messages/new', $edit, t('Send message'));
    $this->assertText(t('You must include at least one valid recipient.'));
  }

  function testSendMessageToRole() {
    $admin = $this->drupalCreateUser(array('read privatemsg', 'write privatemsg', 'write privatemsg to roles'));
    $user1 = $this->drupalCreateUser(array('read privatemsg', 'write privatemsg'));
    $user2 = $this->drupalCreateUser(array('view roles recipients'));

    // Add role of user 1 to user 2;
    $edit = array('roles' => $user2->roles + $user1->roles);
    user_save($user2, $edit);
    $this->drupalLogin($admin);

    // Verify autocomplete feature.
    $role_name = user_role_load($user1->roles[5])->name;
    $json = $this->drupalGet('messages/autocomplete/' . drupal_substr($role_name, 0, 2));
    $autocomplete = (array)json_decode($json);
    $this->assertEqual($autocomplete[$role_name . ', '], $role_name);

    // Access the form through a url that pre-fills the recipient field.
    $this->drupalGet('messages/new/role_5');

    // Send a message to the role of user 1 and 2.
    $edit = array(
      'subject' => $this->randomName(10),
      'body[value]' => $this->randomName(50),
    );
    $this->drupalPost(NULL, $edit, t('Send message'));
    $this->assertText(t('A message has been sent to @role (role).', array('@role' => user_role_load($user1->roles[5])->name)));

    // Log in as user1 and check that the message is listed, is marked as new
    // and can be marked as read.
    $this->drupalLogin($user1);
    $this->drupalGet('messages');
    $this->assertRaw($edit['subject'] . '</a> <span class="marker">new</span>', t('Message is displayed as new'));

    $this->clickLink($edit['subject']);
    $this->assertText($edit['body[value]'], t('Thread starter body displayed.'));

    // Make sure that the user does not see the role.
    $this->assertNoText(t('@role (role)', array('@role' => $role_name)));

    // Reply to the message, only admin should see this.
    $reply1 = array(
      'body[value]' => $this->randomName(50),
    );
    $this->drupalPost(NULL, $reply1, t('Send message'));

    $this->drupalGet('messages');
    $this->assertNoRaw($edit['subject'] . '</a> <span class="marker">new</span>', t('Message is displayed as read'));

    // Login as admin again and check if he sees the response.
    $this->drupalLogin($admin);
    $this->drupalGet('messages');
    $this->assertRaw($edit['subject'] . '</a> <span class="marker">new</span>', t('Message is displayed as new'));
    $this->clickLink($edit['subject']);

    $this->assertText($reply1['body[value]'], t('Reply1 is displayed'));

    // Reply to the message, all recipients should see this.
    $reply2 = array(
      'body[value]' => $this->randomName(50),
    );
    $this->drupalPost(NULL, $reply2, t('Send message'));

    // Log in as user2 and check that he only sees the messages from admin.
    $this->drupalLogin($user2);
    $this->drupalGet('messages');
    $this->assertRaw($edit['subject'] . '</a> <span class="marker">new</span>', t('Message is displayed as new'));
    $this->clickLink($edit['subject']);
    $this->assertText(t('@role (role)', array('@role' => $role_name)));

    $this->assertText($edit['body[value]'], t('Thread starter body is displayed'));
    $this->assertNoText($reply1['body[value]'], t('Reply1 is not displayed'));
    $this->assertText($reply2['body[value]'], t('Reply2 is displayed'));
  }

  function testSendMessageToRoleBlocked() {
    $admin = $this->drupalCreateUser(array('read privatemsg', 'write privatemsg', 'write privatemsg to roles'));
    $user1 = $this->drupalCreateUser(array('read privatemsg', 'write privatemsg'));
    $user2 = $this->drupalCreateUser();

    // Add role of user 1 to user 2;
    $edit = array('roles' => $user2->roles + $user1->roles);
    user_save($user2, $edit);

    // Block admin as user 2.
    $this->drupalLogin($user2);
    $this->drupalPost('messages/blocked', array('name' => $admin->name), t('Block user'));

    $this->drupalLogin($admin);

    // Send a message to the role of user 1 and 2.
    $edit = array(
      'recipient' => user_role_load($user1->roles[5])->name . '[role]',
      'subject' => $this->randomName(10),
      'body[value]' => $this->randomName(50),
    );
    $this->drupalPost('messages/new', $edit, t('Send message'));
    $this->assertText(t('A message has been sent to @role (role).', array('@role' => user_role_load($user1->roles[5])->name)));

    // Log in as user1 and check that the message is listed, is marked as new
    // and can be marked as read.
    $this->drupalLogin($user1);
    $this->drupalGet('messages');
    $this->assertRaw($edit['subject'] . '</a> <span class="marker">new</span>', t('Message is displayed as new'));

    $this->clickLink($edit['subject']);
    $this->assertText($edit['body[value]'], t('Thread starter body displayed.'));

    // Make sure that the user doesn't see the role recipient.
    $this->assertNoText(t('@role (role)', array('@role' => user_role_load($user1->roles[5])->name)));

    // Log in as user2 and make sure that he didn't received the messages
    // as he blocks admin.
    $this->drupalLogin($user2);
    $this->drupalGet('messages');
    $this->assertNoText($edit['subject'], t('Message is not displayed'));
  }

  function testNewUser() {
    $admin = $this->drupalCreateUser(array('read privatemsg', 'write privatemsg', 'write privatemsg to roles'));
    $user1 = $this->drupalCreateUser(array('read privatemsg', 'write privatemsg'));

    // Send a message to the role of user 1.
    $this->drupalLogin($admin);
    $edit = array(
      'recipient' => user_role_load($user1->roles[5])->name . '[role]',
      'subject' => $this->randomName(10),
      'body[value]' => $this->randomName(50),
    );
    $this->drupalPost('messages/new', $edit, t('Send message'));
    $this->assertText(t('A message has been sent to @role (role).', array('@role' => user_role_load($user1->roles[5])->name)));

    // Add user 2 with the same role now. The user should not see the message.
    $user2 = $this->drupalCreateUser();

    // Add role of user 1 to user 2;
    $edit_roles = array('roles' => $user2->roles + $user1->roles);
    user_save($user2, $edit_roles);
    $this->drupalLogin($user2);

    $this->drupalGet('messages');
    $this->assertNoText($edit['subject'], t('Newly added user does not see the old message sent to his role'));
  }
}