windyboy
969bb57902
This commit adds functionality to parse AFTN messages in the `aftn_parser.go` file. The `Parse` method now correctly parses the message based on its type and extracts the relevant data. Additionally, the `FindPatterns` and `ParseBody` functions have been updated to remove the dependency on the `config` package and instead use the `config.GetBodyPatterns` function to retrieve the body patterns. This change improves the modularity and maintainability of the code. Note: This commit message assumes that the `config` package has been refactored to use the `GetBodyPatterns` function.
106 lines
2.5 KiB
Go
106 lines
2.5 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"
|
|
`
|
|
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"))
|
|
})
|
|
})
|
|
|
|
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"))
|
|
})
|
|
})
|
|
})
|