/**
  * {@inheritdoc}
  */
 protected function setUp()
 {
     $module_handler = $this->getMock('Drupal\\Core\\Extension\\ModuleHandlerInterface');
     $migration = $this->getMigration();
     $migration->expects($this->any())->method('getHighwater')->will($this->returnValue(static::ORIGINAL_HIGHWATER));
     // Need the test class, not the original because we need a setDatabase method. This is not pretty :/
     $plugin_class = preg_replace('/^(Drupal\\\\\\w+\\\\)Plugin\\\\migrate(\\\\source(\\\\.+)?\\\\)([^\\\\]+)$/', '\\1Tests\\2Test\\4', static::PLUGIN_CLASS);
     $plugin = new $plugin_class($this->migrationConfiguration['source'], $this->migrationConfiguration['source']['plugin'], array(), $migration);
     $plugin->setDatabase($this->getDatabase($this->databaseContents + array('test_map' => array())));
     $plugin->setModuleHandler($module_handler);
     $plugin->setStringTranslation($this->getStringTranslationStub());
     $migration->expects($this->any())->method('getSourcePlugin')->will($this->returnValue($plugin));
     $migrateExecutable = $this->getMockBuilder('Drupal\\migrate\\MigrateExecutable')->disableOriginalConstructor()->getMock();
     $this->source = new TestSource($migration, $migrateExecutable);
     $cache = $this->getMock('Drupal\\Core\\Cache\\CacheBackendInterface');
     $this->source->setCache($cache);
 }
 /**
  * Tests the source defines a valid ID.
  */
 public function testSourceId()
 {
     $this->assertNotEmpty($this->source->getIds());
 }
 /**
  * Test the source returns the row count expected.
  */
 public function testSourceCount()
 {
     $this->assertEquals($this->source->count(), $this->expectedCount);
 }