invitation process upgrade

client/identity_test.go

@@ -142,14 +142,12 @@ func TestGetRequestJobs(t *testing.T) {
 	// Call GetRequestJobs
 	jobs := id.GetRequestJobs()
 
-	// Check that the returned list is as expected
-	assert.Equal(t, 6, len(jobs), "Expected 6 jobs")
+	// All 10 test peers use server1 and server2, so exactly 2 jobs are expected.
+	assert.Equal(t, 2, len(jobs), "Expected 2 jobs (server1 and server2)")
 
-	// Check that each job has the correct server and lookup keys
 	for _, job := range jobs {
-		//fmt.Println(job.Server.GetUid(), job.LookupKeys)
-		assert.Contains(t, []string{"server1", "server2", "server3", "server4", "server5", "server6"}, job.Server.GetUid(), "Unexpected server UID")
-		assert.Len(t, job.LookupKeys, 1, "Expected 1 lookup key per job")
+		assert.Contains(t, []string{"server1", "server2"}, job.Server.GetUid(), "Unexpected server UID")
+		assert.Len(t, job.LookupKeys, 10, "Expected 10 lookup keys (one per test peer) per job")
 	}
 
 	// Clean up