Add test for stdin generator