feat: return mock from then methods

[mcous]

This PR steals a convenient part of td.js' API that I haven't stolen yet:

// before
const mock = vi.fn()
when(mock).calledWith(1, 2).thenReturn(3)

// after
const mock = when(vi.fn()).calledWith(1, 2).thenReturn(3)