def _check_rotation(self): current_name = datetime.utcnow().strftime(self._format) if current_name != self._fname: if self._writer: self._writer.fo.close() self._fname = current_name f = file(os.path.join(self._fdir, self._fname), 'ab') self._writer = flow.FilteredFlowWriter(f, self._filt)
def test_filter(self): sio = StringIO() fl = filt.parse("~c 200") w = flow.FilteredFlowWriter(sio, fl) f = tutils.tflow(resp=True) f.response.code = 200 w.add(f) f = tutils.tflow(resp=True) f.response.code = 201 w.add(f) sio.seek(0) r = flow.FlowReader(sio) assert len(list(r.stream()))