You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
33 lines
949 B
33 lines
949 B
package testutil
|
|
|
|
import (
|
|
"testing"
|
|
|
|
kschema "github.com/seaweedfs/seaweedfs/weed/mq/kafka/schema"
|
|
)
|
|
|
|
// EnsureValueSchema registers a minimal Avro value schema for the given topic if not present.
|
|
// Returns the latest schema ID if successful.
|
|
func EnsureValueSchema(t *testing.T, registryURL, topic string) (uint32, error) {
|
|
t.Helper()
|
|
subject := topic + "-value"
|
|
rc := kschema.NewRegistryClient(kschema.RegistryConfig{URL: registryURL})
|
|
|
|
// Minimal Avro record schema with string field "value"
|
|
schemaJSON := `{"type":"record","name":"TestRecord","fields":[{"name":"value","type":"string"}]}`
|
|
|
|
// Try to get existing
|
|
if latest, err := rc.GetLatestSchema(subject); err == nil {
|
|
return latest.LatestID, nil
|
|
}
|
|
|
|
// Register and fetch latest
|
|
if _, err := rc.RegisterSchema(subject, schemaJSON); err != nil {
|
|
return 0, err
|
|
}
|
|
latest, err := rc.GetLatestSchema(subject)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
return latest.LatestID, nil
|
|
}
|