pfxhttp/postfix_test.go at main · croessner/pfxhttp · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
package main

import (
	"testing"
)

type MockNetString struct {
	data string
}

func (m *MockNetString) Length() uint16 {
	return 0
}

func (m *MockNetString) Data() []byte {
	return []byte(m.data)
}

func (m *MockNetString) String() string {
	return m.data
}

func TestReadNetString(t *testing.T) {
	tests := []struct {
		name         string
		netString    *MockNetString
		expectErr    bool
		expectedName string
		expectedKey  string
	}{
		{
			name:         "Valid netstring",
			netString:    &MockNetString{data: "username apikey123"},
			expectErr:    false,
			expectedName: "username",
			expectedKey:  "apikey123",
		},
		{
			name:         "Trailing whitespace",
			netString:    &MockNetString{data: "username apikey123 "},
			expectErr:    true,
			expectedName: "",
			expectedKey:  "",
		},
		{
			name:         "Missing space",
			netString:    &MockNetString{data: "usernameapikey123"},
			expectErr:    true,
			expectedName: "",
			expectedKey:  "",
		},
		{
			name:         "Empty strings",
			netString:    &MockNetString{data: ""},
			expectErr:    true,
			expectedName: "",
			expectedKey:  "",
		},
		{
			name:         "Only space",
			netString:    &MockNetString{data: " "},
			expectErr:    true,
			expectedName: "",
			expectedKey:  "",
		},
		{
			name:         "Multiple spaces",
			netString:    &MockNetString{data: "username    apikey123"},
			expectErr:    true,
			expectedName: "",
			expectedKey:  "",
		},
		{
			name:         "Special characters in key",
			netString:    &MockNetString{data: "username @apikey!"},
			expectErr:    false,
			expectedName: "username",
			expectedKey:  "@apikey!",
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			receiver := &PostfixMapReceiver{}
			err := receiver.ReadNetString(tt.netString)

			if (err != nil) != tt.expectErr {
				t.Errorf("ReadNetString() error = %v, expectErr %v", err, tt.expectErr)
			}

			if !tt.expectErr {
				if receiver.GetName() != tt.expectedName {
					t.Errorf("Expected name %s, got %s", tt.expectedName, receiver.GetName())
				}

				if receiver.GetKey() != tt.expectedKey {
					t.Errorf("Expected key %s, got %s", tt.expectedKey, receiver.GetKey())
				}
			}
		})
	}
}