Update specs

Add tests to make sure valid `code-lang` attributes are kept and invalid ones removed.

Signed-off-by: Plastikmensch <plastikmensch@users.noreply.github.com>

spec/lib/advanced_text_formatter_spec.rb

@@ -50,6 +50,14 @@
         it 'does not format links' do
           expect(subject).to include 'return 0; // https://joinmastodon.org/foo'
         end
+
+        context 'with valid language' do
+          let(:text) { "test\n\n```c++\nint main(void) {\n  return 0; // https://joinmastodon.org/foo\n}\n```\n" }
+
+          it 'formats code using <pre> and <code> with data containing set language' do
+            expect(subject).to include '<pre><code data-codelang="c++">int main'
+          end
+        end
       end
 
       context 'with a link in inline code using backticks' do

spec/lib/sanitize_config_spec.rb

@@ -55,6 +55,14 @@
     it 'keeps title in abbr' do
       expect(Sanitize.fragment('<abbr title="HyperText Markup Language">HTML</abbr>', subject)).to eq '<abbr title="HyperText Markup Language">HTML</abbr>'
     end
+
+    it 'keeps data-codelang attribute in code' do
+      expect(Sanitize.fragment('<code data-codelang="c++">int main(void) { return 0; // https://joinmastodon.org/foo }</code>', subject)).to eq '<code data-codelang="c++">int main(void) { return 0; // https://joinmastodon.org/foo }</code>'
+    end
+
+    it 'removes data-codelang attribute in code when unsupported' do
+      expect(Sanitize.fragment('<code data-codelang="invalid">int main(void) { return 0; // https://joinmastodon.org/foo }</code>', subject)).to eq '<code>int main(void) { return 0; // https://joinmastodon.org/foo }</code>'
+    end
   end
 
   describe '::MASTODON_OUTGOING' do