Set test context in setup