From 31601f011c33c1b6fb60c29c4e49d60adf082307 Mon Sep 17 00:00:00 2001
From: Laurent Curau <laurent.curau@digital.cabinet-office.gov.uk>
Date: Mon, 22 Jan 2018 14:16:56 +0000
Subject: [PATCH] Fix undefined method `empty?' for nil:NilClass

Coercing @datafiles and @docs into an array ensures
we can always call array methods in the view
---
 app/models/dataset.rb | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/app/models/dataset.rb b/app/models/dataset.rb
index 1f276a32..dd393d63 100644
--- a/app/models/dataset.rb
+++ b/app/models/dataset.rb
@@ -15,7 +15,7 @@ class Dataset
                 :_index, :_type, :_id, :_score, :_source,
                 :_version
 
-  attr_reader :organisation,:docs, :datafiles
+  attr_reader :organisation
 
   index_name ENV['ES_INDEX'] || "datasets-#{Rails.env}"
 
@@ -62,10 +62,18 @@ def self.datafile_formats
     map_keys(buckets)
   end
 
+  def docs
+    Array(@docs)
+  end
+
   def docs=(docs)
     @docs = docs.map { |file| Doc.new(file) }
   end
 
+  def datafiles
+    Array(@datafiles)
+  end
+
   def datafiles=(datafiles)
     @datafiles = datafiles.map { |file| Datafile.new(file)}
   end