Cover all handlers

m-lab · Feb 12, 2024 · dfd1a87 · dfd1a87
1 parent eddee21
commit dfd1a87
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/handler/handler_test.go b/handler/handler_test.go
@@ -120,3 +120,17 @@ func TestServer_Reload(t *testing.T) {
 		}
 	})
 }
+
+func TestServer_LiveAndReady(t *testing.T) {
+	t.Run("success", func(t *testing.T) {
+		f := &fakeIataFinder{}
+		s := NewServer("fake", f)
+		rw := httptest.NewRecorder()
+		req := httptest.NewRequest(http.MethodGet, "/", nil)
+		s.Live(rw, req)
+		s.Ready(rw, req)
+
+		// TODO: remove once handler has a real implementation.
+		s.Register(rw, req)
+	})
+}