Fix tests.

dist/pigeon/message.rb

@@ -26,7 +26,7 @@ module Pigeon
 
     def self.current
       # TODO: Handle find-or-create logic.
-      @current ||= Marshal.load(Pigeon::Storage.current.get_config(NAME_OF_DRAFT))
+      @current ||= Pigeon::Storage.current.get_config(NAME_OF_DRAFT)
     end
 
     def render
@@ -57,7 +57,7 @@ module Pigeon
     end
 
     def save
-      Pigeon::Storage.current.set_config(NAME_OF_DRAFT, Marshal.dump(self))
+      Pigeon::Storage.current.set_config(NAME_OF_DRAFT, self)
       self
     end
 

spec/pigeon/message_spec.rb

@@ -3,12 +3,21 @@ require "spec_helper"
 RSpec.describe Pigeon::Message do
   it "creates a new message" do
     message = Pigeon::Message.create(kind: "unit_test")
+    expectations = {
+      author: Pigeon::KeyPair.current.public_key,
+      kind: "unit_test",
+      body: { "foo" => "bar".to_json },
+      depth: 0,
+      prev: Pigeon::Message::EMPTY_MESSAGE,
+    }
+    message.append("foo", "bar")
     expect(message.author).to eq(Pigeon::KeyPair.current.public_key)
     expect(message.kind).to eq("unit_test")
-    expect(message.body).to eq({})
+    expect(message.body).to eq(expectations.fetch(:body))
     expect(message.depth).to eq(0)
     expect(message.prev).to eq(Pigeon::Message::EMPTY_MESSAGE)
-    expect(Pigeon::Message.current).to eq(message)
-    message.append("foo", "bar")
+    expectations.map do |k, v|
+      expect(Pigeon::Message.current.send(k)).to eq(v)
+    end
   end
 end