class TestMigrateExecutable

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

}