Adding tests to cover three of the four remaining code paths.