Improved mock generator tests.