[TASK] Add context unit-tests