Address Post Bridge PR review comments

Author: FujiwaraChoki

src/classes/PostBridge.py

@@ -219,6 +219,10 @@ def _request(
 
         for attempt in range(1, self._max_retries + 1):
             try:
+                request_data = kwargs.get("data")
+                if hasattr(request_data, "seek"):
+                    request_data.seek(0)
+
                 response = self._session.request(
                     method,
                     url,

src/config.py

@@ -360,6 +360,8 @@ def get_post_bridge_config() -> dict:
         config_json = json.load(file)
 
     raw_config = config_json.get("post_bridge", {})
+    if not isinstance(raw_config, dict):
+        raw_config = {}
 
     raw_platforms = raw_config.get("platforms")
     normalized_platforms = []

tests/test_config.py

@@ -63,6 +63,22 @@ def test_non_list_platforms_fail_closed(self) -> None:
 
         self.assertEqual(post_bridge_config["platforms"], [])
 
+    def test_non_object_post_bridge_config_falls_back_to_defaults(self) -> None:
+        with tempfile.TemporaryDirectory() as temp_dir:
+            self.write_config(
+                temp_dir,
+                {
+                    "post_bridge": None,
+                },
+            )
+
+            with patch.object(config, "ROOT_DIR", temp_dir):
+                post_bridge_config = config.get_post_bridge_config()
+
+        self.assertEqual(post_bridge_config["platforms"], ["tiktok", "instagram"])
+        self.assertEqual(post_bridge_config["account_ids"], [])
+        self.assertFalse(post_bridge_config["enabled"])
+
 
 if __name__ == "__main__":
     unittest.main()

tests/test_post_bridge_client.py

@@ -126,6 +126,31 @@ def test_upload_media_does_not_forward_api_bearer_token_to_signed_upload_url(sel
         self.assertEqual(upload_call.args[1], "https://signed-upload.example/path")
         self.assertEqual(upload_call.kwargs["headers"], {"Content-Type": "video/mp4"})
 
+    @patch("classes.PostBridge.time.sleep")
+    def test_request_rewinds_streamed_upload_body_before_retry(self, sleep_mock) -> None:
+        session = Mock()
+        session.request.side_effect = [
+            MockResponse(500, {"message": "temporary error"}),
+            MockResponse(200, {}),
+        ]
+        client = PostBridge("token", session=session)
+
+        stream = unittest.mock.mock_open(read_data=b"video").return_value
+        stream.seek = Mock()
+
+        client._request(
+            "PUT",
+            "https://signed-upload.example/path",
+            data=stream,
+            headers={"Content-Type": "video/mp4"},
+            expected_statuses={200},
+            use_default_headers=False,
+        )
+
+        self.assertEqual(stream.seek.call_count, 2)
+        stream.seek.assert_any_call(0)
+        sleep_mock.assert_called_once()
+
 
 if __name__ == "__main__":
     unittest.main()