spec/support/fake_keikoku.rb in keikokuc-0.0.3 vs spec/support/fake_keikoku.rb in keikokuc-0.1

@@ -1,29 +1,61 @@
 class FakeKeikoku
   def initialize
     @publishers = []
     @notifications = []
+    @users = []
   end
 
   def register_publisher(opts)
     @publishers << opts
   end
 
+  def register_user(opts)
+    @users << opts
+  end
+
   def publisher_by_api_key(api_key)
     @publishers.detect { |p| p[:api_key] == api_key }
   end
 
+  def find_user(user, pass)
+    @users.detect { |u| u[:email] == user && u[:password] == pass }
+  end
+
+  def notifications_for_user(email)
+    @notifications.select do |notification|
+      notification.to_hash['account_email'] == email
+    end
+  end
+
   def call(env)
     with_rack_env(env) do
       if request_path == '/api/v1/notifications' && request_verb == 'POST'
         if publisher_by_api_key(request_api_key)
           notification = Notification.new({id: next_id}.merge(request_body))
           @notifications << notification
           [200, { }, [Yajl::Encoder.encode({id: notification.id})]]
         else
           [401, { }, ["Not authorized"]]
         end
+      elsif request_path == '/api/v1/notifications' && request_verb == 'GET'
+        if current_user = authenticate_consumer
+          notifications = notifications_for_user(current_user).map(&:to_hash)
+          [200, { }, [Yajl::Encoder.encode(notifications)]]
+        else
+          [401, { }, ["Not authorized"]]
+        end
+      elsif request_path =~ %r{/api/v1/notifications/([^/]+)/read} && request_verb == 'POST'
+        if current_user = authenticate_consumer
+          notification = notifications_for_user(current_user).detect do |notification|
+            notification.to_hash[:id].to_s == $1.to_s
+          end
+          notification.mark_read_by!(current_user)
+          [200, {}, [Yajl::Encoder.encode({read_by: current_user, read_at: Time.now})]]
+        else
+          [401, { }, ["Not authorized"]]
+        end
       end
     end
   end
 
 private
@@ -55,10 +87,20 @@
 
   def request_api_key
     rack_env["HTTP_X_KEIKOKU_AUTH"]
   end
 
+  def authenticate_consumer
+    auth = Rack::Auth::Basic::Request.new(rack_env)
+    if auth.provided? && auth.basic? && creds = auth.credentials
+      # creds looks like [user, password]
+      if find_user(*creds)
+        creds.first
+      end
+    end
+  end
+
   def next_id
     @@sequence ||= 0
     @@sequence += 1
   end
 
@@ -68,8 +110,16 @@
       opts.each do |key, value|
         self.class.send :define_method, key do
           value
         end
       end
+    end
+
+    def to_hash
+      @opts
+    end
+
+    def mark_read_by!(user)
+      (@read_by ||= []) << user
     end
   end
 end