assert dictionary size

onflow · Apr 23, 2024 · 5504380 · 5504380
1 parent 8faed59
commit 5504380
Show file tree

Hide file tree

Showing 4 changed files with 14 additions and 0 deletions.
diff --git a/migrations/entitlements/migration_test.go b/migrations/entitlements/migration_test.go
@@ -3311,6 +3311,8 @@ func TestRehash(t *testing.T) {
 			optType.ID(),
 		)
 
+		assert.Equal(t, 1, dictValue.Count())
+
 		value, ok := dictValue.Get(inter, locationRange, typeValue)
 		require.True(t, ok)
 

diff --git a/migrations/statictypes/intersection_type_migration_test.go b/migrations/statictypes/intersection_type_migration_test.go
@@ -628,6 +628,8 @@ func TestIntersectionTypeRehash(t *testing.T) {
 			intersectionType.ID(),
 		)
 
+		assert.Equal(t, 1, dictValue.Count())
+
 		value, ok := dictValue.Get(inter, locationRange, typeValue)
 		require.True(t, ok)
 
@@ -800,6 +802,8 @@ func TestRehashNestedIntersectionType(t *testing.T) {
 				intersectionType.ID(),
 			)
 
+			assert.Equal(t, 1, dictValue.Count())
+
 			value, ok := dictValue.Get(inter, locationRange, typeValue)
 			require.True(t, ok)
 
@@ -948,6 +952,8 @@ func TestRehashNestedIntersectionType(t *testing.T) {
 				intersectionType.ID(),
 			)
 
+			assert.Equal(t, 1, dictValue.Count())
+
 			value, ok := dictValue.Get(inter, locationRange, typeValue)
 			require.True(t, ok)
 

diff --git a/migrations/statictypes/statictype_migration_test.go b/migrations/statictypes/statictype_migration_test.go
@@ -1393,6 +1393,8 @@ func TestOptionalTypeRehash(t *testing.T) {
 			optionalType.ID(),
 		)
 
+		assert.Equal(t, 1, dictValue.Count())
+
 		value, ok := dictValue.Get(inter, locationRange, typeValue)
 		require.True(t, ok)
 

diff --git a/migrations/string_normalization/migration_test.go b/migrations/string_normalization/migration_test.go
@@ -488,6 +488,8 @@ func TestStringValueRehash(t *testing.T) {
 			stringValue.HashInput(inter, locationRange, nil),
 		)
 
+		assert.Equal(t, 1, dictValue.Count())
+
 		value, ok := dictValue.Get(inter, locationRange, stringValue)
 		require.True(t, ok)
 
@@ -633,6 +635,8 @@ func TestCharacterValueRehash(t *testing.T) {
 			characterValue.HashInput(inter, locationRange, nil),
 		)
 
+		assert.Equal(t, 1, dictValue.Count())
+
 		value, ok := dictValue.Get(inter, locationRange, characterValue)
 		require.True(t, ok)