windyboy
53f03bbc0a
The code changes in config.dev.toml and sub.go update the configuration and message handling for integrating with Hasura. The config.dev.toml file now includes a new section for Hasura configuration, specifying the endpoint and secret. In sub.go, the UUID field is set to the UUID of the received message, ensuring proper tracking of parsed messages. This refactor enables seamless integration with Hasura and improves the functionality of the code.
113 lines
2.8 KiB
Go
113 lines
2.8 KiB
Go
package config
|
|
|
|
import (
|
|
"os"
|
|
"testing"
|
|
|
|
. "github.com/onsi/ginkgo/v2"
|
|
. "github.com/onsi/gomega"
|
|
"github.com/spf13/viper"
|
|
)
|
|
|
|
func TestConfig(t *testing.T) {
|
|
RegisterFailHandler(Fail)
|
|
RunSpecs(t, "Config Suite")
|
|
}
|
|
|
|
var _ = Describe("Config", func() {
|
|
var originalEnv string
|
|
|
|
BeforeEach(func() {
|
|
// Save the original GO_ENV value
|
|
originalEnv = os.Getenv("GO_ENV")
|
|
// Set up a temporary configuration file for testing
|
|
viper.Reset()
|
|
viper.SetConfigType("toml")
|
|
configContent := `
|
|
[nats]
|
|
client = "test-client"
|
|
url = "nats://localhost:4222"
|
|
cluster = "test-cluster"
|
|
|
|
[subscription]
|
|
topic = "example-topic"
|
|
queue_group = "example-group"
|
|
|
|
[timeouts]
|
|
server_timeout = "30s"
|
|
reconnect_wait = "10s"
|
|
close_timeout = "10s"
|
|
ack_wait_timeout = "5s"
|
|
|
|
[hasura]
|
|
endpoint = "http://localhost:8080/v1/graphql"
|
|
secret = "aviation-test"
|
|
`
|
|
tmpFile, err := os.CreateTemp("", "config.*.toml")
|
|
Expect(err).NotTo(HaveOccurred())
|
|
_, err = tmpFile.Write([]byte(configContent))
|
|
Expect(err).NotTo(HaveOccurred())
|
|
err = tmpFile.Close()
|
|
Expect(err).NotTo(HaveOccurred())
|
|
|
|
viper.SetConfigFile(tmpFile.Name())
|
|
err = viper.ReadInConfig()
|
|
Expect(err).NotTo(HaveOccurred())
|
|
|
|
// Load the configuration
|
|
MyConfig = &Config{}
|
|
err = viper.Unmarshal(MyConfig)
|
|
Expect(err).NotTo(HaveOccurred())
|
|
})
|
|
|
|
AfterEach(func() {
|
|
// Restore the original GO_ENV value
|
|
os.Setenv("GO_ENV", originalEnv)
|
|
})
|
|
|
|
Context("Loading configuration", func() {
|
|
It("should load the configuration correctly", func() {
|
|
cfg := GetMyConfig()
|
|
Expect(cfg).NotTo(BeNil())
|
|
Expect(cfg.Nats.Client).To(Equal("test-client"))
|
|
Expect(cfg.Nats.URL).To(Equal("nats://localhost:4222"))
|
|
Expect(cfg.Subscription.Topic).To(Equal("example-topic"))
|
|
Expect(cfg.Subscription.QueueGroup).To(Equal("example-group"))
|
|
Expect(cfg.Hasura.Endpoint).To(Equal("http://localhost:8080/v1/graphql"))
|
|
Expect(cfg.Hasura.Secret).To(Equal("aviation-test"))
|
|
})
|
|
})
|
|
|
|
Context("Validating configuration", func() {
|
|
It("should validate a valid configuration", func() {
|
|
cfg := GetMyConfig()
|
|
err := ValidateConfig(cfg)
|
|
Expect(err).NotTo(HaveOccurred())
|
|
})
|
|
|
|
It("should return an error for missing NATS client", func() {
|
|
cfg := GetMyConfig()
|
|
cfg.Nats.Client = ""
|
|
err := ValidateConfig(cfg)
|
|
Expect(err).To(HaveOccurred())
|
|
Expect(err.Error()).To(Equal("nats client is required"))
|
|
})
|
|
|
|
It("should return an error for missing NATS URL", func() {
|
|
cfg := GetMyConfig()
|
|
cfg.Nats.URL = ""
|
|
err := ValidateConfig(cfg)
|
|
Expect(err).To(HaveOccurred())
|
|
Expect(err.Error()).To(Equal("nats URL is required"))
|
|
})
|
|
|
|
It("should return an error for missing subscription topic", func() {
|
|
cfg := GetMyConfig()
|
|
cfg.Subscription.Topic = ""
|
|
err := ValidateConfig(cfg)
|
|
Expect(err).To(HaveOccurred())
|
|
Expect(err.Error()).To(Equal("subscription topic is required"))
|
|
})
|
|
})
|
|
})
|