minor fix

armosec · Nov 16, 2023 · cca6cf7 · cca6cf7
1 parent 39315f6
commit cca6cf7
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/s3connector/s3_mock.go b/s3connector/s3_mock.go
@@ -21,10 +21,10 @@ func NewS3Mock() S3Mock {
 	}
 }
 
-func (s3 *S3Mock) DeleteObject(key string) error {
+func (s3 *S3Mock) DeleteObject(objPath S3ObjectPath) error {
 	s3.mux.Lock()
 	defer s3.mux.Unlock()
-	delete(s3.storage, key)
+	delete(s3.storage, objPath.Key)
 	return nil
 }
 

diff --git a/s3connector/s3_mock_test.go b/s3connector/s3_mock_test.go
@@ -71,8 +71,9 @@ func TestS3Mock_DeleteObject(t *testing.T) {
 	mock := NewS3Mock()
 	mock.storage["keyToDelete"] = "value"
 
+	objPath := S3ObjectPath{Key: "keyToDelete"}
 	// Test deleting existing object
-	err := mock.DeleteObject("keyToDelete")
+	err := mock.DeleteObject(objPath)
 	if err != nil {
 		t.Errorf("Failed to delete object: %s", err)
 	}