annotate core/modules/migrate/tests/src/Unit/TestMigrateExecutable.php @ 19:fa3358dc1485 tip

Add ndrum files
author Chris Cannam
date Wed, 28 Aug 2019 13:14:47 +0100
parents 4c8ae668cc8c
children
rev   line source
Chris@0 1 <?php
Chris@0 2
Chris@0 3 namespace Drupal\Tests\migrate\Unit;
Chris@0 4
Chris@0 5 use Drupal\Core\StringTranslation\TranslationInterface;
Chris@0 6 use Drupal\migrate\MigrateExecutable;
Chris@0 7
Chris@0 8 /**
Chris@0 9 * Tests MigrateExecutable.
Chris@0 10 */
Chris@0 11 class TestMigrateExecutable extends MigrateExecutable {
Chris@0 12
Chris@0 13 /**
Chris@0 14 * The fake memory usage in bytes.
Chris@0 15 *
Chris@0 16 * @var int
Chris@0 17 */
Chris@0 18 protected $memoryUsage;
Chris@0 19
Chris@0 20 /**
Chris@0 21 * The cleared memory usage.
Chris@0 22 *
Chris@0 23 * @var int
Chris@0 24 */
Chris@0 25 protected $clearedMemoryUsage;
Chris@0 26
Chris@0 27 /**
Chris@0 28 * Sets the string translation service.
Chris@0 29 *
Chris@0 30 * @param \Drupal\Core\StringTranslation\TranslationInterface $string_translation
Chris@0 31 * The translation manager.
Chris@0 32 */
Chris@0 33 public function setStringTranslation(TranslationInterface $string_translation) {
Chris@0 34 $this->stringTranslation = $string_translation;
Chris@0 35 }
Chris@0 36
Chris@0 37 /**
Chris@0 38 * Allows access to set protected source property.
Chris@0 39 *
Chris@0 40 * @param \Drupal\migrate\Plugin\MigrateSourceInterface $source
Chris@0 41 * The value to set.
Chris@0 42 */
Chris@0 43 public function setSource($source) {
Chris@0 44 $this->source = $source;
Chris@0 45 }
Chris@0 46
Chris@0 47 /**
Chris@0 48 * Allows access to protected sourceIdValues property.
Chris@0 49 *
Chris@0 50 * @param array $source_id_values
Chris@0 51 * The values to set.
Chris@0 52 */
Chris@0 53 public function setSourceIdValues($source_id_values) {
Chris@0 54 $this->sourceIdValues = $source_id_values;
Chris@0 55 }
Chris@0 56
Chris@0 57 /**
Chris@0 58 * {@inheritdoc}
Chris@0 59 */
Chris@0 60 public function handleException(\Exception $exception, $save = TRUE) {
Chris@0 61 $message = $exception->getMessage();
Chris@0 62 if ($save) {
Chris@0 63 $this->saveMessage($message);
Chris@0 64 }
Chris@0 65 $this->message->display($message);
Chris@0 66 }
Chris@0 67
Chris@0 68 /**
Chris@0 69 * Allows access to the protected memoryExceeded method.
Chris@0 70 *
Chris@0 71 * @return bool
Chris@0 72 * The memoryExceeded value.
Chris@0 73 */
Chris@0 74 public function memoryExceeded() {
Chris@0 75 return parent::memoryExceeded();
Chris@0 76 }
Chris@0 77
Chris@0 78 /**
Chris@0 79 * {@inheritdoc}
Chris@0 80 */
Chris@0 81 protected function attemptMemoryReclaim() {
Chris@0 82 return $this->clearedMemoryUsage;
Chris@0 83 }
Chris@0 84
Chris@0 85 /**
Chris@0 86 * {@inheritdoc}
Chris@0 87 */
Chris@0 88 protected function getMemoryUsage() {
Chris@0 89 return $this->memoryUsage;
Chris@0 90 }
Chris@0 91
Chris@0 92 /**
Chris@0 93 * Sets the fake memory usage.
Chris@0 94 *
Chris@0 95 * @param int $memory_usage
Chris@0 96 * The fake memory usage value.
Chris@0 97 * @param int $cleared_memory_usage
Chris@0 98 * (optional) The fake cleared memory value. Defaults to NULL.
Chris@0 99 */
Chris@0 100 public function setMemoryUsage($memory_usage, $cleared_memory_usage = NULL) {
Chris@0 101 $this->memoryUsage = $memory_usage;
Chris@0 102 $this->clearedMemoryUsage = $cleared_memory_usage;
Chris@0 103 }
Chris@0 104
Chris@0 105 /**
Chris@0 106 * Sets the memory limit.
Chris@0 107 *
Chris@0 108 * @param int $memory_limit
Chris@0 109 * The memory limit.
Chris@0 110 */
Chris@0 111 public function setMemoryLimit($memory_limit) {
Chris@0 112 $this->memoryLimit = $memory_limit;
Chris@0 113 }
Chris@0 114
Chris@0 115 /**
Chris@0 116 * Sets the memory threshold.
Chris@0 117 *
Chris@0 118 * @param float $threshold
Chris@0 119 * The new threshold.
Chris@0 120 */
Chris@0 121 public function setMemoryThreshold($threshold) {
Chris@0 122 $this->memoryThreshold = $threshold;
Chris@0 123 }
Chris@0 124
Chris@0 125 /**
Chris@0 126 * {@inheritdoc}
Chris@0 127 */
Chris@0 128 protected function formatSize($size) {
Chris@0 129 return $size;
Chris@0 130 }
Chris@0 131
Chris@0 132 }