1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
|
<?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();
// 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();
// 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"));
}
}
|
