add test with db unit