also test if not in a room

litaio · Jun 17, 2015 · b138ae6 · b138ae6
1 parent e210b85
commit b138ae6
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/spec/lita/rspec_spec.rb b/spec/lita/rspec_spec.rb
@@ -16,8 +16,12 @@ def message(response)
   end
 
   def channel(response)
-    response.reply(response.message.source.room_object.id)
-    response.reply(response.message.source.room_object.name)
+    if (room = response.message.source.room_object)
+      response.reply(room.id)
+      response.reply(room.name)
+    else
+      response.reply("No room")
+    end
   end
 
   def command(response)
@@ -112,6 +116,10 @@ def self.name
       send_message("channel", from: room)
       expect(replies).to eq(%w(1 Room))
     end
+    it "replies with no channel if not sent from room" do
+      send_message("channel")
+      expect(replies).to eq(["No room"])
+    end
   end
 
   describe "#command" do