netcmd = sys.argv[2]
def flush_cache():
- os.system(netcmd + "cache flush")
+ os.system(netcmd + " cache flush")
+
+def fill_cache(inids, idtype='gid'):
+ for inid in inids:
+ if inid is None:
+ continue
+ subprocess.Popen([wbinfo, '--%s-to-sid=%s' % (idtype, inid)],
+ stdout=subprocess.PIPE).communicate()
domain = subprocess.Popen([wbinfo, "--own-domain"],
stdout=subprocess.PIPE).communicate()[0].strip()
stdout=subprocess.PIPE).communicate()[0].strip()
gids=[]
+uids=[]
+idtypes = []
for line in sids2xids.split('\n'):
result = line.split(' ')[2:]
+ idtypes.append(result[0])
- if result[0] == 'gid' or result[0] == 'uid/gid':
+ gid = None
+ uid = None
+ if result[0] == 'gid':
gid = result[1]
- else:
- gid = ''
+ elif result[0] == 'uid':
+ uid = result[1]
+ elif result[0] == 'uid/gid':
+ gid = result[1]
+ uid = result[1]
+
if gid == '-1':
gid = ''
gids.append(gid)
+ if uid == '-1':
+ uid = ''
+ uids.append(uid)
+
# Check the list produced by the sids-to-xids call with the
-# singular variant (sid-to-gid) for each sid in turn.
-def check_singular(sids, gids):
- i=0
+# singular variant (sid-to-xid) for each sid in turn.
+def check_singular(sids, ids, idtype='gid'):
+ i = 0
for sid in sids:
- gid = subprocess.Popen([wbinfo, '--sid-to-gid', sid],
- stdout=subprocess.PIPE).communicate()[0].strip()
- if gid != gids[i]:
- print "Expected %s, got %s\n", gid, gids[i]
+ if ids[i] is None:
+ continue
+
+ outid = subprocess.Popen([wbinfo, '--sid-to-%s' % idtype, sid],
+ stdout=subprocess.PIPE).communicate()[0].strip()
+ if outid != ids[i]:
+ print "Expected %s, got %s\n" % (outid, ids[i])
+ flush_cache()
+ sys.exit(1)
+ i += 1
+
+# Check the list produced by the sids-to-xids call with the
+# multiple variant (sid-to-xid) for each sid in turn.
+def check_multiple(sids, idtypes):
+ sids2xids = subprocess.Popen([wbinfo, '--sids-to-unix-ids=' + ','.join(sids)],
+ stdout=subprocess.PIPE).communicate()[0].strip()
+ # print sids2xids
+ i = 0
+ for line in sids2xids.split('\n'):
+ result = line.split(' ')[2:]
+
+ if result[0] != idtypes[i]:
+ print "Expected %s, got %s\n" % (idtypes[i], result[0])
+ flush_cache()
sys.exit(1)
- i+=1
+ i += 1
-# first round: with filled cache
-check_singular(sids, gids)
+# first round: with filled cache via sid-to-id
+check_singular(sids, gids, 'gid')
+check_singular(sids, uids, 'uid')
# second round: with empty cache
flush_cache()
-check_singular(sids, gids)
+check_singular(sids, gids, 'gid')
+flush_cache()
+check_singular(sids, uids, 'uid')
+
+# third round: with filled cache via uid-to-sid
+flush_cache()
+fill_cache(uids, 'uid')
+check_multiple(sids, idtypes)
+
+# fourth round: with filled cache via gid-to-sid
+flush_cache()
+fill_cache(gids, 'gid')
+check_multiple(sids, idtypes)
+
+# flush the cache so any incorrect mappings don't break other tests
+flush_cache()
sys.exit(0)