--- /dev/null
+<?php
+
+namespace Drupal\Tests\file\Kernel;
+
+use Drupal\file\Entity\File;
+use Drupal\file\FileInterface;
+use Drupal\KernelTests\KernelTestBase;
+use Drupal\user\Entity\User;
+
+/**
+ * Base class for file unit tests that use the file_test module to test uploads and
+ * hooks.
+ */
+abstract class FileManagedUnitTestBase extends KernelTestBase {
+
+ /**
+ * Modules to enable.
+ *
+ * @var array
+ */
+ public static $modules = ['file_test', 'file', 'system', 'field', 'user'];
+
+ protected function setUp() {
+ parent::setUp();
+ // Clear out any hook calls.
+ file_test_reset();
+
+ $this->installConfig(['system']);
+ $this->installEntitySchema('file');
+ $this->installEntitySchema('user');
+ $this->installSchema('file', ['file_usage']);
+
+ // Make sure that a user with uid 1 exists, self::createFile() relies on
+ // it.
+ $user = User::create(['uid' => 1, 'name' => $this->randomMachineName()]);
+ $user->enforceIsNew();
+ $user->save();
+ \Drupal::currentUser()->setAccount($user);
+ }
+
+ /**
+ * Assert that all of the specified hook_file_* hooks were called once, other
+ * values result in failure.
+ *
+ * @param array $expected
+ * Array with string containing with the hook name, e.g. 'load', 'save',
+ * 'insert', etc.
+ */
+ public function assertFileHooksCalled($expected) {
+ \Drupal::state()->resetCache();
+
+ // Determine which hooks were called.
+ $actual = array_keys(array_filter(file_test_get_all_calls()));
+
+ // Determine if there were any expected that were not called.
+ $uncalled = array_diff($expected, $actual);
+ if (count($uncalled)) {
+ $this->assertTrue(FALSE, format_string('Expected hooks %expected to be called but %uncalled was not called.', ['%expected' => implode(', ', $expected), '%uncalled' => implode(', ', $uncalled)]));
+ }
+ else {
+ $this->assertTrue(TRUE, format_string('All the expected hooks were called: %expected', ['%expected' => empty($expected) ? '(none)' : implode(', ', $expected)]));
+ }
+
+ // Determine if there were any unexpected calls.
+ $unexpected = array_diff($actual, $expected);
+ if (count($unexpected)) {
+ $this->assertTrue(FALSE, format_string('Unexpected hooks were called: %unexpected.', ['%unexpected' => empty($unexpected) ? '(none)' : implode(', ', $unexpected)]));
+ }
+ else {
+ $this->assertTrue(TRUE, 'No unexpected hooks were called.');
+ }
+ }
+
+ /**
+ * Assert that a hook_file_* hook was called a certain number of times.
+ *
+ * @param string $hook
+ * String with the hook name, e.g. 'load', 'save', 'insert', etc.
+ * @param int $expected_count
+ * Optional integer count.
+ * @param string $message
+ * Optional translated string message.
+ */
+ public function assertFileHookCalled($hook, $expected_count = 1, $message = NULL) {
+ $actual_count = count(file_test_get_calls($hook));
+
+ if (!isset($message)) {
+ if ($actual_count == $expected_count) {
+ $message = format_string('hook_file_@name was called correctly.', ['@name' => $hook]);
+ }
+ elseif ($expected_count == 0) {
+ $message = \Drupal::translation()->formatPlural($actual_count, 'hook_file_@name was not expected to be called but was actually called once.', 'hook_file_@name was not expected to be called but was actually called @count times.', ['@name' => $hook, '@count' => $actual_count]);
+ }
+ else {
+ $message = format_string('hook_file_@name was expected to be called %expected times but was called %actual times.', ['@name' => $hook, '%expected' => $expected_count, '%actual' => $actual_count]);
+ }
+ }
+ $this->assertEqual($actual_count, $expected_count, $message);
+ }
+
+ /**
+ * Asserts that two files have the same values (except timestamp).
+ *
+ * @param \Drupal\file\FileInterface $before
+ * File object to compare.
+ * @param \Drupal\file\FileInterface $after
+ * File object to compare.
+ */
+ public function assertFileUnchanged(FileInterface $before, FileInterface $after) {
+ $this->assertEqual($before->id(), $after->id(), t('File id is the same: %file1 == %file2.', ['%file1' => $before->id(), '%file2' => $after->id()]), 'File unchanged');
+ $this->assertEqual($before->getOwner()->id(), $after->getOwner()->id(), t('File owner is the same: %file1 == %file2.', ['%file1' => $before->getOwner()->id(), '%file2' => $after->getOwner()->id()]), 'File unchanged');
+ $this->assertEqual($before->getFilename(), $after->getFilename(), t('File name is the same: %file1 == %file2.', ['%file1' => $before->getFilename(), '%file2' => $after->getFilename()]), 'File unchanged');
+ $this->assertEqual($before->getFileUri(), $after->getFileUri(), t('File path is the same: %file1 == %file2.', ['%file1' => $before->getFileUri(), '%file2' => $after->getFileUri()]), 'File unchanged');
+ $this->assertEqual($before->getMimeType(), $after->getMimeType(), t('File MIME type is the same: %file1 == %file2.', ['%file1' => $before->getMimeType(), '%file2' => $after->getMimeType()]), 'File unchanged');
+ $this->assertEqual($before->getSize(), $after->getSize(), t('File size is the same: %file1 == %file2.', ['%file1' => $before->getSize(), '%file2' => $after->getSize()]), 'File unchanged');
+ $this->assertEqual($before->isPermanent(), $after->isPermanent(), t('File status is the same: %file1 == %file2.', ['%file1' => $before->isPermanent(), '%file2' => $after->isPermanent()]), 'File unchanged');
+ }
+
+ /**
+ * Asserts that two files are not the same by comparing the fid and filepath.
+ *
+ * @param \Drupal\file\FileInterface $file1
+ * File object to compare.
+ * @param \Drupal\file\FileInterface $file2
+ * File object to compare.
+ */
+ public function assertDifferentFile(FileInterface $file1, FileInterface $file2) {
+ $this->assertNotEqual($file1->id(), $file2->id(), t('Files have different ids: %file1 != %file2.', ['%file1' => $file1->id(), '%file2' => $file2->id()]), 'Different file');
+ $this->assertNotEqual($file1->getFileUri(), $file2->getFileUri(), t('Files have different paths: %file1 != %file2.', ['%file1' => $file1->getFileUri(), '%file2' => $file2->getFileUri()]), 'Different file');
+ }
+
+ /**
+ * Asserts that two files are the same by comparing the fid and filepath.
+ *
+ * @param \Drupal\file\FileInterface $file1
+ * File object to compare.
+ * @param \Drupal\file\FileInterface $file2
+ * File object to compare.
+ */
+ public function assertSameFile(FileInterface $file1, FileInterface $file2) {
+ $this->assertEqual($file1->id(), $file2->id(), t('Files have the same ids: %file1 == %file2.', ['%file1' => $file1->id(), '%file2-fid' => $file2->id()]), 'Same file');
+ $this->assertEqual($file1->getFileUri(), $file2->getFileUri(), t('Files have the same path: %file1 == %file2.', ['%file1' => $file1->getFileUri(), '%file2' => $file2->getFileUri()]), 'Same file');
+ }
+
+ /**
+ * Create a file and save it to the files table and assert that it occurs
+ * correctly.
+ *
+ * @param string $filepath
+ * Optional string specifying the file path. If none is provided then a
+ * randomly named file will be created in the site's files directory.
+ * @param string $contents
+ * Optional contents to save into the file. If a NULL value is provided an
+ * arbitrary string will be used.
+ * @param string $scheme
+ * Optional string indicating the stream scheme to use. Drupal core includes
+ * public, private, and temporary. The public wrapper is the default.
+ * @return \Drupal\file\FileInterface
+ * File entity.
+ */
+ public function createFile($filepath = NULL, $contents = NULL, $scheme = NULL) {
+ // Don't count hook invocations caused by creating the file.
+ \Drupal::state()->set('file_test.count_hook_invocations', FALSE);
+ $file = File::create([
+ 'uri' => $this->createUri($filepath, $contents, $scheme),
+ 'uid' => 1,
+ ]);
+ $file->save();
+ // Write the record directly rather than using the API so we don't invoke
+ // the hooks.
+ $this->assertTrue($file->id() > 0, 'The file was added to the database.', 'Create test file');
+
+ \Drupal::state()->set('file_test.count_hook_invocations', TRUE);
+ return $file;
+ }
+
+ /**
+ * Creates a file and returns its URI.
+ *
+ * @param string $filepath
+ * Optional string specifying the file path. If none is provided then a
+ * randomly named file will be created in the site's files directory.
+ * @param string $contents
+ * Optional contents to save into the file. If a NULL value is provided an
+ * arbitrary string will be used.
+ * @param string $scheme
+ * Optional string indicating the stream scheme to use. Drupal core includes
+ * public, private, and temporary. The public wrapper is the default.
+ *
+ * @return string
+ * File URI.
+ */
+ public function createUri($filepath = NULL, $contents = NULL, $scheme = NULL) {
+ if (!isset($filepath)) {
+ // Prefix with non-latin characters to ensure that all file-related
+ // tests work with international filenames.
+ $filepath = 'Файл для тестирования ' . $this->randomMachineName();
+ }
+ if (!isset($scheme)) {
+ $scheme = file_default_scheme();
+ }
+ $filepath = $scheme . '://' . $filepath;
+
+ if (!isset($contents)) {
+ $contents = "file_put_contents() doesn't seem to appreciate empty strings so let's put in some data.";
+ }
+
+ file_put_contents($filepath, $contents);
+ $this->assertTrue(is_file($filepath), t('The test file exists on the disk.'), 'Create test file');
+ return $filepath;
+ }
+
+}