Add tests for some of the contexts methods that may have been causing a
bug