lib/usaidwat/service.rb in usaidwat-1.1.1 vs lib/usaidwat/service.rb in usaidwat-1.2.0

@@ -1,34 +1,42 @@
 module USaidWat
   module Service
     class MockComment
-      attr_reader :subreddit, :body, :id, :link_id, :created_utc, :link_title
+      attr_reader :subreddit, :body, :id, :link_id, :created_utc, :link_title, :ups, :downs
 
       def initialize(dict)
         data = dict['data']
         @subreddit = data['subreddit']
         @body = data['body']
         @id = data['id']
         @link_id = data['link_id']
         @created_utc = data['created_utc']
         @link_title = data['link_title']
+        @ups = data['ups']
+        @downs = data['downs']
       end
     end
 
     class MockUser
       def initialize(username)
         @username = username
       end
 
+      def about
+        load_data("user_#{@username}.json")['data']
+      end
+
       def comments(n)
-        path = File.join(File.dirname(__FILE__), "..", "..", "features", "fixtures", "#{@username}.json")
-        if File.exists?(path)
-          json = IO.read(path)
-          json = JSON.parse(json)
-          json['data']['children'].map { |d| MockComment.new(d) }
-        else
-          raise USaidWat::Client::NoSuchUserError, @username
-        end
+        json = load_data("#{@username}.json")
+        json['data']['children'].map { |d| MockComment.new(d) }
+      end
+
+      private
+
+      def load_data(data_file)
+        path = File.join(File.dirname(__FILE__), "..", "..", "features", "fixtures", data_file)
+        raise USaidWat::Client::NoSuchUserError, @username unless File.exists?(path)
+        JSON.parse(IO.read(path))
       end
     end
 
     class MockService
       def user(username)