Unit tests: dnsserver/https (#5251)

- Added unit tests for dnsserver/https.

Signed-off-by: nathannaveen <42319948+nathannaveen@users.noreply.github.com>
This commit is contained in:
nathannaveen 2022-03-14 10:55:25 -04:00 committed by GitHub
parent 74830b35fd
commit 188077d7fc
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -0,0 +1,90 @@
package dnsserver
import (
"net"
"net/http"
"reflect"
"testing"
)
func TestDoHWriter_LocalAddr(t *testing.T) {
tests := []struct {
name string
laddr net.Addr
want net.Addr
}{
{
name: "LocalAddr",
laddr: &net.TCPAddr{},
want: &net.TCPAddr{},
},
{
name: "LocalAddr",
laddr: &net.UDPAddr{},
want: &net.UDPAddr{},
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
d := &DoHWriter{
laddr: tt.laddr,
}
if got := d.LocalAddr(); !reflect.DeepEqual(got, tt.want) {
t.Errorf("LocalAddr() = %v, want %v", got, tt.want)
}
})
}
}
func TestDoHWriter_RemoteAddr(t *testing.T) {
tests := []struct {
name string
want net.Addr
raddr net.Addr
}{
{
name: "RemoteAddr",
want: &net.TCPAddr{},
raddr: &net.TCPAddr{},
},
{
name: "RemoteAddr",
want: &net.UDPAddr{},
raddr: &net.UDPAddr{},
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
d := &DoHWriter{
raddr: tt.raddr,
}
if got := d.RemoteAddr(); !reflect.DeepEqual(got, tt.want) {
t.Errorf("RemoteAddr() = %v, want %v", got, tt.want)
}
})
}
}
func TestDoHWriter_Request(t *testing.T) {
tests := []struct {
name string
request *http.Request
want *http.Request
}{
{
name: "Request",
request: &http.Request{},
want: &http.Request{},
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
d := &DoHWriter{
request: tt.request,
}
if got := d.Request(); !reflect.DeepEqual(got, tt.want) {
t.Errorf("Request() = %v, want %v", got, tt.want)
}
})
}
}