15531: Refactor federation-migrate
[arvados.git] / sdk / python / tests / fed-migrate / check.py
1 import arvados
2 import json
3 import sys
4
5 j = json.load(open(sys.argv[1]))
6
7 apiA = arvados.api(host=j["arvados_api_hosts"][0], token=j["superuser_tokens"][0], insecure=True)
8 apiB = arvados.api(host=j["arvados_api_hosts"][1], token=j["superuser_tokens"][1], insecure=True)
9 apiC = arvados.api(host=j["arvados_api_hosts"][2], token=j["superuser_tokens"][2], insecure=True)
10
11 users = apiA.users().list().execute()
12
13 assert len(users["items"]) == 10
14
15 by_username = {}
16
17 for i in range(1, 9):
18     found = False
19     for u in users["items"]:
20         if u["username"] == ("case%d" % i) and u["email"] == ("case%d@test" % i):
21             found = True
22             by_username[u["username"]] = u["uuid"]
23     assert found
24
25 users = apiB.users().list().execute()
26 assert len(users["items"]) == 10
27
28 for i in range(2, 9):
29     found = False
30     for u in users["items"]:
31         if u["username"] == ("case%d" % i) and u["email"] == ("case%d@test" % i) and u["uuid"] == by_username[u["username"]]:
32             found = True
33     assert found
34
35 users = apiC.users().list().execute()
36 assert len(users["items"]) == 10
37
38 for i in range(2, 9):
39     found = False
40     for u in users["items"]:
41         if u["username"] == ("case%d" % i) and u["email"] == ("case%d@test" % i) and u["uuid"] == by_username[u["username"]]:
42             found = True
43     assert found
44
45 print("Passed checks")