(maint) Fix OS import_export test

Prior to this commit, the open source test was not working anymore. This
commit fixes that by moving the Puppet Enterprise test logic around and
keeping the old open source test method in the helper.