Exemplo n.º 1
0
    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)
Exemplo n.º 2
0
    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()))