Method handle

internal_maint_notif_test.go:502–545

(c net.Conn)

}

func (s *mockMaintNotificationsDowngradeServer) handle(c net.Conn) {
	defer c.Close()
	r := bufio.NewReader(c)
	for {
		args, err := readRESPCommand(r)
		if err != nil {
			return
		}
		if len(args) == 0 {
			continue
		}
		name := strings.ToUpper(args[0])
		full := name
		if len(args) > 1 {
			full = name + " " + strings.ToUpper(args[1])
		}

		switch {
		case name == "HELLO":
			s.record(full)
			_, _ = c.Write([]byte("%1\r\n+proto\r\n:3\r\n"))
		case full == maintNotificationsCommand:
			if call := s.record(full); call == 1 {
				_, _ = c.Write([]byte("+OK\r\n"))
			} else {
				_, _ = c.Write([]byte("-ERR unknown subcommand 'MAINT_NOTIFICATIONS'\r\n"))
			}
		case name == "PING":
			s.record(full)
			shouldBlock := false
			s.blockFirstPing.Do(func() {
				shouldBlock = true
				close(s.firstPingBlocked)
			})
			if shouldBlock {
				<-s.releaseFirstPing
			}
			_, _ = c.Write([]byte("+PONG\r\n"))
		default:
			s.record(full)
			_, _ = c.Write([]byte("+OK\r\n"))
		}
	}
}

func startMockMaintNotificationsDowngradeServer(t *testing.T) *mockMaintNotificationsDowngradeServer {
	t.Helper()