From c4ce5e86234e9f91599b87cff032228adb10fd50 Mon Sep 17 00:00:00 2001
From: Julien Dessaux
Date: Sat, 5 Oct 2024 09:54:35 +0200
Subject: fix(tfstated): add delete handler tests

---
 pkg/database/states.go | 14 +++++++++++---
 1 file changed, 11 insertions(+), 3 deletions(-)

(limited to 'pkg/database')

diff --git a/pkg/database/states.go b/pkg/database/states.go
index e848924..46a5536 100644
--- a/pkg/database/states.go
+++ b/pkg/database/states.go
@@ -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) {
-- 
cgit v1.2.3