createMock(Validator::class); $this->userValidationRules = $this->createMock(\Magento\User\Model\UserValidationRules::class); $this->rules = $objectManager->getObject(\Magento\User\Model\UserValidationRules::class); $this->validator = $this->createMock(DataObject::class); $this->plugin = $objectManager->getObject( UserValidationRules::class, ['validator' => $userExpirationValidator] ); } public function testAfterAddUserInfoRules() { $this->validator->expects(static::exactly(5))->method('addRule')->willReturn($this->validator); static::assertSame($this->validator, $this->rules->addUserInfoRules($this->validator)); static::assertSame($this->validator, $this->callAfterAddUserInfoRulesPlugin($this->validator)); } protected function callAfterAddUserInfoRulesPlugin($validator) { return $this->plugin->afterAddUserInfoRules($this->userValidationRules, $validator); } }