fix api with new heartbeat types

moira-alert · Nov 5, 2024 · 82fcf1c · 82fcf1c
1 parent b82f9ab
commit 82fcf1c
Show file tree

Hide file tree

Showing 5 changed files with 9 additions and 9 deletions.
diff --git a/api/controller/contact_test.go b/api/controller/contact_test.go
@@ -890,7 +890,7 @@ func TestRemoveContact(t *testing.T) {
 			Convey("With emergency contact", func() {
 				emergencyContact := datatypes.EmergencyContact{
 					ContactID:      contactID,
-					HeartbeatTypes: []datatypes.HeartbeatType{datatypes.HeartbeatNotifierOff},
+					HeartbeatTypes: []datatypes.HeartbeatType{datatypes.HeartbeatNotifier},
 				}
 				dataBase.EXPECT().GetTeamSubscriptionIDs(teamID).Return(make([]string, 0), nil)
 				dataBase.EXPECT().GetEmergencyContact(contactID).Return(emergencyContact, nil)

diff --git a/api/controller/emergency_contact_test.go b/api/controller/emergency_contact_test.go
@@ -21,7 +21,7 @@ var (
 
 	testEmergencyContact = datatypes.EmergencyContact{
 		ContactID:      testContactID,
-		HeartbeatTypes: []datatypes.HeartbeatType{datatypes.HeartbeatNotifierOff},
+		HeartbeatTypes: []datatypes.HeartbeatType{datatypes.HeartbeatNotifier},
 	}
 	testEmergencyContact2 = datatypes.EmergencyContact{
 		ContactID:      testContactID2,
@@ -30,7 +30,7 @@ var (
 
 	testEmergencyContactDTO = dto.EmergencyContact{
 		ContactID:      testContactID,
-		HeartbeatTypes: []datatypes.HeartbeatType{datatypes.HeartbeatNotifierOff},
+		HeartbeatTypes: []datatypes.HeartbeatType{datatypes.HeartbeatNotifier},
 	}
 	testEmergencyContact2DTO = dto.EmergencyContact{
 		ContactID:      testContactID2,
@@ -286,7 +286,7 @@ func TestUpdateEmergencyContact(t *testing.T) {
 
 		Convey("With empty contact id", func() {
 			emergencyContactDTO := dto.EmergencyContact{
-				HeartbeatTypes: []datatypes.HeartbeatType{datatypes.HeartbeatNotifierOff},
+				HeartbeatTypes: []datatypes.HeartbeatType{datatypes.HeartbeatNotifier},
 			}
 			database.EXPECT().SaveEmergencyContact(testEmergencyContact).Return(nil)
 

diff --git a/api/dto/emergency_contact_test.go b/api/dto/emergency_contact_test.go
@@ -12,7 +12,7 @@ var (
 
 	testEmergencyContact = datatypes.EmergencyContact{
 		ContactID:      testContactID,
-		HeartbeatTypes: []datatypes.HeartbeatType{datatypes.HeartbeatNotifierOff},
+		HeartbeatTypes: []datatypes.HeartbeatType{datatypes.HeartbeatNotifier},
 	}
 )
 

diff --git a/api/handler/contact_test.go b/api/handler/contact_test.go
@@ -837,7 +837,7 @@ func TestRemoveContact(t *testing.T) {
 
 			emergencyContact := datatypes.EmergencyContact{
 				ContactID:      contactID,
-				HeartbeatTypes: []datatypes.HeartbeatType{datatypes.HeartbeatNotifierOff},
+				HeartbeatTypes: []datatypes.HeartbeatType{datatypes.HeartbeatNotifier},
 			}
 
 			mockDb.EXPECT().GetUserSubscriptionIDs(defaultLogin).Return([]string{"test"}, nil).Times(1)

diff --git a/api/handler/emergency_contact_test.go b/api/handler/emergency_contact_test.go
@@ -28,7 +28,7 @@ var (
 
 	testEmergencyContact = datatypes.EmergencyContact{
 		ContactID:      testContactID,
-		HeartbeatTypes: []datatypes.HeartbeatType{datatypes.HeartbeatNotifierOff},
+		HeartbeatTypes: []datatypes.HeartbeatType{datatypes.HeartbeatNotifier},
 	}
 	testEmergencyContact2 = datatypes.EmergencyContact{
 		ContactID:      testContactID2,
@@ -247,7 +247,7 @@ func TestCreateEmergencyContact(t *testing.T) {
 
 		Convey("Try to create emergency contact without contact id", func() {
 			emergencyContact := datatypes.EmergencyContact{
-				HeartbeatTypes: []datatypes.HeartbeatType{datatypes.HeartbeatNotifierOff},
+				HeartbeatTypes: []datatypes.HeartbeatType{datatypes.HeartbeatNotifier},
 			}
 			emergencyContactDTO := dto.EmergencyContact(emergencyContact)
 
@@ -473,7 +473,7 @@ func TestUpdateEmergencyContact(t *testing.T) {
 
 		Convey("Successfully update emergency contact without contact id in dto", func() {
 			emergencyContact := datatypes.EmergencyContact{
-				HeartbeatTypes: []datatypes.HeartbeatType{datatypes.HeartbeatNotifierOff},
+				HeartbeatTypes: []datatypes.HeartbeatType{datatypes.HeartbeatNotifier},
 			}
 			emergencyContactDTO := dto.EmergencyContact(emergencyContact)