Add missing includes in tests

Caused mysterious test failures on s390x due to (presumably) wrong
return type of some functions.

Signed-off-by: Martin Milata <mmilata@redhat.com>