def test_with(self): sampler.FREQ = 1 with sampler.sampling('bob', 'baz', ('foo', 'foobar')): sleep(0.01) query = models.Query.objects.get() self.assertEquals('bob', query.query_type) self.assertEquals('baz', query.query) self.assertTrue(query.total_duration >= 0.01)