function FileDownloadTest::testPrivateFileTransfer

Test the private file transfer system.

File

modules/simpletest/tests/file.test, line 2610

Class

FileDownloadTest

Tests for download/file transfer functions.

Code

function testPrivateFileTransfer() {
    // Set file downloads to private so handler functions get called.
    // Create a file.
    $contents = $this->randomName(8);
    $file = $this->createFile(NULL, $contents, 'private');
    $url = file_create_url($file->uri);
    // Set file_test access header to allow the download.
    file_test_reset();
    file_test_set_return('download', array(
        'x-foo' => 'Bar',
    ));
    $this->drupalGet($url);
    $headers = $this->drupalGetHeaders();
    $this->assertEqual($headers['x-foo'], 'Bar', 'Found header set by file_test module on private download.');
    $this->assertResponse(200, 'Correctly allowed access to a file when file_test provides headers.');
    // Ensure hook_file_download is fired correctly.
    $hooks_results = file_test_get_all_calls();
    $file_uri = !empty($hooks_results['download']) ? reset($hooks_results['download'][0]) : '';
    $this->assertEqual($file->uri, $file_uri);
    // Test that the file transferred correctly.
    $this->assertEqual($contents, $this->content, 'Contents of the file are correct.');
    // Deny access to all downloads via a -1 header.
    file_test_set_return('download', -1);
    $this->drupalHead($url);
    $this->assertResponse(403, 'Correctly denied access to a file when file_test sets the header to -1.');
    // Try non-existent file.
    file_test_reset();
    $url = file_create_url('private://' . $this->randomName());
    $this->drupalHead($url);
    $this->assertResponse(404, 'Correctly returned 404 response for a non-existent file.');
    // Assert that hook_file_download is not called.
    $hooks_results = file_test_get_all_calls();
    $hook_download_results = isset($hooks_results['download']) ? $hooks_results['download'] : NULL;
    $this->assertEqual(array(), $hook_download_results);
    // Try requesting the private file url without a file specified.
    file_test_reset();
    $this->drupalGet('system/files');
    $this->assertResponse(404, 'Correctly returned 404 response for a private file url without a file specified.');
    // Assert that hook_file_download is not called.
    $hooks_results = file_test_get_all_calls();
    $hook_download_results = isset($hooks_results['download']) ? $hooks_results['download'] : NULL;
    $this->assertEqual(array(), $hook_download_results);
}