X-Git-Url: https://git.cworth.org/git?a=blobdiff_plain;f=lib%2Fsup%2Fmessage.rb;h=afa8f008f354d28802ca6f78062dd353032a11f3;hb=90271a77cd8986ee230afb908e4a1cff01a5a2e3;hp=c69b9e30327ff6e51d70b8b2395f47107c0dc68e;hpb=d4c1bd840f55a07fd5aeb4d08a1f43c3f7c1850e;p=sup

diff --git a/lib/sup/message.rb b/lib/sup/message.rb
index c69b9e3..afa8f00 100644
--- a/lib/sup/message.rb
+++ b/lib/sup/message.rb
@@ -191,11 +191,13 @@ class Message
 
   def has_label? t; @labels.member? t; end
   def add_label l
+    l = l.to_sym
     return if @labels.member? l
     @labels << l
     @dirty = true
   end
   def remove_label l
+    l = l.to_sym
     return unless @labels.member? l
     @labels.delete l
     @dirty = true
@@ -207,6 +209,7 @@ class Message
 
   def labels= l
     raise ArgumentError, "not a set" unless l.is_a?(Set)
+    raise ArgumentError, "not a set of labels" unless l.all? { |ll| ll.is_a?(Symbol) }
     return if @labels == l
     @labels = l
     @dirty = true