added mock context to tests