init

2024-07-19 20:02:18 +08:00
parent d1d427057e
commit dbbf6133be
26 changed files with 2171 additions and 0 deletions
@@ -0,0 +1,194 @@
+package parsers
+
+import (
+	"caatsm/internal/config"
+	"caatsm/internal/domain"
+	"regexp"
+
+	. "github.com/onsi/ginkgo/v2"
+	. "github.com/onsi/gomega"
+)
+
+var _ = Describe("AFTN Parser", func() {
+	var testConfig *config.Config
+
+	BeforeEach(func() {
+		testConfig = &config.Config{
+			Body: []config.BodyConfig{
+				{
+					Name: "ARR",
+					Patterns: []config.PatternConfig{
+						{
+							Pattern:    `^\((?P<type>[A-Z]{3})\-(?P<number>[A-Z0-9]+)\-(?P<ssr>[A-Z0-9]+)\-(?P<departure>[A-Z]{4})\-(?P<arrival>[A-Z]{4})\)$`,
+							Expression: regexp.MustCompile(`^\((?P<type>[A-Z]{3})\-(?P<number>[A-Z0-9]+)\-(?P<ssr>[A-Z0-9]+)\-(?P<departure>[A-Z]{4})\-(?P<arrival>[A-Z]{4})\)$`),
+						},
+					},
+				},
+				{
+					Name: "DEP",
+					Patterns: []config.PatternConfig{
+						{
+							Pattern:    `^\((?P<type>[A-Z]{3})\-(?P<number>[A-Z0-9]+)\-(?P<ssr>[A-Z0-9]+)\-(?P<departure>[A-Z]{4})\-(?P<departure_time>\d{4})\-(?P<arrival>[A-Z]{4})\)$`,
+							Expression: regexp.MustCompile(`^\((?P<type>[A-Z]{3})\-(?P<number>[A-Z0-9]+)\-(?P<ssr>[A-Z0-9]+)\-(?P<departure>[A-Z]{4})\-(?P<departure_time>\d{4})\-(?P<arrival>[A-Z]{4})\)$`),
+						},
+					},
+				},
+			},
+		}
+		config.SetMyConfig(testConfig)
+	})
+
+	Describe("Parse", func() {
+		It("should parse ARR messages correctly", func() {
+			message := "(ARR-AB123-SSR1234-KJFK-KLAX)"
+			parser := AFTNParser{}
+			parsedMessage, err := parser.Parse(message)
+			Expect(err).NotTo(HaveOccurred())
+			Expect(parsedMessage).To(BeAssignableToTypeOf(&domain.ARR{}))
+			arrMessage := parsedMessage.(*domain.ARR)
+			Expect(arrMessage.Category).To(Equal("ARR"))
+			Expect(arrMessage.AircraftID).To(Equal("AB123"))
+			Expect(arrMessage.SSRModeAndCode).To(Equal("SSR1234"))
+			Expect(arrMessage.DepartureAirport).To(Equal("KJFK"))
+			Expect(arrMessage.ArrivalAirport).To(Equal("KLAX"))
+		})
+
+		It("should parse DEP messages correctly", func() {
+			message := "(DEP-AB123-SSR1234-KJFK-1500-KLAX)"
+			parser := AFTNParser{}
+			parsedMessage, err := parser.Parse(message)
+			Expect(err).NotTo(HaveOccurred())
+			Expect(parsedMessage).To(BeAssignableToTypeOf(&domain.DEP{}))
+			depMessage := parsedMessage.(*domain.DEP)
+			Expect(depMessage.AircraftID).To(Equal("AB123"))
+			Expect(depMessage.SSRModeAndCode).To(Equal("SSR1234"))
+			Expect(depMessage.DepartureAirport).To(Equal("KJFK"))
+			Expect(depMessage.DepartureTime).To(Equal("1500"))
+			Expect(depMessage.Destination).To(Equal("KLAX"))
+		})
+
+		It("should return an error for invalid message types", func() {
+			message := "(XYZ-AB123-SSR1234-KJFK-KLAX)"
+			parser := AFTNParser{}
+			_, err := parser.Parse(message)
+			Expect(err).To(HaveOccurred())
+			Expect(err.Error()).To(Equal("invalid message type"))
+		})
+	})
+
+	Describe("ParseAFTN", func() {
+		It("should parse a valid AFTN message", func() {
+			rawMessage := `ZCZC TMQ2611 151524
+FF SENDERAA
+151524 RECEIVER
+(ARR-AB123-SSR1234-KJFK-KLAX)`
+
+			aftnMessage, err := ParseAFTN(rawMessage)
+			Expect(err).NotTo(HaveOccurred())
+			Expect(aftnMessage).NotTo(BeNil())
+			Expect(aftnMessage.Header.StartSignal).To(Equal("ZCZC"))
+			Expect(aftnMessage.Header.SendID).To(Equal("TMQ2611"))
+			Expect(aftnMessage.Header.SendTime).To(Equal("151524"))
+			Expect(aftnMessage.Category).To(Equal("ARR"))
+		})
+
+		It("should return an error for invalid AFTN message format", func() {
+			rawMessage := `ZCZC
+TMQ2611
+151524`
+
+			_, err := ParseAFTN(rawMessage)
+			Expect(err).To(HaveOccurred())
+			Expect(err.Error()).To(Equal("invalid AFTN message format"))
+		})
+	})
+
+	Describe("ValidateAFTN", func() {
+		It("should validate a valid AFTN message", func() {
+			aftnMessage := &domain.AFTN{
+				Header: domain.Header{
+					StartSignal: "ZCZC",
+					SendID:      "TMQ2611",
+					SendTime:    "151524",
+				},
+				PriorityAndSender: domain.PriorityAndSender{
+					Priority: "FF",
+					Sender:   "SENDERAA",
+				},
+				TimeAndReceiver: domain.TimeAndReceiver{
+					Time:     "151524",
+					Receiver: "RECEIVER",
+				},
+				Category: "ARR",
+			}
+			err := ValidateAFTN(aftnMessage)
+			Expect(err).NotTo(HaveOccurred())
+		})
+
+		It("should return an error for missing required fields", func() {
+			aftnMessage := &domain.AFTN{
+				Header: domain.Header{
+					StartSignal: "",
+					SendID:      "TMQ2611",
+					SendTime:    "151524",
+				},
+				PriorityAndSender: domain.PriorityAndSender{
+					Priority: "FF",
+					Sender:   "",
+				},
+				TimeAndReceiver: domain.TimeAndReceiver{
+					Time:     "151524",
+					Receiver: "RECEIVER",
+				},
+				Category: "ARR",
+			}
+			err := ValidateAFTN(aftnMessage)
+			Expect(err).To(HaveOccurred())
+			Expect(err.Error()).To(Equal("invalid AFTN message: missing fields"))
+		})
+
+		It("should return an error for invalid priority code", func() {
+			aftnMessage := &domain.AFTN{
+				Header: domain.Header{
+					StartSignal: "ZCZC",
+					SendID:      "TMQ2611",
+					SendTime:    "151524",
+				},
+				PriorityAndSender: domain.PriorityAndSender{
+					Priority: "ZZ",
+					Sender:   "SENDERAA",
+				},
+				TimeAndReceiver: domain.TimeAndReceiver{
+					Time:     "151524",
+					Receiver: "RECEIVER",
+				},
+				Category: "ARR",
+			}
+			err := ValidateAFTN(aftnMessage)
+			Expect(err).To(HaveOccurred())
+			Expect(err.Error()).To(Equal("invalid priority code"))
+		})
+
+		It("should return an error for invalid address format", func() {
+			aftnMessage := &domain.AFTN{
+				Header: domain.Header{
+					StartSignal: "ZCZC",
+					SendID:      "TMQ2611",
+					SendTime:    "151524",
+				},
+				PriorityAndSender: domain.PriorityAndSender{
+					Priority: "FF",
+					Sender:   "INVALID",
+				},
+				TimeAndReceiver: domain.TimeAndReceiver{
+					Time:     "151524",
+					Receiver: "RECEIVERAA",
+				},
+				Category: "ARR",
+			}
+			err := ValidateAFTN(aftnMessage)
+			Expect(err).To(HaveOccurred())
+			Expect(err.Error()).To(Equal("invalid address format"))
+		})
+	})
+})