Update unit tests.

Signed-off-by: Hugues Peccatte <hugues.peccatte@gmail.com>
Add missing include in unit tests.

Signed-off-by: Hugues Peccatte <hugues.peccatte@gmail.com>