114 lines
		
	
	
		
			3.7 KiB
		
	
	
	
		
			PHP
		
	
	
	
	
	
			
		
		
	
	
			114 lines
		
	
	
		
			3.7 KiB
		
	
	
	
		
			PHP
		
	
	
	
	
	
<?php
 | 
						|
 | 
						|
declare(strict_types=1);
 | 
						|
class UploadTest extends ShimmiePHPUnitTestCase
 | 
						|
{
 | 
						|
    public function testUploadPage()
 | 
						|
    {
 | 
						|
        $this->log_in_as_user();
 | 
						|
 | 
						|
        $this->get_page("upload");
 | 
						|
        $this->assert_title("Upload");
 | 
						|
    }
 | 
						|
 | 
						|
    // Because $this->post_image() sends the event directly
 | 
						|
    public function testRawUpload()
 | 
						|
    {
 | 
						|
        global $database;
 | 
						|
 | 
						|
        $this->log_in_as_user();
 | 
						|
        $_FILES = [
 | 
						|
            'data0' => [
 | 
						|
                'name' => ['puppy-hugs.jpg'],
 | 
						|
                'type' => ['image/jpeg'],
 | 
						|
                'tmp_name' => ['tests/bedroom_workshop.jpg'],
 | 
						|
                'error' => [0],
 | 
						|
                'size' => [271386],
 | 
						|
            ],
 | 
						|
            'data1' => [
 | 
						|
                'name' => ['cat-hugs-2.jpg', 'cat-hugs-3.jpg', 'cat-hugs.jpg'],
 | 
						|
                'type' => ['image/png', 'image/jpeg', 'image/svg'],
 | 
						|
                'tmp_name' => ['tests/favicon.png', 'tests/pbx_screenshot.jpg', 'tests/test.svg'],
 | 
						|
                'error' => [0, 0, 0],
 | 
						|
                'size' => [110361, 64021, 421410],
 | 
						|
            ],
 | 
						|
            'data2' => [
 | 
						|
                'name' => [''],
 | 
						|
                'type' => [''],
 | 
						|
                'tmp_name' => [''],
 | 
						|
                'error' => [4],
 | 
						|
                'size' => [0],
 | 
						|
            ]
 | 
						|
        ];
 | 
						|
        $page = $this->post_page("upload", ["tags0"=>"foo bar"]);
 | 
						|
        $this->assert_response(302);
 | 
						|
        $this->assertStringStartsWith("/test/post/list/poster=test/1", $page->redirect);
 | 
						|
 | 
						|
        $this->assertEquals(4, $database->get_one("SELECT COUNT(*) FROM images"));
 | 
						|
    }
 | 
						|
 | 
						|
    public function testRawReplace()
 | 
						|
    {
 | 
						|
        global $database;
 | 
						|
 | 
						|
        $this->log_in_as_admin();
 | 
						|
        $image_id = $this->post_image("tests/pbx_screenshot.jpg", "pbx computer screenshot");
 | 
						|
 | 
						|
        $_FILES = [
 | 
						|
            'data' => [
 | 
						|
                'name' => ['puppy-hugs.jpg'],
 | 
						|
                'type' => ['image/jpeg'],
 | 
						|
                'tmp_name' => ['tests/bedroom_workshop.jpg'],
 | 
						|
                'error' => [0],
 | 
						|
                'size' => [271386],
 | 
						|
            ]
 | 
						|
        ];
 | 
						|
 | 
						|
        $page = $this->post_page("upload/replace", ["image_id"=>$image_id]);
 | 
						|
        $this->assert_response(302);
 | 
						|
        $this->assertEquals("/test/post/view/$image_id", $page->redirect);
 | 
						|
 | 
						|
        $this->assertEquals(1, $database->get_one("SELECT COUNT(*) FROM images"));
 | 
						|
    }
 | 
						|
 | 
						|
    public function testUpload()
 | 
						|
    {
 | 
						|
        $this->log_in_as_user();
 | 
						|
        $image_id = $this->post_image("tests/pbx_screenshot.jpg", "pbx computer screenshot");
 | 
						|
        $this->assertGreaterThan(0, $image_id);
 | 
						|
 | 
						|
        $this->get_page("post/view/$image_id");
 | 
						|
        $this->assert_title("Post $image_id: computer pbx screenshot");
 | 
						|
    }
 | 
						|
 | 
						|
    public function testRejectDupe()
 | 
						|
    {
 | 
						|
        $this->post_image("tests/pbx_screenshot.jpg", "pbx computer screenshot");
 | 
						|
 | 
						|
        try {
 | 
						|
            $this->post_image("tests/pbx_screenshot.jpg", "pbx computer screenshot");
 | 
						|
        } catch (UploadException $e) {
 | 
						|
            $this->assertStringContainsString("already has hash", $e->getMessage());
 | 
						|
        }
 | 
						|
    }
 | 
						|
 | 
						|
    public function testRejectUnknownFiletype()
 | 
						|
    {
 | 
						|
        $image_id = $this->post_image("index.php", "test");
 | 
						|
        $this->assertEquals(-1, $image_id);  // no file handler claimed this
 | 
						|
    }
 | 
						|
 | 
						|
    public function testRejectHuge()
 | 
						|
    {
 | 
						|
        // FIXME: huge.dat is rejected for other reasons; manual testing shows that this works
 | 
						|
        file_put_contents("data/huge.jpg", file_get_contents("tests/pbx_screenshot.jpg") . str_repeat("U", 1024*1024*3));
 | 
						|
        try {
 | 
						|
            $this->post_image("data/huge.jpg", "test");
 | 
						|
            $this->assertTrue(false, "Uploading huge.jpg didn't fail...");
 | 
						|
        } catch (UploadException $e) {
 | 
						|
            $this->assertEquals("File too large (3.0MB > 1.0MB)", $e->error);
 | 
						|
        }
 | 
						|
        unlink("data/huge.jpg");
 | 
						|
    }
 | 
						|
}
 |