clean: Extensions remove_and_replace!

Author: thekuwayama

Diff for: spec/extensions_spec.rb

@@ -196,19 +196,25 @@
     end
 
     it 'should be equal remove_and_replace! with []' do
-      cloned = extensions.clone
-      expect(extensions.remove_and_replace!([]))
-        .to eq cloned
+      expected = extensions.clone
+      got = extensions.remove_and_replace!([])
+
+      expect(got.keys).to eq expected.keys
+      expect(got[ExtensionType::ECH_OUTER_EXTENSIONS]).to eq nil
+      expect(extensions.keys - got.keys).to eq []
     end
 
     it 'should be equal remove_and_replace! with [key_share]' do
       expected = extensions.filter { |k, _| k != ExtensionType::KEY_SHARE }
       expected[ExtensionType::ECH_OUTER_EXTENSIONS] = \
         Extension::ECHOuterExtensions.new([ExtensionType::KEY_SHARE])
       got = extensions.remove_and_replace!([ExtensionType::KEY_SHARE])
+
       expect(got.keys).to eq expected.keys
       expect(got[ExtensionType::ECH_OUTER_EXTENSIONS].outer_extensions)
         .to eq expected[ExtensionType::ECH_OUTER_EXTENSIONS].outer_extensions
+      expect(extensions.keys - got.keys)
+        .to eq expected[ExtensionType::ECH_OUTER_EXTENSIONS].outer_extensions
     end
 
     it 'should be equal remove_and_replace! with' \
@@ -223,16 +229,22 @@
           extensions.filter { |k, _| outer_extensions.include?(k) }.keys
         )
       got = extensions.remove_and_replace!(outer_extensions)
+
       expect(got.keys).to eq expected.keys
       expect(got[ExtensionType::ECH_OUTER_EXTENSIONS].outer_extensions)
         .to eq expected[ExtensionType::ECH_OUTER_EXTENSIONS].outer_extensions
+      expect(extensions.keys - got.keys)
+        .to eq expected[ExtensionType::ECH_OUTER_EXTENSIONS].outer_extensions
     end
 
     it 'should be equal remove_and_replace! with no key_share extensions' \
        ' & [key_share]' do
-      cloned = no_key_share_exs.clone
-      expect(no_key_share_exs.remove_and_replace!([ExtensionType::KEY_SHARE]))
-        .to eq cloned
+      expected = no_key_share_exs.clone
+      got = no_key_share_exs.remove_and_replace!([ExtensionType::KEY_SHARE])
+
+      expect(got).to eq expected
+      expect(got[ExtensionType::ECH_OUTER_EXTENSIONS]).to eq nil
+      expect(no_key_share_exs.keys - got.keys).to eq []
     end
   end
 end