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)