added expect into the test example