Adds tests for entity builder

Signed-off-by: Ahmed H. Ismail <ahm3d.hisham@gmail.com>