Ensure bidirectional linkage in WithStory

app/models/concerns/with_story.rb

@@ -7,16 +7,16 @@ module WithStory
     def with_story(&)
       belongs_to :story, optional: true, dependent: :destroy
 
-      before_update do
+      after_update do
         story_data = instance_eval(&).data
         story.update!(data: story_data) if story_data != story.data
       end
 
-      before_create do
+      after_create do
         story = instance_eval(&)
         story.created_at = created_at
         story.save!
-        self.story_id = story.id
+        update_attribute(:story_id, story.id)
       end
     end
   end

spec/models/concerns/with_story_spec.rb

@@ -23,7 +23,10 @@
     end
 
     it 'creates a story when the record is created' do
-      expect { StoryOwner.create! }.to change(Story, :count).by(1)
+      owner = StoryOwner.new
+      expect { owner.save! }.to change(Story, :count).by(1)
+      expect(Story.last.data['test_id']).to eq(owner.id)
+      expect(owner.story_id).to eq(Story.last.id)
     end
 
     it 'updates the story when the data changes' do