--- /dev/null
+<?php
+
+namespace Drupal\Tests\drupalmoduleupgrader\Unit\Plugin\DMU\Converter\Functions;
+
+use Pharborist\Filter;
+use Pharborist\Parser;
+
+/**
+ * @group DMU.Converter.Functions
+ * @covers \Drupal\drupalmoduleupgrader\Plugin\DMU\Converter\Functions\CommentLoad
+ */
+class CommentLoadTest extends FunctionCallModifierTestBase {
+
+ public function testRewriteWithoutCacheReset() {
+ $function_call = Parser::parseExpression('comment_load(30)');
+ $rewritten = $this->plugin->rewrite($function_call, $this->target);
+ $this->assertInstanceOf('\Pharborist\Objects\ObjectMethodCallNode', $rewritten);
+ $this->assertEquals('\Drupal::entityManager()->getStorage(\'comment\')->load(30)', $rewritten->getText());
+ }
+
+ public function testRewriteWithCacheReset() {
+ $original = <<<'END'
+comment_load(30, TRUE);
+END;
+ $expected = <<<'END'
+// @FIXME
+// To reset the comment cache, use EntityStorageInterface::resetCache().
+\Drupal::entityManager()->getStorage('comment')->load(30);
+END;
+ $snippet = Parser::parseSnippet($original);
+ $function_call = $snippet->children(Filter::isFunctionCall('comment_load'))->get(0);
+ $rewritten = $this->plugin->rewrite($function_call, $this->target);
+ $function_call->replaceWith($rewritten);
+ $this->assertEquals($expected, $snippet->getText());
+ }
+
+}