dialect/sql/sqlgraph: delete nodes in the graph

This commit is contained in:
Ariel Mashraki
2019-12-11 15:38:29 +02:00
parent ddd69ec90a
commit 9f324ce030
2 changed files with 51 additions and 1 deletions

View File

@@ -1124,6 +1124,23 @@ func TestUpdateNodes(t *testing.T) {
}
}
func TestDeleteNodes(t *testing.T) {
db, mock, err := sqlmock.New()
require.NoError(t, err)
mock.ExpectBegin()
mock.ExpectExec(escape("DELETE FROM `users`")).
WillReturnResult(sqlmock.NewResult(0, 2))
mock.ExpectCommit()
affected, err := DeleteNodes(context.Background(), OpenDB("", db), &DeleteSpec{
Node: &NodeSpec{
Table: "users",
ID: &FieldSpec{Column: "id", Type: field.TypeInt},
},
})
require.NoError(t, err)
require.Equal(t, 2, affected)
}
func escape(query string) string {
rows := strings.Split(query, "\n")
for i := range rows {