diff --git a/test/test_mpw.rb b/test/test_mpw.rb index 5697a7b..870a66d 100644 --- a/test/test_mpw.rb +++ b/test/test_mpw.rb @@ -3,6 +3,7 @@ require_relative '../lib/MPW' require 'test/unit' require 'yaml' +require 'csv' class TestMPW < Test::Unit::TestCase @@ -112,4 +113,21 @@ class TestMPW < Test::Unit::TestCase assert(@mpw.import('fixtures.yml')) assert_equal(2, @mpw.search.length) end + + def test_export_yaml + assert(@mpw.import('fixtures.yml')) + assert_equal(2, @mpw.search.length) + assert(@mpw.export('export.yml', :yaml)) + export = YAML::load_file('export.yml') + assert_equal(2, export.length) + end + + def test_export_csv + assert(@mpw.import('fixtures.yml')) + assert_equal(2, @mpw.search.length) + assert(@mpw.export('export.yml', :csv)) + export = CSV.parse(File.read('export.yml'), headers: true) + assert_equal(2, export.length) + + end end