fix missing constants + add license

gravitational · Dec 5, 2024 · 06d909d · 06d909d
1 parent 09b8f86
commit 06d909d
Show file tree

Hide file tree

Showing 4 changed files with 48 additions and 6 deletions.
diff --git a/lib/autoupdate/rollout/controller.go b/lib/autoupdate/rollout/controller.go
@@ -61,8 +61,8 @@ func NewController(client Client, log *slog.Logger, clock clockwork.Clock) (*Con
 		clock: clock,
 		log:   log,
 		reconciler: reconciler{
-			clt: client,
-			log: log,
+			clt:               client,
+			log:               log,
 			rolloutStrategies: []rolloutStrategy{
 				// TODO(hugoShaka): add the strategies here as we implement them
 			},

diff --git a/lib/autoupdate/rollout/reconciler.go b/lib/autoupdate/rollout/reconciler.go
@@ -42,10 +42,6 @@ const (
 
 	defaultGroupName = "default"
 	defaultStartHour = 12
-
-	// Common reasons
-	updateReasonCreated         = "created"
-	updateReasonReconcilerError = "reconciler_error"
 )
 
 var (

diff --git a/lib/autoupdate/rollout/strategy.go b/lib/autoupdate/rollout/strategy.go
@@ -1,3 +1,21 @@
+/*
+ * Teleport
+ * Copyright (C) 2024  Gravitational, Inc.
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Affero General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU Affero General Public License for more details.
+ *
+ * You should have received a copy of the GNU Affero General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ */
+
 package rollout
 
 import (
@@ -11,6 +29,12 @@ import (
 	"github.com/gravitational/teleport/api/types"
 )
 
+const (
+	// Common update reasons
+	updateReasonCreated         = "created"
+	updateReasonReconcilerError = "reconciler_error"
+)
+
 // rolloutStrategy is responsible for rolling out the update across groups.
 // This interface allows us to inject dummy strategies for simpler testing.
 type rolloutStrategy interface {

diff --git a/lib/autoupdate/rollout/strategy_test.go b/lib/autoupdate/rollout/strategy_test.go
@@ -1,3 +1,21 @@
+/*
+ * Teleport
+ * Copyright (C) 2024  Gravitational, Inc.
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Affero General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU Affero General Public License for more details.
+ *
+ * You should have received a copy of the GNU Affero General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ */
+
 package rollout
 
 import (
@@ -136,6 +154,10 @@ func Test_inWindow(t *testing.T) {
 func Test_setGroupState(t *testing.T) {
 	groupName := "test-group"
 
+	// TODO(hugoShaka) remove those two variables once the strategies are merged and the constants are defined.
+	updateReasonCanStart := "can_start"
+	updateReasonCannotStart := "cannot_start"
+
 	clock := clockwork.NewFakeClock()
 	// oldUpdateTime is 5 minutes in the past
 	oldUpdateTime := clock.Now()