add unit test for local file upload

Author: sasha-scale

tests/helpers.py

@@ -29,6 +29,8 @@
     DatasetItem(TEST_IMG_URLS[2], "3"),
     DatasetItem(TEST_IMG_URLS[3], "4"),
 ]
+
+LOCAL_FILENAME = "tests/test_img.jpg"
 TEST_PREDS = [
     BoxPrediction("[Pytest Box Prediction 1]", 0, 0, 100, 100, "1"),
     BoxPrediction("[Pytest Box Prediction 2]", 0, 0, 100, 100, "2"),
@@ -60,7 +62,6 @@ def s3_sign(bucket, key):
         ExpiresIn=PRESIGN_EXPIRY_SECONDS,
     )
 
-
 def reference_id_from_url(url):
     return Path(url).name
 

tests/test_dataset.py

@@ -4,6 +4,7 @@
     TEST_SLICE_NAME,
     TEST_DATASET_NAME,
     TEST_IMG_URLS,
+    LOCAL_FILENAME,
     reference_id_from_url,
 )
 
@@ -84,6 +85,19 @@ def check_is_expected_response(response):
     check_is_expected_response(response)
 
 
+def test_dataset_append_local(CLIENT, dataset):
+    ds_items_local = [DatasetItem(image_location=LOCAL_FILENAME)]
+    response = dataset.append(ds_items_local)
+    assert isinstance(response, UploadResponse)
+    resp_json = response.json()
+    assert resp_json[DATASET_ID_KEY] == dataset.id
+    assert resp_json[NEW_ITEMS] == 1
+    assert resp_json[UPDATED_ITEMS] == 0
+    assert resp_json[IGNORED_ITEMS] == 0
+    assert resp_json[ERROR_ITEMS] == 0
+    assert ERROR_PAYLOAD not in resp_json
+
+
 def test_dataset_list_autotags(CLIENT, dataset):
     # Creation
     # List of Autotags should be empty