Add datastore - restore tests