refactor test. one test - one assert