typemoq/src/InterceptorSetup.ts

A simple mocking library for TypeScript

import * as all from "./_all";

export class InterceptorSetup<T> implements all.ICallInterceptor {
    private _interceptedCall: all.ICallContext;

    get interceptedCall(): all.ICallContext { return this._interceptedCall; }

    intercept(invocation: all.ICallContext) {
        invocation.invocationType = all.InvocationType.SETUP;
        
        if (invocation.proxyType == all.ProxyType.DYNAMIC &&
            invocation.callType == all.CallType.UNKNOWN)
            invocation.callType = all.CallType.PROPERTY;

        if (this._interceptedCall) {
            throw new all.MockException(all.MockExceptionReason.MoreThanOneSetup,
                invocation, `'${invocation}'; setup should contain only one expression`);
        }

        this._interceptedCall = invocation;
    }

    removeInvocation(invocation: all.ICallContext) {
        if(this._interceptedCall && 
            this._interceptedCall === invocation)
                this._interceptedCall = undefined;
    }
}