gallery3/modules/watermark/tests/Admin_Watermarks_Controller_Test.php

<?php defined("SYSPATH") or die("No direct script access.");
/**
 * Gallery - a web based photo album viewer and editor
 * Copyright (C) 2000-2013 Bharat Mediratta
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or (at
 * your option) any later version.
 *
 * This program 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
 * General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street - Fifth Floor, Boston, MA  02110-1301, USA.
 */
class Admin_Watermarks_Controller_Test extends Gallery_Unit_Test_Case {
  public function setup() {
    $this->_save = array($_POST, $_SERVER);
    $_SERVER["HTTP_REFERER"] = "HTTP_REFERER";
  }

  public function teardown() {
    list($_POST, $_SERVER) = $this->_save;
  }

  public function add_watermark_test() {
    // Source is a jpg file, watermark path has extension jpg
    $name = test::random_name();
    $source_path = MODPATH . "gallery/images/imagemagick.jpg";
    $watermark_path = TMPPATH . "uploadfile-123-{$name}.jpg";
    copy($source_path, $watermark_path);

    // Setup and run Admin_Watermarks_Controller::add
    $controller = new Admin_Watermarks_Controller();
    $_POST["file"] = $watermark_path;
    $_POST["csrf"] = access::csrf_token();
    ob_start();
    $controller->add();
    $results = ob_get_clean();

    // Add should be successful
    $this->assert_equal(json_encode(array("result" => "success",
                                          "location" => url::site("admin/watermarks"))), $results);
    $this->assert_equal(file_get_contents($source_path),
                        file_get_contents(VARPATH . "modules/watermark/$name.jpg"));
    $this->assert_equal("$name.jpg", module::get_var("watermark", "name"));
    $this->assert_equal(114, module::get_var("watermark", "width"));
    $this->assert_equal(118, module::get_var("watermark", "height"));
    $this->assert_equal("image/jpeg", module::get_var("watermark", "mime_type"));
  }

  public function add_watermark_reject_illegal_file_test() {
    // Source is a php file, watermark path has extension php
    $name = test::random_name();
    $source_path = MODPATH . "watermark/tests/Admin_Watermarks_Controller_Test.php";
    $watermark_path = TMPPATH . "uploadfile-123-{$name}.php";
    copy($source_path, $watermark_path);

    // Setup and run Admin_Watermarks_Controller::add
    $controller = new Admin_Watermarks_Controller();
    $_POST["file"] = $watermark_path;
    $_POST["csrf"] = access::csrf_token();
    ob_start();
    $controller->add();
    $results = ob_get_clean();

    // Delete all files marked using system::delete_later (from gallery_event::gallery_shutdown)
    system::delete_marked_files();

    // Add should *not* be successful, and watermark should be deleted
    $this->assert_equal("", $results);
    $this->assert_false(file_exists($watermark_path));
    $this->assert_false(file_exists(VARPATH . "modules/watermark/$name.php"));
  }

  public function add_watermark_rename_legal_file_with_illegal_extension_test() {
    // Source is a jpg file, watermark path has extension php
    $name = test::random_name();
    $source_path = MODPATH . "gallery/images/imagemagick.jpg";
    $watermark_path = TMPPATH . "uploadfile-123-{$name}.php";
    copy($source_path, $watermark_path);

    // Setup and run Admin_Watermarks_Controller::add
    $controller = new Admin_Watermarks_Controller();
    $_POST["file"] = $watermark_path;
    $_POST["csrf"] = access::csrf_token();
    ob_start();
    $controller->add();
    $results = ob_get_clean();

    // Add should be successful with file renamed as jpg
    $this->assert_equal(json_encode(array("result" => "success",
                                          "location" => url::site("admin/watermarks"))), $results);
    $this->assert_equal(file_get_contents($source_path),
                        file_get_contents(VARPATH . "modules/watermark/$name.jpg"));
    $this->assert_equal("$name.jpg", module::get_var("watermark", "name"));
    $this->assert_equal(114, module::get_var("watermark", "width"));
    $this->assert_equal(118, module::get_var("watermark", "height"));
    $this->assert_equal("image/jpeg", module::get_var("watermark", "mime_type"));
  }

  public function add_watermark_reject_illegal_file_with_legal_extension_test() {
    // Source is a php file, watermark path has extension jpg
    $name = test::random_name();
    $source_path = MODPATH . "watermark/tests/Admin_Watermarks_Controller_Test.php";
    $watermark_path = TMPPATH . "uploadfile-123-{$name}.jpg";
    copy($source_path, $watermark_path);

    // Setup and run Admin_Watermarks_Controller::add
    $controller = new Admin_Watermarks_Controller();
    $_POST["file"] = $watermark_path;
    $_POST["csrf"] = access::csrf_token();
    ob_start();
    $controller->add();
    $results = ob_get_clean();

    // Delete all files marked using system::delete_later (from gallery_event::gallery_shutdown)
    system::delete_marked_files();

    // Add should *not* be successful, and watermark should be deleted
    $this->assert_equal("", $results);
    $this->assert_false(file_exists($watermark_path));
    $this->assert_false(file_exists(VARPATH . "modules/watermark/$name.php"));
    $this->assert_false(file_exists(VARPATH . "modules/watermark/$name.jpg"));
  }
}