src/mock/MethodMocker.ps1
|
# Copyright 2019, Adam Edwards # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. # You may obtain a copy of the License at # # http://www.apache.org/licenses/LICENSE-2.0 # # Unless required by applicable law or agreed to in writing, software # distributed under the License is distributed on an "AS IS" BASIS, # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. $__MethodMocker = [PSCustomObject] @{ MethodPatcher = __MethodPatcher_Get } function __MethodMocker_Get { $script:__MethodMocker } function __MethodMocker_Mock($mockManager, $className, $methodName, $isStatic, $object, $mockScriptBlock, $parameterFilter, $isVerifiableMock) { $patchedMethod = __MethodPatcher_PatchMethod $mockManager.MethodPatcher $className $MethodName $isStatic $object __MethodMocker_MockPatchedMethod $patchedMethod $object $mockScriptBlock $parameterFilter $isVerifiableMock } function __MethodMocker_MockPatchedMethod($patchedMethod, $object, $mockScriptBlock, $parameterFilter, $isVerifiableMock) { $adjustedParameterFilter = if ( $object ) { $patchedObject = __PatchedClassMethod_GetPatchedObject $patchedMethod $Object __PatchedObject_Mock $patchedObject $mockScriptBlock $parameterFilter $patchedObject.ParameterFilter } else { $parameterFilter } Mock $patchedMethod.FunctionName -parameterfilter $adjustedParameterfilter -Verifiable:$IsVerifiableMock -MockWith $mockScriptBlock } function __MethodMocker_Unmock($className, $methodName, $isStatic, $object, $allMocks) { $patcher = __MethodPatcher_Get $targetMethods = if ( $allMocks ) { __MethodPatcher_GetPatchedMethods $patcher } else { __MethodPatcher_QueryPatchedMethods $patcher $ClassName $MethodName $Static $Object } $targetMethods | foreach { __MethodPatcher_Unpatch $patcher $_ $object } } |