Skip to content

Commit

Permalink
sql: escape 'key'
Browse files Browse the repository at this point in the history
  • Loading branch information
ChrisSchinnerl committed Aug 13, 2024
1 parent 2d5b469 commit db33cb5
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions stores/metadata_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -3857,7 +3857,7 @@ func TestSlabHealthInvalidation(t *testing.T) {
t.Helper()

var validUntil int64
if err := ss.DB().QueryRow(context.Background(), "SELECT health_valid_until FROM slabs WHERE key = ?", sql.EncryptionKey(slabKey)).Scan(&validUntil); err != nil {
if err := ss.DB().QueryRow(context.Background(), "SELECT health_valid_until FROM slabs WHERE `key` = ?", sql.EncryptionKey(slabKey)).Scan(&validUntil); err != nil {
t.Fatal(err)
} else if valid := time.Now().Before(time.Unix(validUntil, 0)); valid != expected {
t.Fatal("unexpected health valid", valid)
Expand Down Expand Up @@ -4172,7 +4172,7 @@ func TestSlabCleanup(t *testing.T) {

// create a slab
var slabID int64
if res, err := ss.DB().Exec(context.Background(), "INSERT INTO slabs (db_contract_set_id, key, health_valid_until) VALUES (?, ?, ?);", csID, sql.EncryptionKey(object.GenerateEncryptionKey()), 100); err != nil {
if res, err := ss.DB().Exec(context.Background(), "INSERT INTO slabs (db_contract_set_id, `key`, health_valid_until) VALUES (?, ?, ?);", csID, sql.EncryptionKey(object.GenerateEncryptionKey()), 100); err != nil {
t.Fatal(err)
} else if slabID, err = res.LastInsertId(); err != nil {
t.Fatal(err)
Expand Down Expand Up @@ -4213,7 +4213,7 @@ func TestSlabCleanup(t *testing.T) {

// create another slab referencing the buffered slab
var bufferedSlabID int64
if res, err := ss.DB().Exec(context.Background(), "INSERT INTO slabs (db_buffered_slab_id, db_contract_set_id, key, health_valid_until) VALUES (?, ?, ?, ?);", bsID, csID, sql.EncryptionKey(object.GenerateEncryptionKey()), 100); err != nil {
if res, err := ss.DB().Exec(context.Background(), "INSERT INTO slabs (db_buffered_slab_id, db_contract_set_id, `key`, health_valid_until) VALUES (?, ?, ?, ?);", bsID, csID, sql.EncryptionKey(object.GenerateEncryptionKey()), 100); err != nil {
t.Fatal(err)
} else if bufferedSlabID, err = res.LastInsertId(); err != nil {
t.Fatal(err)
Expand Down

0 comments on commit db33cb5

Please sign in to comment.