From b50678e307eace3df074d4468f5928d1c370b1bd Mon Sep 17 00:00:00 2001
From: RebeccaMahany <rebeccamahany@gmail.com>
Date: Wed, 17 Jul 2024 10:21:45 -0400
Subject: [PATCH] Add test for knownSubsystem

---
 ee/control/control_test.go | 65 ++++++++++++++++++++++++++++++++++++++
 1 file changed, 65 insertions(+)

diff --git a/ee/control/control_test.go b/ee/control/control_test.go
index 270ebd1b3..76bfa66a2 100644
--- a/ee/control/control_test.go
+++ b/ee/control/control_test.go
@@ -410,6 +410,71 @@ func TestControlServicePersistLastFetched(t *testing.T) {
 	}
 }
 
+func Test_knownSubsystem(t *testing.T) {
+	t.Parallel()
+
+	for _, tt := range []struct {
+		testCaseName   string
+		subsystemName  string
+		consumers      []string
+		subscribers    []string
+		subsystemKnown bool
+	}{
+		{
+			testCaseName:   "subsystem in consumers",
+			subsystemName:  "subsystem_to_query",
+			consumers:      []string{"subsystem_to_query", "some_other_subsystem"},
+			subscribers:    []string{"some_other_subsystem", "a_third_subsystem"},
+			subsystemKnown: true,
+		},
+		{
+			testCaseName:   "subsystem in subscribers",
+			subsystemName:  "subsystem_to_query",
+			consumers:      []string{"some_other_subsystem"},
+			subscribers:    []string{"some_other_subsystem", "subsystem_to_query"},
+			subsystemKnown: true,
+		},
+		{
+			testCaseName:   "subsystem in consumers and subscribers",
+			subsystemName:  "subsystem_to_query",
+			consumers:      []string{"subsystem_to_query", "some_other_subsystem"},
+			subscribers:    []string{"some_other_subsystem", "subsystem_to_query"},
+			subsystemKnown: true,
+		},
+		{
+			testCaseName:   "subsystem is unknown",
+			subsystemName:  "subsystem_to_query",
+			consumers:      []string{"some_other_subsystem", "another_test_subsystem"},
+			subscribers:    []string{"some_other_subsystem"},
+			subsystemKnown: false,
+		},
+	} {
+		tt := tt
+		t.Run(tt.testCaseName, func(t *testing.T) {
+			t.Parallel()
+
+			mockKnapsack := typesMocks.NewKnapsack(t)
+			mockKnapsack.On("RegisterChangeObserver", mock.Anything, keys.ControlRequestInterval)
+			mockKnapsack.On("ControlRequestInterval").Return(60 * time.Second)
+			mockKnapsack.On("Slogger").Return(multislogger.NewNopLogger())
+
+			controlOpts := []Option{}
+			cs := New(mockKnapsack, nil, controlOpts...)
+
+			// Register known consumers and subscribers
+			for _, c := range tt.consumers {
+				require.NoError(t, cs.RegisterConsumer(c, &mockConsumer{}))
+			}
+			for _, s := range tt.subscribers {
+				cs.RegisterSubscriber(s, &mockSubscriber{})
+			}
+
+			known := cs.knownSubsystem(tt.subsystemName)
+			require.Equal(t, tt.subsystemKnown, known)
+		})
+	}
+}
+
 func TestInterrupt_Multiple(t *testing.T) {
 	t.Parallel()