def test_payload_destroy(self):
x = [1,2,3]
y = ['a','b','c']
self.s.smuggle(x=x,y=y)
#TODO handle sub one secon smuggle intervals
time.sleep(1.5)
a = [4,5,6]
b = ['d','e','f']
self.s.smuggle(x=a,y=b)
p = Payload(self.testdir)
#for name in list(os.listdir(self.testdir)) + ['Done']:
# print name
#print len([name for name in os.listdir('testdir') if os.path.isfile(name)])
#print p.destroy(verbose=False)
num_del = p.destroy(verbose=False)
#print num_del
exp_num_del = 4
#for name in list(os.listdir(self.testdir)) + ['Done']:
# print name
self.assertEqual(exp_num_del,num_del,"Problem destroying payloads")
def test_payload(self):
x = [1,2,3]
y = ['a','b','c']
self.s.smuggle(x=x,y=y)
#TODO handle sub one secon smuggle intervals
time.sleep(1.5)
a = [4,5,6]
b = ['d','e','f']
self.s.smuggle(x=a,y=b)
p = Payload(self.testdir)
msg = "Problem testing payloads"
def sorteddictoflists(d):
for key in list(d.keys()):
d[key] = [item for sublist in d[key] for item in sublist]
d[key].sort()
return d
exp = sorteddictoflists({'x' : [x,a], 'y' : [y,b]})
act = sorteddictoflists(p.asvardict())
self.assertEqual(exp,act,msg)
exp = [x, y, a, b]
act = p.aslist()
exp = [item for sublist in exp for item in sublist]
act = [item for sublist in act for item in sublist]
exp = [str(ex) for ex in exp]
act = [str(ac) for ac in act]
exp.sort()
act.sort()
self.assertEqual(exp,act,msg)
