[refactor] passing test to be a bit more abstract and doesn't depend on having a class