Fixed Steam::Liquid::Drops::ContentEntry#to_hash for the case when belong_to content type can be nil

Roman Trofimov committed Jan 15, 2018

commit 2a9eefeb620f9ba9189f1a34122743b010487647

Showing 2 changed files with 32 additions and 1 deletions

locomotive/steam/liquid/drops/content_entry.rb b/lib/locomotive/steam/liquid/drops/content_entry.rb +1 -1

@@	@@ -67,7 +67,7 @@ module Locomotive
	@_source.content_type.fields_by_name.each do \|name, field\|
	case field.type
	when :belongs_to
-	hash[name] = liquify_entry(@_source.send(name))._slug
+	hash[name] = @_source.send(name).presence && liquify_entry(@_source.send(name))._slug
	when :many_to_many
	hash[name] = (@_source.send(name) \|\| []).all.map { \|e\| liquify_entry(e)._slug }.compact
	when :file

spec/unit/liquid/drops/content_entry_spec.rb +31 -0

@@	@@ -106,6 +106,37 @@ describe Locomotive::Steam::Liquid::Drops::ContentEntry do

	end

+	describe '#to_hash' do
+
+	let(:entry) { instance_double('Article', _id: 42, localized_attributes: {}, content_type: type, title: 'Hello world', _label: 'Hello world', _slug: 'hello-world', _translated: false, seo_title: 'seo title', meta_keywords: 'keywords', meta_description: 'description', created_at: 0, updated_at: 1, author: author) }
+	let(:type) { instance_double('Type', fields_by_name: { title: instance_double('StringField', type: :string ), author: instance_double('Author', type: :belongs_to), picture: instance_double('FileField', type: :file), category: instance_double('SelectField', type: :select) }) }
+	let(:picture_field) { Locomotive::Steam::ContentEntry::FileField.new('foo.png', 'http://assets.dev', 0, 42) }
+
+	before do
+	allow(entry).to receive(:to_hash).and_return({ '_id' => 1, 'title' => 'Hello world', 'picture' => picture_field, 'category_id' => 42 })
+	allow(entry).to receive(:category).and_return('Test')
+	end
+
+	subject { drop.to_hash.stringify_keys }
+
+	context 'belongs to content type is not nil' do
+
+	let(:author) { instance_double('Author', _slug: 'john-doe', localized_attributes: {}) }
+
+	it { is_expected.to eq('id' => 1, '_id' => 1, 'title' => 'Hello world', 'picture' => 'http://assets.dev/foo.png?42', 'picture_url' => 'http://assets.dev/foo.png?42', 'category_id' => 42, 'category' => 'Test', 'author' => 'john-doe') }
+
+	end
+
+	context 'belongs to content type is nil' do
+
+	let(:author) { nil }
+
+	it { is_expected.to eq('id' => 1, '_id' => 1, 'title' => 'Hello world', 'picture' => 'http://assets.dev/foo.png?42', 'picture_url' => 'http://assets.dev/foo.png?42', 'category_id' => 42, 'category' => 'Test', 'author' => nil) }
+
+	end
+
+	end
+
	describe '#as_json' do

	let(:entry) { instance_double('Article', _id: 42, localized_attributes: {}, content_type: type, title: 'Hello world', _label: 'Hello world', _slug: 'hello-world', _translated: false, seo_title: 'seo title', meta_keywords: 'keywords', meta_description: 'description', created_at: 0, updated_at: 1, author: author, authors: authors) }