Issues with `Layer.mock` and Missing Method Implementations

class ServiceA extends Effect.Service<ServiceA>()('ServiceA', {
    succeed: { foo: () => {}, bar: () => {} }
}) {}

const mockA1 = testLayer(ServiceA, { foo: () => {} });

const mockA2 = Layer.mock(ServiceA, { _tag: 'ServiceA', foo: () => {} }); // missing `bar`

class ServiceB extends Context.Tag('ServiceB')<ServiceB, { foo: () => void, bar: () => void }>(){}

const mockB1 = testLayer(ServiceB, { foo: () => {} });

const mockB2 = Layer.mock(ServiceB, { foo: () => {} }); // missing `bar`

const testLayer = <I, S extends object>(tag: Context.Tag<I, S>, impl = {} as Partial<S>): Layer.Layer<I> =>
  Layer.succeed(tag, makeProxy(tag.key, impl));

const makeProxy = <T extends object>(
    service: string,
    impl: Partial<T>
): T =>
    new Proxy({ ...impl } as T, {
    get(target, prop, _receiver) {
        if (prop in target) {
            return target[prop as keyof T];
        }
        return ((target as any)[prop] = unimplementedMethod(service, prop));
    },
    has: () => true
});

const unimplementedMethod = (id: string, prop: PropertyKey) => {
    const dead = Effect.die(`${id}: Unimplemented method "${prop.toString()}"`);
    function method() { return dead; }
    const yieldable = Object.assign(method, dead);
    Object.setPrototypeOf(yieldable, Object.getPrototypeOf(dead));
    return yieldable;
};