gorm/model_struct_test.go

package gorm_test

import (
	"sync"
	"testing"

	"github.com/jinzhu/gorm"
)

type ModelA struct {
	gorm.Model
	Name string

	ModelCs []ModelC `gorm:"foreignkey:OtherAID"`
}

type ModelB struct {
	gorm.Model
	Name string

	ModelCs []ModelC `gorm:"foreignkey:OtherBID"`
}

type ModelC struct {
	gorm.Model
	Name string

	OtherAID uint64
	OtherA   *ModelA `gorm:"foreignkey:OtherAID"`
	OtherBID uint64
	OtherB   *ModelB `gorm:"foreignkey:OtherBID"`
}

// This test will try to cause a race condition on the model's foreignkey metadata
func TestModelStructRaceSameModel(t *testing.T) {
	// use a WaitGroup to execute as much in-sync as possible
	// it's more likely to hit a race condition than without
	n := 32
	start := sync.WaitGroup{}
	start.Add(n)

	// use another WaitGroup to know when the test is done
	done := sync.WaitGroup{}
	done.Add(n)

	for i := 0; i < n; i++ {
		go func() {
			start.Wait()

			// call GetStructFields, this had a race condition before we fixed it
			DB.NewScope(&ModelA{}).GetStructFields()

			done.Done()
		}()

		start.Done()
	}

	done.Wait()
}

// This test will try to cause a race condition on the model's foreignkey metadata
func TestModelStructRaceDifferentModel(t *testing.T) {
	// use a WaitGroup to execute as much in-sync as possible
	// it's more likely to hit a race condition than without
	n := 32
	start := sync.WaitGroup{}
	start.Add(n)

	// use another WaitGroup to know when the test is done
	done := sync.WaitGroup{}
	done.Add(n)

	for i := 0; i < n; i++ {
		i := i
		go func() {
			start.Wait()

			// call GetStructFields, this had a race condition before we fixed it
			if i%2 == 0 {
				DB.NewScope(&ModelA{}).GetStructFields()
			} else {
				DB.NewScope(&ModelB{}).GetStructFields()
			}

			done.Done()
		}()

		start.Done()
	}

	done.Wait()
}