refactor: changing return of methods

Author: Guilospanck

go.mod

@@ -7,13 +7,15 @@ require github.com/gocql/gocql v1.0.0
 require (
 	github.com/davecgh/go-spew v1.1.0 // indirect
 	github.com/pmezard/go-difflib v1.0.0 // indirect
+	github.com/scylladb/go-reflectx v1.0.1 // indirect
 	github.com/stretchr/objx v0.1.0 // indirect
 	gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c // indirect
 )
 
 require (
 	github.com/golang/snappy v0.0.3 // indirect
 	github.com/hailocab/go-hostpool v0.0.0-20160125115350-e80d13ce29ed // indirect
+	github.com/scylladb/gocqlx/v2 v2.7.0
 	github.com/stretchr/testify v1.7.1
 	gopkg.in/inf.v0 v0.9.1 // indirect
 )

go.sum

@@ -13,6 +13,10 @@ github.com/kr/pty v1.1.1/go.mod h1:pFQYn66WHrOpPYNljwOMqo10TkYh1fy3cYio2l3bCsQ=
 github.com/kr/text v0.1.0/go.mod h1:4Jbv+DJW3UT/LiOwJeYQe1efqtUx/iVham/4vfdArNI=
 github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
 github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
+github.com/scylladb/go-reflectx v1.0.1 h1:b917wZM7189pZdlND9PbIJ6NQxfDPfBvUaQ7cjj1iZQ=
+github.com/scylladb/go-reflectx v1.0.1/go.mod h1:rWnOfDIRWBGN0miMLIcoPt/Dhi2doCMZqwMCJ3KupFc=
+github.com/scylladb/gocqlx/v2 v2.7.0 h1:/w1VeJHCEAsg9eTculTvIS9eIe/VmEu0clhlH1CF7lc=
+github.com/scylladb/gocqlx/v2 v2.7.0/go.mod h1:jKhM0/LkEAhEOSwd10TCMQdlC5x8aEzK7cXjQcPyMJ0=
 github.com/stretchr/objx v0.1.0 h1:4G4v2dO3VZwixGIRoQ5Lfboy6nUhCyYzaqnIAPPhYs4=
 github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
 github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI=

iterx.go

@@ -3,6 +3,7 @@ package gocqlxmock
 import (
 	"reflect"
 
+	"github.com/gocql/gocql"
 	"github.com/stretchr/testify/mock"
 )
 
@@ -27,86 +28,152 @@ type IterxMock struct {
 	mock.Mock
 }
 
-func (mock *IterxMock) Unsafe() IIterx {
+func (mock IterxMock) Unsafe() IIterx {
 	args := mock.Called()
 
 	return args.Get(0).(IIterx)
 }
 
-func (mock *IterxMock) StructOnly() IIterx {
+func (mock IterxMock) StructOnly() IIterx {
 	args := mock.Called()
 
 	return args.Get(0).(IIterx)
 }
 
-func (mock *IterxMock) Get(dest interface{}) error {
+func (mock IterxMock) Get(dest interface{}) error {
 	args := mock.Called(dest)
 
 	return args.Error(0)
 }
 
-func (mock *IterxMock) scanAny(dest interface{}) bool {
+func (mock IterxMock) scanAny(dest interface{}) bool {
 	args := mock.Called(dest)
 
 	return args.Get(0).(bool)
 }
 
-func (mock *IterxMock) Select(dest interface{}) error {
+func (mock IterxMock) Select(dest interface{}) error {
 	args := mock.Called(dest)
 
 	return args.Error(0)
 }
 
-func (mock *IterxMock) scanAll(dest interface{}) bool {
+func (mock IterxMock) scanAll(dest interface{}) bool {
 	args := mock.Called(dest)
 
 	return args.Get(0).(bool)
 }
 
-func (mock *IterxMock) isScannable(t reflect.Type) bool {
+func (mock IterxMock) isScannable(t reflect.Type) bool {
 	args := mock.Called(t)
 
 	return args.Get(0).(bool)
 }
 
-func (mock *IterxMock) scan(value reflect.Value) bool {
+func (mock IterxMock) scan(value reflect.Value) bool {
 	args := mock.Called(value)
 
 	return args.Get(0).(bool)
 }
 
-func (mock *IterxMock) StructScan(dest interface{}) bool {
+func (mock IterxMock) StructScan(dest interface{}) bool {
 	args := mock.Called(dest)
 
 	return args.Get(0).(bool)
 }
 
-func (mock *IterxMock) structScan(value reflect.Value) bool {
+func (mock IterxMock) structScan(value reflect.Value) bool {
 	args := mock.Called(value)
 
 	return args.Get(0).(bool)
 }
 
-func (mock *IterxMock) fieldsByTraversal(value reflect.Value, traversals [][]int, values []interface{}) error {
+func (mock IterxMock) fieldsByTraversal(value reflect.Value, traversals [][]int, values []interface{}) error {
 	args := mock.Called(value, traversals, values)
 
 	return args.Error(0)
 }
 
-func (mock *IterxMock) Scan(dest ...interface{}) bool {
+func (mock IterxMock) Scan(dest ...interface{}) bool {
 	args := mock.Called(dest)
 
 	return args.Get(0).(bool)
 }
 
-func (mock *IterxMock) Close() error {
+func (mock IterxMock) Close() error {
 	args := mock.Called()
 
 	return args.Error(0)
 }
 
-func (mock *IterxMock) checkErrAndNotFound() error {
+func (mock IterxMock) checkErrAndNotFound() error {
 	args := mock.Called()
 
 	return args.Error(0)
 }
+
+// "Interface assertion"
+var (
+	_ IIterx = IterxMock{}
+	_ IIterx = iterx{}
+)
+
+type iterx struct {
+	i *gocql.Iter
+}
+
+func (i iterx) Unsafe() IIterx {
+	return i
+}
+
+func (i iterx) StructOnly() IIterx {
+	return i
+}
+
+func (i iterx) Get(dest interface{}) error {
+	return nil
+}
+
+func (i iterx) scanAny(dest interface{}) bool {
+	return true
+}
+
+func (i iterx) Select(dest interface{}) error {
+	return nil
+}
+
+func (i iterx) scanAll(dest interface{}) bool {
+	return true
+}
+
+func (i iterx) isScannable(t reflect.Type) bool {
+	return true
+}
+
+func (i iterx) scan(value reflect.Value) bool {
+	return true
+}
+
+func (i iterx) StructScan(dest interface{}) bool {
+	return true
+}
+
+func (i iterx) structScan(value reflect.Value) bool {
+	return true
+}
+
+func (i iterx) fieldsByTraversal(value reflect.Value, traversals [][]int, values []interface{}) error {
+	return nil
+}
+
+func (i iterx) Scan(dest ...interface{}) bool {
+	return true
+}
+
+func (i iterx) Close() error {
+	return nil
+}
+
+func (i iterx) checkErrAndNotFound() error {
+	return nil
+}