Expose state and update_state method

lib/s2/connection.rb

@@ -2,7 +2,7 @@ module S2
   class Connection
     include S2::MessageHandler
 
-    attr_reader :sent_messages
+    attr_reader :state, :sent_messages
 
     on S2::Messages::ReceptionStatus do |message|
       if message_sent?(message.subject_message_id)
@@ -16,8 +16,9 @@ class Connection
     def initialize(ws, logger: Rails.logger)
       @ws = ws
       @logger = logger
-      @state = :connected
       @sent_messages = {}
+
+      update_state :connected
     end
 
     def receive_message(message_json)
@@ -64,6 +65,10 @@ def reply(message, status:)
       end
     end
 
+    def update_state(new_state)
+      @state = new_state
+    end
+
     def message_sent?(message_id)
       @sent_messages.has_key?(message_id)
     end