Fix row count stmt when it's returning emtpy

Author: edocsss

internal/google/store/row_test.go

@@ -3,12 +3,13 @@ package store
 import (
 	"context"
 	"fmt"
-	"github.com/FreeLeh/GoFreeDB/internal/common"
-	"github.com/FreeLeh/GoFreeDB/internal/models"
 	"strconv"
 	"testing"
 	"time"
 
+	"github.com/FreeLeh/GoFreeDB/internal/common"
+	"github.com/FreeLeh/GoFreeDB/internal/models"
+
 	"github.com/FreeLeh/GoFreeDB/google/auth"
 	"github.com/stretchr/testify/assert"
 )
@@ -43,6 +44,11 @@ func TestGoogleSheetRowStore_Integration(t *testing.T) {
 		_ = db.Close(context.Background())
 	}()
 
+	time.Sleep(time.Second)
+	res, err := db.Count().Exec(context.Background())
+	assert.Equal(t, uint64(0), res)
+	assert.Nil(t, err)
+
 	var out []testPerson
 
 	time.Sleep(time.Second)

internal/google/store/stmt.go

@@ -4,12 +4,13 @@ import (
 	"context"
 	"errors"
 	"fmt"
-	"github.com/FreeLeh/GoFreeDB/internal/common"
-	"github.com/FreeLeh/GoFreeDB/internal/models"
 	"reflect"
 	"strconv"
 	"strings"
 
+	"github.com/FreeLeh/GoFreeDB/internal/common"
+	"github.com/FreeLeh/GoFreeDB/internal/models"
+
 	"github.com/FreeLeh/GoFreeDB/internal/google/sheets"
 )
 
@@ -604,8 +605,9 @@ func (s *GoogleSheetCountStmt) Exec(ctx context.Context) (uint64, error) {
 		return 0, err
 	}
 
+	// When COUNT() returns 0, somehow it returns empty row slice.
 	if len(result.Rows) != 1 || len(result.Rows[0]) != 1 {
-		return 0, errors.New("")
+		return 0, nil
 	}
 
 	count := result.Rows[0][0].(float64)