Function TestRevokeAPIKey_DatabaseError

database/api_key_test.go:206–236

(t *testing.T)

}

func TestRevokeAPIKey_DatabaseError(t *testing.T) {
	db, mock, err := sqlmock.New()
	assert.NoError(t, err)
	defer func() { _ = db.Close() }()

	ds := Datasource{Conn: db}

	apiKeyID := "api_key_123"
	ownerID := "owner123"
	hashedKey := "some-hashed-key-value"

	// Expect the SELECT query first (succeeds)
	selectRow := sqlmock.NewRows([]string{"key"}).
		AddRow(hashedKey)

	mock.ExpectQuery("SELECT.*FROM ledgerforge.api_keys.*WHERE api_key_id.*AND owner_id.*").
		WithArgs(apiKeyID, ownerID).
		WillReturnRows(selectRow)

	// Expect the UPDATE query (fails with database error)
	mock.ExpectExec("UPDATE ledgerforge.api_keys.*SET is_revoked.*WHERE api_key_id.*AND owner_id.*").
		WithArgs(sqlmock.AnyArg(), apiKeyID, ownerID).
		WillReturnError(sql.ErrConnDone)

	err = ds.RevokeAPIKey(context.Background(), apiKeyID, ownerID)
	assert.Error(t, err)
	assert.Equal(t, sql.ErrConnDone, err)

	err = mock.ExpectationsWereMet()
	assert.NoError(t, err)
}

func TestUpdateLastUsed_Success(t *testing.T) {
	db, mock, err := sqlmock.New()