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 }
|