Minor code formatting improvements

alphagov · Dec 23, 2016 · 8f180a9 · 8f180a9
1 parent 4f178c3
commit 8f180a9
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 7 deletions.
diff --git a/app/commands/v2/discard_draft.rb b/app/commands/v2/discard_draft.rb
@@ -64,16 +64,14 @@ def increment_live_lock_version
 
       def draft
         @draft ||= ContentItem.joins(:document).find_by(
-          'documents.content_id': content_id,
-          'documents.locale': locale,
+          documents: { content_id: content_id, locale: locale },
           state: "draft",
         )
       end
 
       def live
         @live ||= ContentItem.joins(:document).find_by(
-          'documents.content_id': content_id,
-          'documents.locale': locale,
+          documents: { content_id: content_id, locale: locale },
           state: %w(published unpublished),
         )
       end

diff --git a/app/commands/v2/patch_link_set.rb b/app/commands/v2/patch_link_set.rb
@@ -136,7 +136,7 @@ def schema_validator
 
       def schema_name
         @schema_name ||= Queries::GetLatest.(
-          ContentItem.where('documents.content_id': payload[:content_id])
+          ContentItem.where(documents: { content_id: content_id })
         ).pluck(:schema_name).first
       end
     end

diff --git a/app/commands/v2/post_action.rb b/app/commands/v2/post_action.rb
@@ -43,8 +43,7 @@ def action_type
 
       def find_content_item
         content_item = ContentItem.joins(:document).find_by(
-          'documents.content_id': content_id,
-          'documents.locale': locale,
+          documents: { content_id: content_id, locale: locale },
           state: draft? ? %w(draft) : %w(published unpublished),
         )