add unit tests for the migration

1 files changed, 84 insertions, 0 deletions

diff --git a/lib/Migration/FixCollectedAddresses.php b/lib/Migration/FixCollectedAddresses.php
new file mode 100644
index 000000000..05be85466
--- /dev/null
+++ b/lib/Migration/FixCollectedAddresses.php
@@ -0,0 +1,84 @@
+<?php
+
+/**
+ * @author Christoph Wurst <christoph@winzerhof-wurst.at>
+ *
+ * Mail
+ *
+ * This code is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Affero General Public License, version 3,
+ * as published by the Free Software Foundation.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU Affero General Public License for more details.
+ *
+ * You should have received a copy of the GNU Affero General Public License, version 3,
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>
+ *
+ */
+
+namespace OCA\Mail\Migration;
+
+use Horde_Mail_Exception;
+use Horde_Mail_Rfc822_Address;
+use OCA\Mail\Db\CollectedAddress;
+use OCA\Mail\Db\CollectedAddressMapper;
+use OCP\Migration\IOutput;
+use OCP\Migration\IRepairStep;
+
+class FixCollectedAddresses implements IRepairStep {
+
+	/** @var CollectedAddressMapper */
+	private $mapper;
+
+	public function __construct(CollectedAddressMapper $mapper) {
+		$this->mapper = $mapper;
+	}
+
+	public function getName() {
+		return 'Purify and migrate collected mail addresses';
+	}
+
+	public function run(IOutput $output) {
+		$nrOfAddresses = $this->mapper->getTotal();
+		$output->startProgress($nrOfAddresses);
+
+		$chunk = $this->mapper->getChunk();
+		while (count($chunk) > 0) {
+			$maxId = null;
+			foreach ($chunk as $address) {
+				/* @var $address CollectedAddress */
+				$maxId = $address->getId();
+				$this->fixAddress($address, $output);
+			}
+
+			$output->advance(count($chunk));
+			$chunk = $this->mapper->getChunk($maxId + 1);
+		}
+		$output->finishProgress();
+	}
+
+	private function fixAddress(CollectedAddress $address, IOutput $output) {
+		if (!is_null($address->getDisplayName())) {
+			// Nothing to fix
+			return;
+		}
+
+		try {
+			$hordeAddress = new Horde_Mail_Rfc822_Address($address->getEmail());
+			if (!$hordeAddress->valid) {
+				throw new Horde_Mail_Exception();
+			}
+			$address->setDisplayName($hordeAddress->label);
+			$address->setEmail($hordeAddress->bare_address);
+			$this->mapper->update($address);
+		} catch (Horde_Mail_Exception $ex) {
+			$output->warning('removed invalid address <' . $address->getEmail() . '>');
+			// Invalid address, let's delete it to prevent further errors
+			$this->mapper->delete($address);
+		}
+	}
+
+}