[refactor] use builder pattern for data model in tests