def test_compareAtoms(self): #self.resetEngine() pengine = self.pengine atom = self.atom assert p_engine.compareAtoms(atom,atom) == 1, 'compare to self' atom_two = pengine.Atom(atom.dest_ip,atom.dest_port,\ atom.protocol,atom.last_timestamp) assert p_engine.compareAtoms(atom,atom_two) == 1, 'compare to equivalent' atom_two = pengine.Atom(atom.dest_ip,atom.dest_port-1,\ atom.protocol,atom.last_timestamp) assert p_engine.compareAtoms(atom,atom_two) == 0, 'compare to different' assert p_engine.compareAtoms(atom_two,atom_two) == 1, 'compare to self' atom_two = pengine.Atom(atom.dest_ip,atom.dest_port,\ atom.protocol,atom.last_timestamp+1) assert p_engine.compareAtoms(atom_two,atom) == 1, 'different timestamp'
def filterWhites(self): white_lock.acquire() whiteList = open(white_file, "r") wll = [] for line in whiteList: if(line != "\n"): wll.append(self.stringToAtom(line)) for index, atom in enumerate(self.p_atoms): for j in wll: if(p_engine.compareAtoms(atom,j)): self.p_atoms.pop(index) break white_lock.release()