This covers all public methods and overall test coverage is above 80% again. Signed-off-by: Rodrigo Campos <rodrigoca@microsoft.com>