diff --git a/requirements.txt b/requirements.txt index 9b8f0b0..6b008a8 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,5 +1,5 @@ androidtvremote2==0.2.1 -ucapi==0.3.0 +ucapi==0.3.1 pyee~=13.0.0 google_play_scraper==1.2.7 pillow>=11.2.1 diff --git a/tests/test_util.py b/tests/test_util.py index c067060..d027ab9 100644 --- a/tests/test_util.py +++ b/tests/test_util.py @@ -21,21 +21,25 @@ def test_data_with_non_data_uri_icon(self): data = {"icon": "http://example.com/icon.png"} result = filter_data_img_properties(data) self.assertEqual(result, data) + self.assertEqual(data, {"icon": "http://example.com/icon.png"}) def test_data_with_data_uri_icon(self): data = {"icon": "data:image/png;base64,somedata"} result = filter_data_img_properties(data) self.assertEqual(result["icon"], "data:***") + self.assertEqual(data["icon"], "data:image/png;base64,somedata") def test_data_with_media_image_url(self): data = {MediaAttr.MEDIA_IMAGE_URL: "data:image/png;base64,somedata"} result = filter_data_img_properties(data) self.assertEqual(result[MediaAttr.MEDIA_IMAGE_URL], "data:***") + self.assertEqual(data[MediaAttr.MEDIA_IMAGE_URL], "data:image/png;base64,somedata") def test_msg_data_with_attributes(self): data = {"msg_data": {"attributes": {MediaAttr.MEDIA_IMAGE_URL: "data:image/png;base64,somedata"}}} result = filter_data_img_properties(deepcopy(data)) self.assertEqual(result["msg_data"]["attributes"][MediaAttr.MEDIA_IMAGE_URL], "data:***") + self.assertEqual(data["msg_data"]["attributes"][MediaAttr.MEDIA_IMAGE_URL], "data:image/png;base64,somedata") def test_msg_data_list_with_attributes_mixed(self): data = { @@ -46,7 +50,10 @@ def test_msg_data_list_with_attributes_mixed(self): } result = filter_data_img_properties(deepcopy(data)) self.assertEqual(result["msg_data"][0]["attributes"][MediaAttr.MEDIA_IMAGE_URL], "data:***") + self.assertEqual(data["msg_data"][0]["attributes"][MediaAttr.MEDIA_IMAGE_URL], "data:image/png;base64,somedata") + self.assertEqual(result["msg_data"][1]["attributes"][MediaAttr.MEDIA_IMAGE_URL], "http://example.com/icon.png") + self.assertEqual(data["msg_data"][1]["attributes"][MediaAttr.MEDIA_IMAGE_URL], "http://example.com/icon.png") def test_msg_data_list_with_attributes(self): data = { @@ -59,6 +66,9 @@ def test_msg_data_list_with_attributes(self): self.assertEqual(result["msg_data"][0]["attributes"][MediaAttr.MEDIA_IMAGE_URL], "data:***") self.assertEqual(result["msg_data"][1]["attributes"][MediaAttr.MEDIA_IMAGE_URL], "data:***") + self.assertEqual(data["msg_data"][0]["attributes"][MediaAttr.MEDIA_IMAGE_URL], "data:image/png;base64,somedata") + self.assertEqual(data["msg_data"][1]["attributes"][MediaAttr.MEDIA_IMAGE_URL], "data:image/png;base64,somedata") + def test_no_modifications_needed(self): data = { "icon": "http://example.com/icon.png", @@ -67,6 +77,8 @@ def test_no_modifications_needed(self): } result = filter_data_img_properties(deepcopy(data)) self.assertEqual(result, data) + self.assertEqual(result[MediaAttr.MEDIA_IMAGE_URL], "http://example.com/image.png") + self.assertEqual(result["msg_data"][0]["attributes"][MediaAttr.MEDIA_IMAGE_URL], "http://example.com/image.png") if __name__ == "__main__":