Add test for missing edge case

jupp0r · May 31, 2016 · 04aad4a · 04aad4a
1 parent 58d0658
commit 04aad4a
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/priorty_queue_test.go b/priorty_queue_test.go
@@ -78,3 +78,19 @@ func TestPopEmptyQueue(t *testing.T) {
 		t.Fatal("should produce error when performing pop on empty queue")
 	}
 }
+
+func TestUpdateNonExistingItem(t *testing.T) {
+	pq := New()
+
+	pq.Insert("foo", 4)
+	pq.UpdatePriority("bar", 5)
+
+	if pq.Len() != 1 {
+		t.Fatal("update should not add items")
+	}
+
+	item, _ := pq.Pop()
+	if item.(string) != "foo" {
+		t.Fatalf("update should not overwrite item, expected \"foo\", got \"%v\"", item.(string))
+	}
+}