diff options
Diffstat (limited to 'pkg/kafkacomm/publisher/pdp-pap-registration_test.go')
-rw-r--r-- | pkg/kafkacomm/publisher/pdp-pap-registration_test.go | 58 |
1 files changed, 58 insertions, 0 deletions
diff --git a/pkg/kafkacomm/publisher/pdp-pap-registration_test.go b/pkg/kafkacomm/publisher/pdp-pap-registration_test.go new file mode 100644 index 0000000..03749de --- /dev/null +++ b/pkg/kafkacomm/publisher/pdp-pap-registration_test.go @@ -0,0 +1,58 @@ +// - +// ========================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 publisher + +import ( + "errors" + "github.com/stretchr/testify/assert" + "github.com/stretchr/testify/mock" + "policy-opa-pdp/pkg/kafkacomm/publisher/mocks" + "policy-opa-pdp/pkg/model" + "testing" +) + +type MockPdpStatusSender struct { + mock.Mock +} + +func (m *MockPdpStatusSender) SendPdpStatus(pdpStatus model.PdpStatus) error { + return m.Called(pdpStatus).Error(0) + +} + +func TestSendPdpPapRegistration_Success(t *testing.T) { + mockSender := new(mocks.PdpStatusSender) + + mockSender.On("SendPdpStatus", mock.AnythingOfType("model.PdpStatus")).Return(nil) + + err := SendPdpPapRegistration(mockSender) + assert.NoError(t, err) + mockSender.AssertCalled(t, "SendPdpStatus", mock.AnythingOfType("model.PdpStatus")) +} + +func TestSendPdpPapRegistration_Failure(t *testing.T) { + mockSender := new(mocks.PdpStatusSender) + + mockSender.On("SendPdpStatus", mock.AnythingOfType("model.PdpStatus")).Return(errors.New("failed To Send")) + + err := SendPdpPapRegistration(mockSender) + assert.Error(t, err, "Expected an error for failure") + assert.EqualError(t, err, "failed To Send", "Error messages should match") + mockSender.AssertCalled(t, "SendPdpStatus", mock.AnythingOfType("model.PdpStatus")) +} |