jochen
be92762969
The live tests reached one shared bus and assert, read and remove the mesh's own objects by their fixed names, so packages run in parallel deleted what each other read and the suite passed only one package at a time; a red suite read as noise. internal/testbus starts a server per test, linked in at the nats-server release go.mod pins, and a test holds that pin to the catalogue's bus image and to the facts snapshot's bus when there is one, so the tests never run a bus the mesh does not. The waiter test read a timing (the most connections held at one look) and now reads the state it means (the fewest held across the wait). make check runs the packages in parallel under the race detector, with a timeout.
112 lines
2.1 KiB
Go
112 lines
2.1 KiB
Go
//go:build enable_antithesis_sdk
|
|
|
|
package assert
|
|
|
|
import (
|
|
"runtime"
|
|
"sync"
|
|
"sync/atomic"
|
|
|
|
"github.com/antithesishq/antithesis-sdk-go/internal"
|
|
)
|
|
|
|
type trackerInfo struct {
|
|
Filename string
|
|
Classname string
|
|
PassCount int
|
|
FailCount int
|
|
}
|
|
|
|
type emitTracker map[string]*trackerInfo
|
|
|
|
// assert_tracker (global) keeps track of the unique asserts evaluated
|
|
var (
|
|
assertTracker emitTracker = make(emitTracker)
|
|
trackerMutex sync.Mutex
|
|
trackerInfoMutex sync.Mutex
|
|
)
|
|
|
|
func (tracker emitTracker) getTrackerEntry(messageKey string, filename, classname string) *trackerInfo {
|
|
var trackerEntry *trackerInfo
|
|
var ok bool
|
|
|
|
if tracker == nil {
|
|
return nil
|
|
}
|
|
|
|
trackerMutex.Lock()
|
|
defer trackerMutex.Unlock()
|
|
if trackerEntry, ok = tracker[messageKey]; !ok {
|
|
trackerEntry = newTrackerInfo(filename, classname)
|
|
tracker[messageKey] = trackerEntry
|
|
}
|
|
return trackerEntry
|
|
}
|
|
|
|
func newTrackerInfo(filename, classname string) *trackerInfo {
|
|
trackerInfo := trackerInfo{
|
|
PassCount: 0,
|
|
FailCount: 0,
|
|
Filename: filename,
|
|
Classname: classname,
|
|
}
|
|
return &trackerInfo
|
|
}
|
|
|
|
func (ti *trackerInfo) emit(ai *assertInfo) {
|
|
if ti == nil || ai == nil {
|
|
return
|
|
}
|
|
|
|
// Registrations are just sent to voidstar
|
|
hit := ai.Hit
|
|
if !hit {
|
|
emitAssert(ai)
|
|
return
|
|
}
|
|
|
|
var err error
|
|
cond := ai.Condition
|
|
|
|
trackerInfoMutex.Lock()
|
|
defer trackerInfoMutex.Unlock()
|
|
if cond {
|
|
if ti.PassCount == 0 {
|
|
err = emitAssert(ai)
|
|
}
|
|
if err == nil {
|
|
ti.PassCount++
|
|
}
|
|
return
|
|
}
|
|
if ti.FailCount == 0 {
|
|
err = emitAssert(ai)
|
|
}
|
|
if err == nil {
|
|
ti.FailCount++
|
|
}
|
|
}
|
|
|
|
func versionMessage() {
|
|
languageBlock := map[string]any{
|
|
"name": "Go",
|
|
"version": runtime.Version(),
|
|
}
|
|
versionBlock := map[string]any{
|
|
"language": languageBlock,
|
|
"sdk_version": internal.SDK_Version,
|
|
"protocol_version": internal.Protocol_Version,
|
|
}
|
|
internal.Json_data(map[string]any{"antithesis_sdk": versionBlock})
|
|
}
|
|
|
|
// package-level flag
|
|
var hasEmitted atomic.Bool // initialzed to false
|
|
|
|
func emitAssert(ai *assertInfo) error {
|
|
if hasEmitted.CompareAndSwap(false, true) {
|
|
versionMessage()
|
|
}
|
|
return internal.Json_data(wrappedAssertInfo{ai})
|
|
}
|