please split it into two tests and do not put logic in tests