From e859ac1d5d4e898541b6ee75cfb9d6cff540f071 Mon Sep 17 00:00:00 2001 From: V2Ray Date: Thu, 1 Oct 2015 16:14:05 +0200 Subject: [PATCH] Test case for capability --- capability.go | 2 +- capability_test.go | 21 +++++++++++++++++++++ 2 files changed, 22 insertions(+), 1 deletion(-) create mode 100644 capability_test.go diff --git a/capability.go b/capability.go index ed84adc3..7b291f38 100644 --- a/capability.go +++ b/capability.go @@ -16,7 +16,7 @@ type listCapabilities struct { data []Capability } -func NewCapabilites() Capabilities { +func NewCapabilities() Capabilities { return &listCapabilities{ data: make([]Capability, 0, 16), } diff --git a/capability_test.go b/capability_test.go new file mode 100644 index 00000000..100daeb7 --- /dev/null +++ b/capability_test.go @@ -0,0 +1,21 @@ +package core + +import ( + "testing" + + "github.com/v2ray/v2ray-core/testing/unit" +) + +func TestCapabilities(t *testing.T) { + assert := unit.Assert(t) + + caps := NewCapabilities() + assert.Bool(caps.HasCapability(TCPConnection)).IsFalse() + + caps.AddCapability(TCPConnection) + assert.Bool(caps.HasCapability(TCPConnection)).IsTrue() + + caps.AddCapability(UDPConnection) + assert.Bool(caps.HasCapability(TCPConnection)).IsTrue() + assert.Bool(caps.HasCapability(UDPConnection)).IsTrue() +}