for completeness, also migrate the path-capability storage maps

onflow · Jun 11, 2024 · 10c9308 · 10c9308
1 parent 145dea1
commit 10c9308
Show file tree

Hide file tree

Showing 2 changed files with 43 additions and 9 deletions.
diff --git a/migrations/migration.go b/migrations/migration.go
@@ -127,6 +127,12 @@ func (m *StorageMigration) Migrate(migrator StorageMapKeyMigrator) {
 		stdlib.CapabilityControllerStorageDomain,
 		migrator,
 	)
+
+	accountStorage.MigrateStringKeys(
+		m.interpreter,
+		stdlib.PathCapabilityStorageDomain,
+		migrator,
+	)
 }
 
 func (m *StorageMigration) NewValueMigrationsPathMigrator(

diff --git a/migrations/migration_test.go b/migrations/migration_test.go
@@ -242,6 +242,9 @@ func (testCapConMigration) Migrate(
 			value.BorrowType,
 			value.CapabilityID+10,
 		), nil
+
+	case interpreter.UInt64Value:
+		return value + 10, nil
 	}
 
 	return nil, nil
@@ -798,28 +801,40 @@ func TestCapConMigration(t *testing.T) {
 	})
 	require.NoError(t, err)
 
-	storageMap := storage.GetStorageMap(
+	capConStorageMap := storage.GetStorageMap(
 		testAddress,
 		stdlib.CapabilityControllerStorageDomain,
 		false,
 	)
 
-	assert.Equal(t, uint64(2), storageMap.Count())
+	assert.Equal(t, uint64(2), capConStorageMap.Count())
 
-	controller1 := storageMap.ReadValue(nil, interpreter.Uint64StorageMapKey(1))
+	controller1 := capConStorageMap.ReadValue(nil, interpreter.Uint64StorageMapKey(1))
 	require.IsType(t, &interpreter.StorageCapabilityControllerValue{}, controller1)
 	assert.Equal(t,
 		interpreter.UInt64Value(1),
 		controller1.(*interpreter.StorageCapabilityControllerValue).CapabilityID,
 	)
 
-	controller2 := storageMap.ReadValue(nil, interpreter.Uint64StorageMapKey(2))
+	controller2 := capConStorageMap.ReadValue(nil, interpreter.Uint64StorageMapKey(2))
 	require.IsType(t, &interpreter.AccountCapabilityControllerValue{}, controller2)
 	assert.Equal(t,
 		interpreter.UInt64Value(2),
 		controller2.(*interpreter.AccountCapabilityControllerValue).CapabilityID,
 	)
 
+	pathCapStorageMap := storage.GetStorageMap(
+		testAddress,
+		stdlib.PathCapabilityStorageDomain,
+		false,
+	)
+
+	capSet := pathCapStorageMap.ReadValue(nil, interpreter.StringStorageMapKey("foo"))
+	require.IsType(t, &interpreter.DictionaryValue{}, capSet)
+	capSetDict := capSet.(*interpreter.DictionaryValue)
+	assert.Equal(t, 1, capSetDict.Count())
+	assert.True(t, bool(capSetDict.ContainsKey(inter, emptyLocationRange, interpreter.UInt64Value(1))))
+
 	// Migrate
 
 	migration, err := NewStorageMigration(inter, storage, "test", testAddress)
@@ -839,34 +854,47 @@ func TestCapConMigration(t *testing.T) {
 
 	// Assert
 
-	assert.Len(t, reporter.migrated, 2)
+	assert.Len(t, reporter.migrated, 3)
 
 	require.Empty(t, reporter.errors)
 
 	err = storage.CheckHealth()
 	require.NoError(t, err)
 
-	storageMap = storage.GetStorageMap(
+	capConStorageMap = storage.GetStorageMap(
 		testAddress,
 		stdlib.CapabilityControllerStorageDomain,
 		false,
 	)
 
-	assert.Equal(t, uint64(2), storageMap.Count())
+	assert.Equal(t, uint64(2), capConStorageMap.Count())
 
-	controller1 = storageMap.ReadValue(nil, interpreter.Uint64StorageMapKey(1))
+	controller1 = capConStorageMap.ReadValue(nil, interpreter.Uint64StorageMapKey(1))
 	require.IsType(t, &interpreter.StorageCapabilityControllerValue{}, controller1)
 	assert.Equal(t,
 		interpreter.UInt64Value(11),
 		controller1.(*interpreter.StorageCapabilityControllerValue).CapabilityID,
 	)
 
-	controller2 = storageMap.ReadValue(nil, interpreter.Uint64StorageMapKey(2))
+	controller2 = capConStorageMap.ReadValue(nil, interpreter.Uint64StorageMapKey(2))
 	require.IsType(t, &interpreter.AccountCapabilityControllerValue{}, controller2)
 	assert.Equal(t,
 		interpreter.UInt64Value(12),
 		controller2.(*interpreter.AccountCapabilityControllerValue).CapabilityID,
 	)
+
+	pathCapStorageMap = storage.GetStorageMap(
+		testAddress,
+		stdlib.PathCapabilityStorageDomain,
+		false,
+	)
+
+	capSet = pathCapStorageMap.ReadValue(nil, interpreter.StringStorageMapKey("foo"))
+	require.IsType(t, &interpreter.DictionaryValue{}, capSet)
+	capSetDict = capSet.(*interpreter.DictionaryValue)
+	assert.Equal(t, 1, capSetDict.Count())
+	assert.True(t, bool(capSetDict.ContainsKey(inter, emptyLocationRange, interpreter.UInt64Value(11))))
+
 }
 
 func TestContractMigration(t *testing.T) {