From 72838f0b6f5a8fb70b209db5fa430c2b2e5e60a6 Mon Sep 17 00:00:00 2001
From: Brian Waldon <bcwaldon@gmail.com>
Date: Fri, 31 Jan 2014 16:33:00 -0800
Subject: [PATCH] feat(set): Implement set.Remove

---
 dbus/set.go      |  4 ++++
 dbus/set_test.go | 26 ++++++++++++++++++++++++++
 2 files changed, 30 insertions(+)
 create mode 100644 dbus/set_test.go

diff --git a/dbus/set.go b/dbus/set.go
index e36aaa03..88378b29 100644
--- a/dbus/set.go
+++ b/dbus/set.go
@@ -8,6 +8,10 @@ func (s *set) Add(value string) {
 	s.data[value] = true
 }
 
+func (s *set) Remove(value string) {
+	delete(s.data, value)
+}
+
 func (s *set) Contains(value string) (exists bool) {
 	_, exists = s.data[value]
 	return
diff --git a/dbus/set_test.go b/dbus/set_test.go
new file mode 100644
index 00000000..0cd3b3e6
--- /dev/null
+++ b/dbus/set_test.go
@@ -0,0 +1,26 @@
+package dbus
+
+import (
+	"testing"
+)
+
+// TestBasicSetActions asserts that Add & Remove behavior is correct
+func TestBasicSetActions(t *testing.T) {
+	s := set{}
+
+	if s.Contains("foo") {
+		t.Fatal("set should not contain 'foo'")
+	}
+
+	s.Add("foo")
+
+	if !s.Contains("foo") {
+		t.Fatal("set should contain 'foo'")
+	}
+
+	s.Remove("foo")
+
+	if s.Contains("foo") {
+		t.Fatal("set should not contain 'foo'")
+	}
+}