|
9 | 9 | "github.com/bodgit/tsig/internal/util" |
10 | 10 | "github.com/miekg/dns" |
11 | 11 | "github.com/stretchr/testify/assert" |
| 12 | + "github.com/stretchr/testify/require" |
12 | 13 | ) |
13 | 14 |
|
14 | 15 | type FakeClient struct { |
@@ -161,3 +162,43 @@ func TestCopyDNSClient(t *testing.T) { |
161 | 162 | }) |
162 | 163 | } |
163 | 164 | } |
| 165 | + |
| 166 | +type mockTsigProvider struct { |
| 167 | + Name string |
| 168 | +} |
| 169 | + |
| 170 | +func (f mockTsigProvider) Generate(_ []byte, _ *dns.TSIG) ([]byte, error) { |
| 171 | + return nil, nil |
| 172 | +} |
| 173 | + |
| 174 | +func (f mockTsigProvider) Verify(_ []byte, _ *dns.TSIG) error { |
| 175 | + return nil |
| 176 | +} |
| 177 | + |
| 178 | +func TestCopyDNSClient_shallow_copy(t *testing.T) { |
| 179 | + t.Parallel() |
| 180 | + |
| 181 | + dnsClient := &dns.Client{ |
| 182 | + Net: "udp", |
| 183 | + TsigProvider: &mockTsigProvider{Name: "original"}, |
| 184 | + } |
| 185 | + |
| 186 | + client, err := util.CopyDNSClient(dnsClient) |
| 187 | + require.NoError(t, err) |
| 188 | + |
| 189 | + client.TsigProvider = &mockTsigProvider{Name: "copy"} |
| 190 | + |
| 191 | + originalProvider, ok := dnsClient.TsigProvider.(*mockTsigProvider) |
| 192 | + require.True(t, ok) |
| 193 | + |
| 194 | + assert.Equal(t, "original", originalProvider.Name) |
| 195 | + assert.Equal(t, "udp", dnsClient.Net) |
| 196 | + assert.Nil(t, dnsClient.TsigSecret) |
| 197 | + |
| 198 | + copyProvider, ok := client.TsigProvider.(*mockTsigProvider) |
| 199 | + require.True(t, ok) |
| 200 | + |
| 201 | + assert.Equal(t, "copy", copyProvider.Name) |
| 202 | + assert.Equal(t, "tcp", client.Net) |
| 203 | + assert.NotNil(t, client.TsigSecret) |
| 204 | +} |
0 commit comments