test: Migrate tests to new direct API's

Signed-off-by: Michael Ikey Doherty <michael.i.doherty@intel.com>