aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorAaron Patterson <aaron.patterson@gmail.com>2011-05-10 10:42:03 -0700
committerAaron Patterson <aaron.patterson@gmail.com>2011-05-23 15:06:12 -0700
commit081b36c6ce799f2e4c755bdaf2fa978c9e494567 (patch)
tree5b8c74ae96d10764e7b5f24bdaf4dcfc36be92b6
parent5278af3d8efbe348864baca0c40a532e8b137ce1 (diff)
downloadrails-081b36c6ce799f2e4c755bdaf2fa978c9e494567.tar.gz
rails-081b36c6ce799f2e4c755bdaf2fa978c9e494567.tar.bz2
rails-081b36c6ce799f2e4c755bdaf2fa978c9e494567.zip
fixture file will validate fixture format
-rw-r--r--activerecord/lib/active_record/fixtures/file.rb9
-rw-r--r--activerecord/test/cases/fixtures/file_test.rb18
2 files changed, 26 insertions, 1 deletions
diff --git a/activerecord/lib/active_record/fixtures/file.rb b/activerecord/lib/active_record/fixtures/file.rb
index beda5a65df..da07cf5da1 100644
--- a/activerecord/lib/active_record/fixtures/file.rb
+++ b/activerecord/lib/active_record/fixtures/file.rb
@@ -34,12 +34,19 @@ module ActiveRecord
return @rows if @rows
data = YAML.load(render(IO.read(@file)))
- @rows = data ? data.to_a : []
+ @rows = data ? validate(data).to_a : []
end
def render(content)
ERB.new(content).result
end
+
+ # Validate our unmarshalled data.
+ def validate(data)
+ raise Fixture::FormatError, 'fixture is not a hash' unless Hash === data
+ raise Fixture::FormatError unless data.all? { |name, row| Hash === row }
+ data
+ end
end
end
end
diff --git a/activerecord/test/cases/fixtures/file_test.rb b/activerecord/test/cases/fixtures/file_test.rb
index 38494a7857..174f41f412 100644
--- a/activerecord/test/cases/fixtures/file_test.rb
+++ b/activerecord/test/cases/fixtures/file_test.rb
@@ -48,6 +48,24 @@ module ActiveRecord
end
end
+ # A valid YAML file is not necessarily a value Fixture file. Make sure
+ # an exception is raised if the format is not valid Fixture format.
+ def test_wrong_fixture_format_string
+ tmp_yaml ['empty', 'yml'], 'qwerty' do |t|
+ assert_raises(ActiveRecord::Fixture::FormatError) do
+ File.open(t.path) { |fh| fh.to_a }
+ end
+ end
+ end
+
+ def test_wrong_fixture_format_nested
+ tmp_yaml ['empty', 'yml'], 'one: two' do |t|
+ assert_raises(ActiveRecord::Fixture::FormatError) do
+ File.open(t.path) { |fh| fh.to_a }
+ end
+ end
+ end
+
private
def tmp_yaml(name, contents)
t = Tempfile.new name