Fix flaky/platform-dependent test

Kwpolska · Kwpolska · commit e09ea791120f · 2025-08-02T20:13:10.000+02:00
diff --git a/tests/test_command_import_wordpress.py b/tests/test_command_import_wordpress.py
@@ -339,14 +339,26 @@ def test_write_content_does_not_destroy_text(import_command):
     with mock.patch("nikola.plugins.basic_import.open", open_mock, create=True):
         import_command.write_content("some_file", content)
 
-    open_mock.assert_has_calls(
-        [
-            mock.call(u"some_file", u"wb+"),
-            mock.call().__enter__(),
-            mock.call().write(b"<html><body><p>FOO</p></body></html>"),
-            mock.call().__exit__(None, None, None),
-        ]
-    )
+    actual_calls = open_mock.mock_calls
+    expected_base_calls = [
+        mock.call(u"some_file", u"wb+"),
+        mock.call().__enter__(),
+        # write call should happen here, but we can't check it as easily
+        mock.call().__exit__(None, None, None),
+    ]
+
+    # Verify open/enter/exit calls
+    assert actual_calls[0] == expected_base_calls[0]
+    assert actual_calls[1] == expected_base_calls[1]
+    assert actual_calls[3] == expected_base_calls[2]
+
+    # Verify write call (which may or may not contain <p> tags)
+    write_call = actual_calls[2]
+    assert write_call[0] == '().write'
+    assert write_call[1][0] in [
+        b"<html><body><p>FOO</p></body></html>",
+        b"<html><body>FOO</body></html>"
+    ]
 
 
 def test_configure_redirections(import_command):
