View source
<?php
namespace Drupal\Tests\file\Functional;
use Drupal\Component\Render\FormattableMarkup;
use Drupal\field\Entity\FieldStorageConfig;
use Drupal\field\Entity\FieldConfig;
use Drupal\file\FileInterface;
use Drupal\Tests\BrowserTestBase;
use Drupal\file\Entity\File;
use Drupal\Tests\TestFileCreationTrait;
abstract class FileFieldTestBase extends BrowserTestBase {
use FileFieldCreationTrait;
use TestFileCreationTrait {
getTestFiles as drupalGetTestFiles;
}
protected static $modules = [
'node',
'file',
'file_module_test',
'field_ui',
];
protected $adminUser;
protected function setUp() {
parent::setUp();
$this->adminUser = $this
->drupalCreateUser([
'access content',
'access administration pages',
'administer site configuration',
'administer users',
'administer permissions',
'administer content types',
'administer node fields',
'administer node display',
'administer nodes',
'bypass node access',
]);
$this
->drupalLogin($this->adminUser);
$this
->drupalCreateContentType([
'type' => 'article',
'name' => 'Article',
]);
}
public function getTestFile($type_name, $size = NULL) {
$file = current($this
->drupalGetTestFiles($type_name, $size));
$file->filesize = filesize($file->uri);
return File::create((array) $file);
}
public function getLastFileId() {
return (int) \Drupal::entityQueryAggregate('file')
->accessCheck(FALSE)
->aggregate('fid', 'max')
->execute()[0]['fid_max'];
}
public function updateFileField($name, $type_name, $field_settings = [], $widget_settings = []) {
$field = FieldConfig::loadByName('node', $type_name, $name);
$field
->setSettings(array_merge($field
->getSettings(), $field_settings));
$field
->save();
\Drupal::service('entity_display.repository')
->getFormDisplay('node', $type_name)
->setComponent($name, [
'settings' => $widget_settings,
])
->save();
}
public function uploadNodeFile(FileInterface $file, $field_name, $nid_or_type, $new_revision = TRUE, array $extras = []) {
return $this
->uploadNodeFiles([
$file,
], $field_name, $nid_or_type, $new_revision, $extras);
}
public function uploadNodeFiles(array $files, $field_name, $nid_or_type, $new_revision = TRUE, array $extras = []) {
$edit = [
'title[0][value]' => $this
->randomMachineName(),
'revision' => (string) (int) $new_revision,
];
$node_storage = $this->container
->get('entity_type.manager')
->getStorage('node');
if (is_numeric($nid_or_type)) {
$nid = $nid_or_type;
$node_storage
->resetCache([
$nid,
]);
$node = $node_storage
->load($nid);
}
else {
$extras['type'] = $nid_or_type;
$node = $this
->drupalCreateNode($extras);
$nid = $node
->id();
$node
->setNewRevision();
$node
->save();
$node_storage
->resetCache([
$nid,
]);
$node = $node_storage
->load($nid);
$this
->assertNotEquals($nid, $node
->getRevisionId(), 'Node revision exists.');
}
$this
->drupalGet("node/{$nid}/edit");
$page = $this
->getSession()
->getPage();
$field_storage = FieldStorageConfig::loadByName('node', $field_name);
$field_num = count($node->{$field_name});
foreach ($files as $i => $file) {
$delta = $field_num + $i;
$file_path = $this->container
->get('file_system')
->realpath($file
->getFileUri());
$name = 'files[' . $field_name . '_' . $delta . ']';
if ($field_storage
->getCardinality() != 1) {
$name .= '[]';
}
if (count($files) == 1) {
$edit[$name] = $file_path;
}
else {
$page
->attachFileToField($name, $file_path);
$this
->submitForm([], 'Upload');
}
}
$this
->submitForm($edit, 'Save');
return $nid;
}
public function removeNodeFile($nid, $new_revision = TRUE) {
$edit = [
'revision' => (string) (int) $new_revision,
];
$this
->drupalGet('node/' . $nid . '/edit');
$this
->submitForm([], 'Remove');
$this
->submitForm($edit, 'Save');
}
public function replaceNodeFile($file, $field_name, $nid, $new_revision = TRUE) {
$edit = [
'files[' . $field_name . '_0]' => \Drupal::service('file_system')
->realpath($file
->getFileUri()),
'revision' => (string) (int) $new_revision,
];
$this
->drupalGet('node/' . $nid . '/edit');
$this
->submitForm([], 'Remove');
$this
->submitForm($edit, 'Save');
}
public function assertFileEntryExists($file, $message = NULL) {
$this->container
->get('entity_type.manager')
->getStorage('file')
->resetCache();
$db_file = File::load($file
->id());
$message = isset($message) ? $message : new FormattableMarkup('File %file exists in database at the correct path.', [
'%file' => $file
->getFileUri(),
]);
$this
->assertEquals($file
->getFileUri(), $db_file
->getFileUri(), $message);
}
public function assertFileEntryNotExists($file, $message) {
$this->container
->get('entity_type.manager')
->getStorage('file')
->resetCache();
$message = isset($message) ? $message : new FormattableMarkup('File %file exists in database at the correct path.', [
'%file' => $file
->getFileUri(),
]);
$this
->assertNull(File::load($file
->id()), $message);
}
public function assertFileIsPermanent(FileInterface $file, $message = NULL) {
$message = isset($message) ? $message : new FormattableMarkup('File %file is permanent.', [
'%file' => $file
->getFileUri(),
]);
$this
->assertTrue($file
->isPermanent(), $message);
}
}