fix(tfstated): add delete handler tests

This commit is contained in:
Julien Dessaux 2024-10-05 09:54:35 +02:00
parent 7ec826921c
commit c4ce5e8623
Signed by: adyxax
GPG key ID: F92E51B86E07177E
3 changed files with 58 additions and 13 deletions

View file

@ -5,9 +5,17 @@ import (
"fmt"
)
func (db *DB) DeleteState(name string) error {
_, err := db.Exec(`DELETE FROM states WHERE name = ?;`, name)
return err
// returns true in case of successful deletion
func (db *DB) DeleteState(name string) (bool, error) {
result, err := db.Exec(`DELETE FROM states WHERE name = ?;`, name)
if err != nil {
return false, err
}
n, err := result.RowsAffected()
if err != nil {
return false, err
}
return n == 1, nil
}
func (db *DB) GetState(name string) ([]byte, error) {