From 337fe4085a3936cef5014576d3a4f33b61d71da4 Mon Sep 17 00:00:00 2001
From: Armon Dadgar <armon.dadgar@gmail.com>
Date: Thu, 28 Aug 2014 14:38:00 -0700
Subject: [PATCH] agent: Testing event endpoints

---
 command/agent/event_endpoint_test.go | 145 +++++++++++++++++++++++++++
 1 file changed, 145 insertions(+)

diff --git a/command/agent/event_endpoint_test.go b/command/agent/event_endpoint_test.go
index 47a299a57b..6c64a44159 100644
--- a/command/agent/event_endpoint_test.go
+++ b/command/agent/event_endpoint_test.go
@@ -1,9 +1,154 @@
 package agent
 
 import (
+	"bytes"
+	"fmt"
+	"net/http"
+	"net/http/httptest"
 	"testing"
+	"time"
+
+	"github.com/hashicorp/consul/testutil"
 )
 
+func TestEventFire(t *testing.T) {
+	httpTest(t, func(srv *HTTPServer) {
+		body := bytes.NewBuffer([]byte("test"))
+		url := "/v1/event/fire/test?node=Node&service=foo&tag=bar"
+		req, err := http.NewRequest("PUT", url, body)
+		if err != nil {
+			t.Fatalf("err: %v", err)
+		}
+		resp := httptest.NewRecorder()
+		obj, err := srv.EventFire(resp, req)
+		if err != nil {
+			t.Fatalf("err: %v", err)
+		}
+
+		event, ok := obj.(*UserEvent)
+		if !ok {
+			t.Fatalf("bad: %#v", obj)
+		}
+
+		if event.ID == "" {
+			t.Fatalf("bad: %#v", event)
+		}
+		if event.Name != "test" {
+			t.Fatalf("bad: %#v", event)
+		}
+		if string(event.Payload) != "test" {
+			t.Fatalf("bad: %#v", event)
+		}
+		if event.NodeFilter != "Node" {
+			t.Fatalf("bad: %#v", event)
+		}
+		if event.ServiceFilter != "foo" {
+			t.Fatalf("bad: %#v", event)
+		}
+		if event.TagFilter != "bar" {
+			t.Fatalf("bad: %#v", event)
+		}
+	})
+}
+
+func TestEventList(t *testing.T) {
+	httpTest(t, func(srv *HTTPServer) {
+		p := &UserEvent{Name: "test"}
+		if err := srv.agent.UserEvent(p); err != nil {
+			t.Fatalf("err: %v", err)
+		}
+
+		testutil.WaitForResult(func() (bool, error) {
+			req, err := http.NewRequest("GET", "/v1/event/list", nil)
+			if err != nil {
+				return false, err
+			}
+			resp := httptest.NewRecorder()
+			obj, err := srv.EventList(resp, req)
+			if err != nil {
+				return false, err
+			}
+
+			list, ok := obj.([]*UserEvent)
+			if !ok {
+				return false, fmt.Errorf("bad: %#v", obj)
+			}
+			if len(list) != 1 || list[0].Name != "test" {
+				return false, fmt.Errorf("bad: %#v", list)
+			}
+			header := resp.Header().Get("X-Consul-Index")
+			if header == "" {
+				return false, fmt.Errorf("bad: %#v", header)
+			}
+			return true, nil
+		}, func(err error) {
+			t.Fatalf("err: %v", err)
+		})
+	})
+}
+
+func TestEventList_Blocking(t *testing.T) {
+	httpTest(t, func(srv *HTTPServer) {
+		p := &UserEvent{Name: "test"}
+		if err := srv.agent.UserEvent(p); err != nil {
+			t.Fatalf("err: %v", err)
+		}
+
+		var index string
+		testutil.WaitForResult(func() (bool, error) {
+			req, err := http.NewRequest("GET", "/v1/event/list", nil)
+			if err != nil {
+				return false, err
+			}
+			resp := httptest.NewRecorder()
+			_, err = srv.EventList(resp, req)
+			if err != nil {
+				return false, err
+			}
+			header := resp.Header().Get("X-Consul-Index")
+			if header == "" {
+				return false, fmt.Errorf("bad: %#v", header)
+			}
+			index = header
+			return true, nil
+		}, func(err error) {
+			t.Fatalf("err: %v", err)
+		})
+
+		go func() {
+			time.Sleep(50 * time.Millisecond)
+			p := &UserEvent{Name: "second"}
+			if err := srv.agent.UserEvent(p); err != nil {
+				t.Fatalf("err: %v", err)
+			}
+		}()
+
+		testutil.WaitForResult(func() (bool, error) {
+			url := "/v1/event/list?index=" + index
+			req, err := http.NewRequest("GET", url, nil)
+			if err != nil {
+				return false, err
+			}
+			resp := httptest.NewRecorder()
+			obj, err := srv.EventList(resp, req)
+			if err != nil {
+				return false, err
+			}
+
+			list, ok := obj.([]*UserEvent)
+			if !ok {
+				return false, fmt.Errorf("bad: %#v", obj)
+			}
+			if len(list) != 2 || list[1].Name != "second" {
+				return false, fmt.Errorf("bad: %#v", list)
+			}
+			return true, nil
+		}, func(err error) {
+			t.Fatalf("err: %v", err)
+		})
+	})
+}
+
 func TestUUIDToUint64(t *testing.T) {
 	inp := "cb9a81ad-fff6-52ac-92a7-5f70687805ec"