Add db unit support

Change-Id: I2df486f502f2d24bf1171c0bb891afecd0ace582