diff -u b/core/modules/migrate/tests/src/Unit/MigrateTestCase.php b/core/modules/migrate/tests/src/Unit/MigrateTestCase.php
--- b/core/modules/migrate/tests/src/Unit/MigrateTestCase.php
+++ b/core/modules/migrate/tests/src/Unit/MigrateTestCase.php
@@ -50,12 +50,12 @@
     // We need the state to be toggled throughout the test so we store the value
     // on the test class and use a return callback.
     $migration->expects($this->any())
-      ->method('getMigrationStatus')
+      ->method('getStatus')
       ->will($this->returnCallback(function() {
         return $this->migrationStatus;
       }));
     $migration->expects($this->any())
-      ->method('setMigrationStatus')
+      ->method('setStatus')
       ->will($this->returnCallback(function($status) {
         $this->migrationStatus = $status;
       }));