Chris@0: <?php
Chris@0: 
Chris@0: namespace Drupal\Tests\migrate\Unit;
Chris@0: 
Chris@0: use Drupal\Core\StringTranslation\TranslationInterface;
Chris@0: use Drupal\migrate\MigrateExecutable;
Chris@0: 
Chris@0: /**
Chris@0:  * Tests MigrateExecutable.
Chris@0:  */
Chris@0: class TestMigrateExecutable extends MigrateExecutable {
Chris@0: 
Chris@0:   /**
Chris@0:    * The fake memory usage in bytes.
Chris@0:    *
Chris@0:    * @var int
Chris@0:    */
Chris@0:   protected $memoryUsage;
Chris@0: 
Chris@0:   /**
Chris@0:    * The cleared memory usage.
Chris@0:    *
Chris@0:    * @var int
Chris@0:    */
Chris@0:   protected $clearedMemoryUsage;
Chris@0: 
Chris@0:   /**
Chris@0:    * Sets the string translation service.
Chris@0:    *
Chris@0:    * @param \Drupal\Core\StringTranslation\TranslationInterface $string_translation
Chris@0:    *   The translation manager.
Chris@0:    */
Chris@0:   public function setStringTranslation(TranslationInterface $string_translation) {
Chris@0:     $this->stringTranslation = $string_translation;
Chris@0:   }
Chris@0: 
Chris@0:   /**
Chris@0:    * Allows access to set protected source property.
Chris@0:    *
Chris@0:    * @param \Drupal\migrate\Plugin\MigrateSourceInterface $source
Chris@0:    *   The value to set.
Chris@0:    */
Chris@0:   public function setSource($source) {
Chris@0:     $this->source = $source;
Chris@0:   }
Chris@0: 
Chris@0:   /**
Chris@0:    * Allows access to protected sourceIdValues property.
Chris@0:    *
Chris@0:    * @param array $source_id_values
Chris@0:    *   The values to set.
Chris@0:    */
Chris@0:   public function setSourceIdValues($source_id_values) {
Chris@0:     $this->sourceIdValues = $source_id_values;
Chris@0:   }
Chris@0: 
Chris@0:   /**
Chris@0:    * {@inheritdoc}
Chris@0:    */
Chris@0:   public function handleException(\Exception $exception, $save = TRUE) {
Chris@0:     $message = $exception->getMessage();
Chris@0:     if ($save) {
Chris@0:       $this->saveMessage($message);
Chris@0:     }
Chris@0:     $this->message->display($message);
Chris@0:   }
Chris@0: 
Chris@0:   /**
Chris@0:    * Allows access to the protected memoryExceeded method.
Chris@0:    *
Chris@0:    * @return bool
Chris@0:    *   The memoryExceeded value.
Chris@0:    */
Chris@0:   public function memoryExceeded() {
Chris@0:     return parent::memoryExceeded();
Chris@0:   }
Chris@0: 
Chris@0:   /**
Chris@0:    * {@inheritdoc}
Chris@0:    */
Chris@0:   protected function attemptMemoryReclaim() {
Chris@0:     return $this->clearedMemoryUsage;
Chris@0:   }
Chris@0: 
Chris@0:   /**
Chris@0:    * {@inheritdoc}
Chris@0:    */
Chris@0:   protected function getMemoryUsage() {
Chris@0:     return $this->memoryUsage;
Chris@0:   }
Chris@0: 
Chris@0:   /**
Chris@0:    * Sets the fake memory usage.
Chris@0:    *
Chris@0:    * @param int $memory_usage
Chris@0:    *   The fake memory usage value.
Chris@0:    * @param int $cleared_memory_usage
Chris@0:    *   (optional) The fake cleared memory value. Defaults to NULL.
Chris@0:    */
Chris@0:   public function setMemoryUsage($memory_usage, $cleared_memory_usage = NULL) {
Chris@0:     $this->memoryUsage = $memory_usage;
Chris@0:     $this->clearedMemoryUsage = $cleared_memory_usage;
Chris@0:   }
Chris@0: 
Chris@0:   /**
Chris@0:    * Sets the memory limit.
Chris@0:    *
Chris@0:    * @param int $memory_limit
Chris@0:    *   The memory limit.
Chris@0:    */
Chris@0:   public function setMemoryLimit($memory_limit) {
Chris@0:     $this->memoryLimit = $memory_limit;
Chris@0:   }
Chris@0: 
Chris@0:   /**
Chris@0:    * Sets the memory threshold.
Chris@0:    *
Chris@0:    * @param float $threshold
Chris@0:    *   The new threshold.
Chris@0:    */
Chris@0:   public function setMemoryThreshold($threshold) {
Chris@0:     $this->memoryThreshold = $threshold;
Chris@0:   }
Chris@0: 
Chris@0:   /**
Chris@0:    * {@inheritdoc}
Chris@0:    */
Chris@0:   protected function formatSize($size) {
Chris@0:     return $size;
Chris@0:   }
Chris@0: 
Chris@0: }