Empty row count returns error instead (#20)

* Fix default mode set KV V2 bug

* Fix row count stmt when it's returning emtpy

* Apply Code Coverage Badge

---------

Co-authored-by: edocsss <[email protected]>

Author: edocsss

README.md

@@ -19,7 +19,7 @@
 
   ![Unit Test](https://github.com/FreeLeh/GoFreeDB/actions/workflows/unit_test.yml/badge.svg)
   ![Integration Test](https://github.com/FreeLeh/GoFreeDB/actions/workflows/full_test.yml/badge.svg)
-![Coverage](https://img.shields.io/badge/Coverage-83.0%25-brightgreen)
+![Coverage](https://img.shields.io/badge/Coverage-83.1%25-brightgreen)
   [![Go Report Card](https://goreportcard.com/badge/github.com/FreeLeh/GoFreeDB)](https://goreportcard.com/report/github.com/FreeLeh/GoFreeDB)
   [![Go Reference](https://pkg.go.dev/badge/github.com/FreeLeh/GoFreeDB.svg)](https://pkg.go.dev/github.com/FreeLeh/GoFreeDB)
 

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,12 @@ 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, nil
+	}
 	if len(result.Rows) != 1 || len(result.Rows[0]) != 1 {
-		return 0, errors.New("")
+		return 0, errors.New("unexpected number of rows or columns")
 	}
 
 	count := result.Rows[0][0].(float64)