Change the context.Context to an interface

[bkielbasa]

Is your feature request related to a problem? Please describe.

While mocking etc having interfaces can make life easier. Another argument behind this is that there's a general rule that we should accept interfaces (in test code too) and return structs.

The change would help to introduce the support for custom feature contexts #52