X-Git-Url: http://www.aleph1.co.uk/gitweb/?p=yaffs-website;a=blobdiff_plain;f=web%2Fmodules%2Fcontrib%2Fdrupalmoduleupgrader%2Ftests%2Fsrc%2FUnit%2FContainerMockTrait.php;fp=web%2Fmodules%2Fcontrib%2Fdrupalmoduleupgrader%2Ftests%2Fsrc%2FUnit%2FContainerMockTrait.php;h=0e115f3fbecaa8b9abe54704805ed08898030fc7;hp=0000000000000000000000000000000000000000;hb=8acec36f19c470dfcda1ae2336826a782f41874c;hpb=e0411c4e83ba0d079034db83c3f7f55be24a0e35

diff --git a/web/modules/contrib/drupalmoduleupgrader/tests/src/Unit/ContainerMockTrait.php b/web/modules/contrib/drupalmoduleupgrader/tests/src/Unit/ContainerMockTrait.php
new file mode 100644
index 000000000..0e115f3fb
--- /dev/null
+++ b/web/modules/contrib/drupalmoduleupgrader/tests/src/Unit/ContainerMockTrait.php
@@ -0,0 +1,46 @@
+<?php
+
+namespace Drupal\Tests\drupalmoduleupgrader\Unit;
+
+use Drupal\Core\DependencyInjection\ContainerBuilder;
+
+/**
+ * A trait for tests that need a mock container; contains (deprecated) methods
+ * to mock basic translation and logging services as well.
+ */
+trait ContainerMockTrait {
+
+  /**
+   * @var \Symfony\Component\DependencyInjection\ContainerInterface
+   */
+  protected $container;
+
+  protected function mockContainer() {
+    if (empty($this->container)) {
+      // Using a ContainerBuilder lets us simply stick services into the
+      // container, which is a whole lot easier than mocking it!
+      $this->container = new ContainerBuilder();
+    }
+  }
+
+  protected function mockTranslator() {
+    $this->mockContainer();
+
+    // Mock the string_translation service; calling its translate()
+    // method will return the original, unprocessed string.
+    $translator = $this->getMock('\Drupal\Core\StringTranslation\TranslationInterface');
+    $translator->method('translate')->willReturnArgument(0);
+    $this->container->set('string_translation', $translator);
+  }
+
+  protected function mockLogger() {
+    $this->mockContainer();
+
+    // Mock the logger.factory service and a logger channel.
+    $factory = $this->getMock('\Drupal\Core\Logger\LoggerChannelFactoryInterface');
+    $channel = $this->getMock('\Drupal\Core\Logger\LoggerChannelInterface');
+    $factory->method('get')->willReturn($channel);
+    $this->container->set('logger.factory', $factory);
+  }
+
+}