Add Riak integration tests