From ec9b22ec2fd7e4efcac36304917be91cacdf0d88 Mon Sep 17 00:00:00 2001
From: sankari gopalakrishnan <sankari@uber.com>
Date: Wed, 20 Dec 2023 15:03:19 +0100
Subject: [PATCH] Adding unit tests to workflowHandler_test.go (#5500)

* Adding unit tests to workflowHandler_test.go

* Update workflowHandler_test.go
---
 service/frontend/workflowHandler_test.go | 44 ++++++++++++++++++++++++
 1 file changed, 44 insertions(+)

diff --git a/service/frontend/workflowHandler_test.go b/service/frontend/workflowHandler_test.go
index 50b0ec47a86..96c8614329c 100644
--- a/service/frontend/workflowHandler_test.go
+++ b/service/frontend/workflowHandler_test.go
@@ -702,6 +702,50 @@ func (s *workflowHandlerSuite) TestRegisterDomain_Success_NotEnabled() {
 	s.NoError(err)
 }
 
+func (s *workflowHandlerSuite) TestListDomains_Success() {
+	domain := persistenceGetDomainResponse(
+		&domain.ArchivalState{},
+		&domain.ArchivalState{},
+	)
+	listDomainResp := &persistence.ListDomainsResponse{
+		Domains: []*persistence.GetDomainResponse{
+			domain,
+			domain,
+		},
+	}
+	s.mockMetadataMgr.
+		On("ListDomains", mock.Anything, mock.Anything).
+		Return(listDomainResp, nil)
+	wh := s.getWorkflowHandler(s.newConfig(dc.NewInMemoryClient()))
+
+	result, err := wh.ListDomains(context.Background(), &types.ListDomainsRequest{})
+	s.NoError(err)
+	s.Equal(2, len(result.GetDomains()))
+}
+
+func (s *workflowHandlerSuite) TestListDomains_RequestNotSet() {
+	wh := s.getWorkflowHandler(s.newConfig(dc.NewInMemoryClient()))
+
+	result, err := wh.ListDomains(context.Background(), nil /* list request is not set */)
+	s.Error(err)
+	s.Equal(errRequestNotSet, err)
+	s.Nil(result)
+}
+
+func (s *workflowHandlerSuite) TestHealth_StatusOK() {
+	wh := s.getWorkflowHandler(s.newConfig(dc.NewInMemoryClient())) // workflow handler gets initial health status as HealthStatusWarmingUp
+
+	result, err := wh.Health(context.Background()) // Health check looks for HealthStatusOK
+	s.NoError(err)
+	s.False(result.Ok)
+
+	wh.UpdateHealthStatus(HealthStatusOK)
+
+	result, err = wh.Health(context.Background())
+	s.NoError(err)
+	s.True(result.Ok)
+}
+
 func (s *workflowHandlerSuite) TestDescribeDomain_Success_ArchivalDisabled() {
 	getDomainResp := persistenceGetDomainResponse(
 		&domain.ArchivalState{Status: types.ArchivalStatusDisabled, URI: ""},