class ExifOrientationTest

class ExifOrientationTest extends DrupalWebTestCase {
    private $admin;
    public static function getInfo() {
        return array(
            'name' => 'Exif Orientation',
            'description' => 'Tests automatic image orientation.',
            'group' => 'Exif Orientation',
        );
    }
    public function setUp() {
        $modules = array(
            'image',
            'exif_orientation',
        );
        parent::setUp($modules);
        // Set to high quality because we don't want encoding artifacts.
        variable_set('image_jpeg_quality', 98);
        // Code from UserPictureTestCase:
        // Enable user pictures.
        variable_set('user_pictures', 1);
        // Test if directories specified in settings exist in filesystem.
        $file_dir = 'public://';
        $file_check = file_prepare_directory($file_dir, FILE_CREATE_DIRECTORY);
        $picture_dir = variable_get('user_picture_path', 'pictures');
        $picture_path = $file_dir . $picture_dir;
        $pic_check = file_prepare_directory($picture_path, FILE_CREATE_DIRECTORY);
        $this->_directory_test = is_writable($picture_path);
        $this->assertTrue($this->_directory_test, "The directory {$picture_path} doesn't exist or is not writable. Further tests won't be made.");
    }
    
    /**
     * Test auto rotation of uploaded user profile pictures.
     */
    public function testUserPicture() {
        // No user picture style or dimensions.
        variable_set('user_picture_style', '');
        variable_set('user_picture_dimensions', '');
        $user1 = $this->drupalCreateUser();
        $this->saveUserPicture($user1);
        $this->assertImageIsRotated($user1->picture->uri);
        // Applied user picture style.
        variable_set('user_picture_style', 'medium');
        variable_set('user_picture_dimensions', '');
        $user2 = $this->drupalCreateUser();
        $this->saveUserPicture($user2);
        $this->assertImageIsRotated($user2->picture->uri);
        // Defined picture dimensions.
        variable_set('user_picture_style', '');
        variable_set('user_picture_dimensions', '50x50');
        $user3 = $this->drupalCreateUser();
        $this->saveUserPicture($user3);
        $this->assertImageIsRotated($user3->picture->uri);
        // Defined picture style and dimensions.
        variable_set('user_picture_style', 'medium');
        variable_set('user_picture_dimensions', '50x50');
        $user4 = $this->drupalCreateUser();
        $this->saveUserPicture($user4);
        $this->assertImageIsRotated($user4->picture->uri);
    }
    
    /**
     * Uploads a user picture.
     */
    private function saveUserPicture(&$account) {
        $this->drupalLogin($account);
        $edit = array(
            'files[picture_upload]' => drupal_realpath(drupal_get_path('module', 'exif_orientation') . '/tests/rotate90cw.jpg'),
        );
        $this->drupalPost('user/' . $account->uid . '/edit', $edit, t('Save'));
        $account = user_load($account->uid, TRUE);
        $this->assertTrue(isset($account->picture) && isset($account->picture->uri), 'The picture has been uploaded successfully.');
    }
    
    /**
     * Verify that an image is landscape and has a red top left corner.
     */
    private function assertImageIsRotated($uri) {
        $uri = drupal_realpath($uri);
        $img = image_load($uri);
        $this->assertTrue(is_object($img), 'Image data is available.');
        // Test the aspect ratio.
        $this->assertTrue($img->info['width'] > $img->info['height'], 'The image format is landscape.');
        // Verify the rotation by color inspection.
        $rgb = imagecolorat($img->resource, 10, 10);
        $r = $rgb >> 16 & 0xff;
        $g = $rgb >> 8 & 0xff;
        $b = $rgb & 0xff;
        // The top left corner should be red.
        $this->assertTrue(abs($r - 255) < 5, 'Red color component is close to 255.');
        $this->assertTrue($g < 5, 'Green color component is close to 0.');
        $this->assertTrue($b < 5, 'Blue color component is close to 0.');
    }

}