Let's add some ifdefs around our testing specific code

This way this code is disabled during non test builds.

Signed-off-by: Toon Schoenmakers <nighteyes1993@gmail.com>