Handle association validation on Album model

Author: estolfo

app/models/album.rb

@@ -1,6 +1,20 @@
+class Album
+ class Validator < ActiveModel::Validator
+
+ ERROR_MESSAGE = 'An album must be associated with an artist.'.freeze
+
+ def validate(album)
+ unless album.title && album.artist && album.artist.persisted?
+ album.errors.add(:base, ERROR_MESSAGE)
+ end
+ end
+ end
+end
+
 class Album
  include ActiveModel::Model
  include ActiveModel::Validations
+ validates_with Validator
 
  ATTRIBUTES = [:id,
  :artist,

app/repositories/album_repository.rb

@@ -27,6 +27,16 @@ def album_count_by_artist(artist)
  count(query: { match: { artist_id: artist.id } })
  end
 
+ def serialize(album)
+ album.validate!
+ album.to_hash.tap do |hash|
+ hash.merge!(artist_id: album.artist.id)
+ suggest = { title: { input: [hash[:title]] } }
+ suggest[:tracklist] = { input: hash[:tracklist].collect(&:strip) } if hash[:tracklist].present?
+ hash.merge!(:album_suggest => suggest)
+ end
+ end
+
  def all(options = {})
  search({ query: { match_all: { } } },
  { sort: 'title' }.merge(options))

test/repositories/album_repository_test.rb

@@ -63,4 +63,18 @@ class AlbumRepositoryTest < ActiveSupport::TestCase
  assert @repository.save(album)
  assert @repository.find(@repository.save(album)['_id']).is_a?(Album)
  end
+
+ test 'AlbumRepository #serialize' do
+ artist = Artist.new(id: 1, name: 'Common')
+ album = Album.new(title: 'Like Water for Chocolate',
+ artist: artist,
+ tracklist: [' The Light ', ' The Questions '],
+ label: 'MCA')
+ doc = @repository.serialize(album)
+ assert doc[:artist].nil?
+ assert doc[:artist_id] == 1
+ assert doc[:tracklist] == [' The Light ' , ' The Questions ']
+ assert doc[:album_suggest][:title] == { input: ['Like Water for Chocolate'] }
+ assert doc[:album_suggest][:tracklist] == { input: ['The Light', 'The Questions'] }
+ end
 end