use pending.next to determine next

1 files changed, 18 insertions, 0 deletions

diff --git a/internal/server/repository/pending_test.go b/internal/server/repository/pending_test.go
index 44fe84c..28563a6 100644
--- a/internal/server/repository/pending_test.go
+++ b/internal/server/repository/pending_test.go
@@ -46,3 +46,21 @@ func TestPendingDelete(t *testing.T) {
 		t.Error("expected not an ok", entries)
 	}
 }
+
+func TestPendingNext(t *testing.T) {
+	pending := newPending()
+
+	id, ok := pending.next(types.LinkedIn)
+	if ok {
+		t.Error("not expected ok return status", id)
+	}
+
+	pending.add(types.LinkedIn, "fooid")
+	id, ok = pending.next(types.LinkedIn)
+	if !ok {
+		t.Error("expected ok return status")
+	}
+	if id != "fooid" {
+		t.Error("expected entry ID fooid")
+	}
+}