refactoring tests to use DataProvider