shimmie2/ext/upload/test.php

<?php
class UploadTest extends ShimmiePHPUnitTestCase
{
    public function testUploadPage()
    {
        $this->log_in_as_user();

        $this->get_page("upload");
        $this->assert_title("Upload");
    }

    public function testUpload()
    {
        $this->log_in_as_user();
        $this->post_image("tests/pbx_screenshot.jpg", "pbx computer 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->assertContains("already has hash", $e->getMessage());
        }
    }

    public function testRejectUnknownFiletype()
    {
        try {
            $this->post_image("index.php", "test");
        } catch (UploadException $e) {
            $this->assertContains("Invalid or corrupted file", $e->getMessage());
        }
    }

    public function testRejectHuge()
    {
        $this->markTestIncomplete();

        // FIXME: huge.dat is rejected for other reasons; manual testing shows that this works
        file_put_contents("huge.dat", file_get_contents("tests/pbx_screenshot.jpg") . str_repeat("U", 1024*1024*3));
        $this->post_image("index.php", "test");
        $this->assert_response(200);
        $this->assert_title("Upload Status");
        $this->assert_text("File too large");
        unlink("huge.dat");
    }
}