Use interface in unit tests to ensure conformity

Signed-off-by: Onsi Fakhouri <onsi@pivotallabs.com>