updating golden-files

roderm · Nov 17, 2021 · 1f2d574 · 1f2d574
1 parent ab1534f
commit 1f2d574
Show file tree

Hide file tree

Showing 6 changed files with 29 additions and 3 deletions.
diff --git a/test/proto/join/test.sqlmap.go b/test/proto/join/test.sqlmap.go
@@ -37,6 +37,10 @@ func (m *Employee) Value() (driver.Value, error) {
 	return m.EmployeeID, nil
 }
 
+func (m *Employee) GetIdentifier() interface{} {
+	return m.EmployeeID
+}
+
 type queryEmployeeConfig struct {
 	Store        *TestStore
 	filter       pg.Where
@@ -80,9 +84,11 @@ func EmployeeWithManager(opts ...EmployeeOption) EmployeeOption {
 			EmployeeOnRow(func(row *Employee) {
 
 				// one-to-one
-				item := mapManager[row.EmployeeID]
-				if config.rows[item.EmployeeID] != nil {
-					config.rows[item.EmployeeID].Manager = row
+				item, ok := mapManager[row.EmployeeID]
+				if ok && item != nil {
+					if config.rows[item.EmployeeID] != nil {
+						config.rows[item.EmployeeID].Manager = row
+					}
 				}
 
 			}),

diff --git a/test/proto/jsonb/test.sqlmap.go b/test/proto/jsonb/test.sqlmap.go
@@ -37,6 +37,10 @@ func (m *Product) Value() (driver.Value, error) {
 	return m.ProductID, nil
 }
 
+func (m *Product) GetIdentifier() interface{} {
+	return m.ProductID
+}
+
 type queryProductConfig struct {
 	Store        *TestStore
 	filter       pg.Where

diff --git a/test/proto/simple/create/test.sqlmap.go b/test/proto/simple/create/test.sqlmap.go
@@ -37,6 +37,10 @@ func (m *Employee) Value() (driver.Value, error) {
 	return m.Id, nil
 }
 
+func (m *Employee) GetIdentifier() interface{} {
+	return m.Id
+}
+
 func (m *Employee) Insert(s *TestStore, ctx context.Context) error {
 	ins := pg.NewInsert()
 	ins.Add(m.Firstname, m.Lastname)

diff --git a/test/proto/simple/delete/test.sqlmap.go b/test/proto/simple/delete/test.sqlmap.go
@@ -37,6 +37,10 @@ func (m *Employee) Value() (driver.Value, error) {
 	return m.Id, nil
 }
 
+func (m *Employee) GetIdentifier() interface{} {
+	return m.Id
+}
+
 func (m *Employee) Delete(s *TestStore, ctx context.Context) error {
 
 	stmt, err := s.conn.PrepareContext(ctx, `

diff --git a/test/proto/simple/read/test.sqlmap.go b/test/proto/simple/read/test.sqlmap.go
@@ -37,6 +37,10 @@ func (m *Employee) Value() (driver.Value, error) {
 	return m.Id, nil
 }
 
+func (m *Employee) GetIdentifier() interface{} {
+	return m.Id
+}
+
 type queryEmployeeConfig struct {
 	Store        *TestStore
 	filter       pg.Where

diff --git a/test/proto/simple/update/test.sqlmap.go b/test/proto/simple/update/test.sqlmap.go
@@ -37,6 +37,10 @@ func (m *Employee) Value() (driver.Value, error) {
 	return m.Id, nil
 }
 
+func (m *Employee) GetIdentifier() interface{} {
+	return m.Id
+}
+
 func (m *Employee) Update(s *TestStore, ctx context.Context, conf *pg.UpdateSQL) error {
 	base := 1
 	if conf == nil {