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
125
126
127
128
129
130
|
<?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"));
}
}
|
