refine unit and e2e tests