Adding mocks for EndpointManager
Change-Id: I9adbc93e5d30b8294d689a923d6058c76cc176fe
diff --git a/pkg/kafka/endpoint_manager.go b/pkg/kafka/endpoint_manager.go
index 4c13c76..1258382 100644
--- a/pkg/kafka/endpoint_manager.go
+++ b/pkg/kafka/endpoint_manager.go
@@ -56,9 +56,9 @@
// devices owned by that service need to be reconciled
IsDeviceOwnedByService(deviceID string, serviceType string, replicaNumber int32) (bool, error)
- // getReplicaAssignment returns the replica number of the service that owns the deviceID. This is used by the
+ // GetReplicaAssignment returns the replica number of the service that owns the deviceID. This is used by the
// test only
- getReplicaAssignment(deviceID string, serviceType string) (ReplicaID, error)
+ GetReplicaAssignment(deviceID string, serviceType string) (ReplicaID, error)
}
type service struct {
@@ -150,7 +150,7 @@
return m.getReplica() == ReplicaID(replicaNumber), nil
}
-func (ep *endpointManager) getReplicaAssignment(deviceID string, serviceType string) (ReplicaID, error) {
+func (ep *endpointManager) GetReplicaAssignment(deviceID string, serviceType string) (ReplicaID, error) {
owner, err := ep.getOwner(deviceID, serviceType)
if err != nil {
return 0, nil
diff --git a/pkg/kafka/endpoint_manager_test.go b/pkg/kafka/endpoint_manager_test.go
index 0ed5a3d..20b330a 100644
--- a/pkg/kafka/endpoint_manager_test.go
+++ b/pkg/kafka/endpoint_manager_test.go
@@ -195,7 +195,7 @@
logger.Fatalw("error-getting-endpoint", log.Fields{"error": err})
}
deviceIDs[deviceID] = endpoint
- replicaID, err := tm.getReplicaAssignment(deviceID, serviceType)
+ replicaID, err := tm.GetReplicaAssignment(deviceID, serviceType)
if err != nil {
logger.Fatalw("error-getting-endpoint", log.Fields{"error": err})
}
@@ -219,7 +219,7 @@
// Verify that a device belong to the correct node
for deviceID := range deviceIDs {
- replicaID, err := tm.getReplicaAssignment(deviceID, serviceType)
+ replicaID, err := tm.GetReplicaAssignment(deviceID, serviceType)
if err != nil {
logger.Fatalw("error-getting-topic", log.Fields{"error": err})
}