Added tests for assertion flow