test: improve code readability

General context approach is used in multi-instances functions -- context
is passed as a first argument.

Author: elhimov