test: single table migration as transaction (#736)

ory · Sep 28, 2021 · 9eda48c · 9eda48c
1 parent 9f3e8c6
commit 9eda48c
Showing 1 changed file with 14 additions and 1 deletion.
diff --git a/internal/persistence/sql/migrations/single_table_test.go b/internal/persistence/sql/migrations/single_table_test.go
@@ -126,6 +126,12 @@ func TestToSingleTableMigrator(t *testing.T) {
 			t.Run("case=non-deserializable tuple", func(t *testing.T) {
 				n := setup(t)
 
+				valid := &relationtuple.InternalRelationTuple{
+					Namespace: n.Name,
+					Object:    "o1",
+					Relation:  "r",
+					Subject:   &relationtuple.SubjectID{ID: "s"},
+				}
 				require.NoError(t, m.insertOldRelationTuples(ctx, n, &relationtuple.InternalRelationTuple{
 					Namespace: n.Name,
 					Object:    "o0",
@@ -136,12 +142,19 @@ func TestToSingleTableMigrator(t *testing.T) {
 					Object:    "o1",
 					Relation:  "r",
 					Subject:   &relationtuple.SubjectID{ID: "invalid#subject-id"},
-				}))
+				}, valid))
 				err := m.MigrateNamespace(ctx, n)
 				require.Error(t, err)
 				invalid, ok := err.(ErrInvalidTuples)
 				require.True(t, ok)
 				assert.Len(t, invalid, 2)
+
+				rts, next, err := r.Persister().GetRelationTuples(ctx, &relationtuple.RelationQuery{
+					Namespace: n.Name,
+				})
+				require.NoError(t, err)
+				require.Equal(t, "", next)
+				assert.Equal(t, []*relationtuple.InternalRelationTuple{valid}, rts)
 			})
 		})
 	}