mirror of
https://github.com/pion/webrtc.git
synced 2024-05-03 11:54:53 +08:00
56 lines
1.1 KiB
Go
56 lines
1.1 KiB
Go
// SPDX-FileCopyrightText: 2023 The Pion community <https://pion.ly>
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package webrtc
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestNewICEProtocol(t *testing.T) {
|
|
testCases := []struct {
|
|
protoString string
|
|
shouldFail bool
|
|
expectedProto ICEProtocol
|
|
}{
|
|
{unknownStr, true, ICEProtocol(Unknown)},
|
|
{"udp", false, ICEProtocolUDP},
|
|
{"tcp", false, ICEProtocolTCP},
|
|
{"UDP", false, ICEProtocolUDP},
|
|
{"TCP", false, ICEProtocolTCP},
|
|
}
|
|
|
|
for i, testCase := range testCases {
|
|
actual, err := NewICEProtocol(testCase.protoString)
|
|
if (err != nil) != testCase.shouldFail {
|
|
t.Error(err)
|
|
}
|
|
assert.Equal(t,
|
|
testCase.expectedProto,
|
|
actual,
|
|
"testCase: %d %v", i, testCase,
|
|
)
|
|
}
|
|
}
|
|
|
|
func TestICEProtocol_String(t *testing.T) {
|
|
testCases := []struct {
|
|
proto ICEProtocol
|
|
expectedString string
|
|
}{
|
|
{ICEProtocol(Unknown), unknownStr},
|
|
{ICEProtocolUDP, "udp"},
|
|
{ICEProtocolTCP, "tcp"},
|
|
}
|
|
|
|
for i, testCase := range testCases {
|
|
assert.Equal(t,
|
|
testCase.expectedString,
|
|
testCase.proto.String(),
|
|
"testCase: %d %v", i, testCase,
|
|
)
|
|
}
|
|
}
|