Refactoring of three test into single test with data provider