Update riak_test modules for new riak_test api