setting Unknown bucket type

internal/storage/gcs/bucket.go

@@ -27,6 +27,7 @@ type BucketTypes int
 const (
 	NonHierarchical BucketTypes = iota
 	Hierarchical
+	Unknown
 )
 
 // Bucket represents a GCS bucket, pre-bound with a bucket name and necessary

internal/storage/mock_control_client.go

@@ -30,7 +30,7 @@ func mockGetBucketType(controlClient *MockStorageControlClient, bucketName strin
 	}, nil...)
 
 	if err != nil {
-		return gcs.NonHierarchical
+		return gcs.Unknown
 	}
 
 	if stoargeLayout.GetHierarchicalNamespace() != nil && stoargeLayout.GetHierarchicalNamespace().Enabled {

internal/storage/storage_handle.go

@@ -214,9 +214,10 @@ func getBucketType(controlClient *control.StorageControlClient, bucketName strin
 		RequestId: "",
 	}, callOptions...)
 
+	// In case bucket does not exist, set type unknown instead of panic.
 	if err != nil {
 		log.Printf("GetStorageLayout: %v", err)
-		return gcs.NonHierarchical
+		return gcs.Unknown
 	}
 
 	if stoargeLayout.HierarchicalNamespace != nil && stoargeLayout.GetHierarchicalNamespace().Enabled {

internal/storage/storage_handle_test.go

@@ -332,5 +332,5 @@ func (testSuite *StorageHandleTest) TestGetBucketTypeWithError() {
 		Return(x, errors.New("mocked error"))
 
 	bucketType := mockGetBucketType(testSuite.mockClient, TestBucketName)
-	assert.Equal(testSuite.T(), gcs.NonHierarchical, bucketType, "Expected NonHierarchical when there's an error")
+	assert.Equal(testSuite.T(), gcs.Unknown, bucketType, "Expected Unknown when there's an error")
 }