From 72a15b8d2dd51cc2717ac0c842f5ac9098a10916 Mon Sep 17 00:00:00 2001
From: Louis Ruch <louisruch@gmail.com>
Date: Wed, 15 Dec 2021 10:06:35 -0800
Subject: [PATCH] bug(target): Add *-host-sources actions to target service
 (#1794)

---
 internal/servers/controller/handlers/targets/target_service.go | 3 +++
 .../controller/handlers/targets/tcp/target_service_test.go     | 3 +++
 2 files changed, 6 insertions(+)

diff --git a/internal/servers/controller/handlers/targets/target_service.go b/internal/servers/controller/handlers/targets/target_service.go
index 95f7a196e..7b91c6f22 100644
--- a/internal/servers/controller/handlers/targets/target_service.go
+++ b/internal/servers/controller/handlers/targets/target_service.go
@@ -55,6 +55,9 @@ var (
 		action.AddHostSets,
 		action.SetHostSets,
 		action.RemoveHostSets,
+		action.AddHostSources,
+		action.SetHostSources,
+		action.RemoveHostSources,
 		action.AddCredentialLibraries,
 		action.SetCredentialLibraries,
 		action.RemoveCredentialLibraries,
diff --git a/internal/servers/controller/handlers/targets/tcp/target_service_test.go b/internal/servers/controller/handlers/targets/tcp/target_service_test.go
index 15963bfa4..41b872227 100644
--- a/internal/servers/controller/handlers/targets/tcp/target_service_test.go
+++ b/internal/servers/controller/handlers/targets/tcp/target_service_test.go
@@ -65,6 +65,9 @@ var testAuthorizedActions = []string{
 	"add-host-sets",
 	"set-host-sets",
 	"remove-host-sets",
+	"add-host-sources",
+	"set-host-sources",
+	"remove-host-sources",
 	"add-credential-libraries",
 	"set-credential-libraries",
 	"remove-credential-libraries",