aboutsummaryrefslogtreecommitdiffstats
path: root/pkg/pdpstate
diff options
context:
space:
mode:
Diffstat (limited to 'pkg/pdpstate')
-rw-r--r--pkg/pdpstate/pdpstate.go45
-rw-r--r--pkg/pdpstate/pdpstate_test.go43
2 files changed, 88 insertions, 0 deletions
diff --git a/pkg/pdpstate/pdpstate.go b/pkg/pdpstate/pdpstate.go
new file mode 100644
index 0000000..0adaa2e
--- /dev/null
+++ b/pkg/pdpstate/pdpstate.go
@@ -0,0 +1,45 @@
+// -
+// ========================LICENSE_START=================================
+// Copyright (C) 2024: Deutsche Telecom
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+// ========================LICENSE_END===================================
+
+// The pdpstate package manages the state of the Policy Decision Point (PDP), allowing for dynamic updates
+// and retrieval of the PDP's current operational state. States are represented using the model.PdpState type.
+package pdpstate
+
+import (
+ "policy-opa-pdp/pkg/model"
+)
+
+var (
+ State model.PdpState = model.Passive // The current state of the PDP.
+ GetCurrentState = GetState // An alias for the GetState function.
+)
+
+// sets the Pdp State retrieved from the message from Pap
+func SetState(stringState string) error {
+ newState, err := model.ConvertStringToEnumState(stringState)
+ if err != nil {
+ return err
+ }
+
+ State = newState
+ return nil
+}
+
+// Retrieves the current PDP state.
+func GetState() model.PdpState {
+ return State
+}
diff --git a/pkg/pdpstate/pdpstate_test.go b/pkg/pdpstate/pdpstate_test.go
new file mode 100644
index 0000000..6b7078c
--- /dev/null
+++ b/pkg/pdpstate/pdpstate_test.go
@@ -0,0 +1,43 @@
+// -
+// ========================LICENSE_START=================================
+// Copyright (C) 2024: Deutsche Telecom
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+// ========================LICENSE_END===================================
+//
+
+package pdpstate
+
+import (
+ "testing"
+
+ "github.com/stretchr/testify/assert"
+ "policy-opa-pdp/pkg/model"
+)
+
+func TestSetState_Success(t *testing.T) {
+ t.Run("ValidState", func(t *testing.T) {
+ err := SetState("ACTIVE")
+ assert.NoError(t, err, "Expected no error for valid state")
+ assert.Equal(t, model.Active, GetState(), "Expected state to be set to Active")
+ })
+}
+
+func TestSetState_Failure(t *testing.T) {
+ State = model.Passive
+ t.Run("InvalidState", func(t *testing.T) {
+ err := SetState("InvalidState")
+ assert.Error(t, err, "Expected an error for invalid state")
+ assert.Equal(t, model.Passive, GetState(), "Expected state to remain unchanged when setting invalid state")
+ })
+}