Utilise "setup" and "teardown" methods when testing.

Signed-off-by: Krzysztof Wilczynski <krzysztof.wilczynski@linux.com>