add-test

Change-Id: I829e3ce10e6f6a9d406ebf79662a4d8c83b48f9d

Author: MarkDaoust

tests/test_content.py

@@ -35,6 +35,9 @@
 TEST_JPG_URL = "https://storage.googleapis.com/generativeai-downloads/data/test_img.jpg"
 TEST_JPG_DATA = TEST_JPG_PATH.read_bytes()
 
+TEST_WEBP_PATH = HERE / "test_img.webp"
+TEST_WEBP_DATA = TEST_WEBP_PATH.read_bytes()
+
 
 # simple test function
 def datetime():

tests/test_files.py

@@ -28,6 +28,18 @@
 from google.generativeai import protos
 from absl.testing import parameterized
 
+HERE = pathlib.Path(__file__).parent
+TEST_PNG_PATH = HERE / "test_img.png"
+TEST_PNG_URL = "https://storage.googleapis.com/generativeai-downloads/data/test_img.png"
+TEST_PNG_DATA = TEST_PNG_PATH.read_bytes()
+
+TEST_JPG_PATH = HERE / "test_img.jpg"
+TEST_JPG_URL = "https://storage.googleapis.com/generativeai-downloads/data/test_img.jpg"
+TEST_JPG_DATA = TEST_JPG_PATH.read_bytes()
+
+TEST_WEBP_PATH = HERE / "test_img.webp"
+TEST_WEBP_DATA = TEST_WEBP_PATH.read_bytes()
+
 
 class FileServiceClient(client_lib.FileServiceClient):
     def __init__(self, test):
@@ -139,3 +151,10 @@ def test_video_metadata(self):
     def test_to_file_data(self, file_data):
         file_data = file_types.to_file_data(file_data)
         self.assertEqual(protos.FileData(file_uri="https://test_uri"), file_data)
+
+    def test_webp_mimetype(self):
+        self.responses["create_file"].append(protos.File())
+
+        response = genai.upload_file("test.webp")
+
+        self.assertEqual("image/webp", self.observed_requests[0]["mime_type"])