def setup_method(self, func_method): stop_event = threading.Event() counter = worker.SafeCounter() worker_queue = queue.Queue() folder = None self.extract_worker = worker.ExtractWorker(1, worker_queue, counter, stop_event, folder)
def test_get(self): counter = worker.SafeCounter() counter.max = 100 counter.increment() counter.increment() counter.increment() assert counter.get() == 3
def setup_method(self, func_method): test_stop_event = threading.Event() error_stop_event = threading.Event() test_counter = worker.SafeCounter() error_counter = worker.SafeCounter() test_worker_queue = queue.Queue() error_worker_queue = queue.Queue() for item in range(1, 100): test_worker_queue.put(item) error_worker_queue.put(item) self.processed = {} self.test_worker = FalseWorker(1, test_worker_queue, test_counter, test_stop_event, self.processed) self.error_worker = ErrorWorker(2, error_worker_queue, error_counter, error_stop_event, self.processed)
def test_increment_five_start_ten(self): counter = worker.SafeCounter(start=10, incr=5) counter.max = 100 return_value = counter.increment() assert return_value[0] == 15 assert return_value[1] == 100 return_value = counter.increment() assert return_value[0] == 20 assert return_value[1] == 100
def test_increment_one(self): counter = worker.SafeCounter() counter.max = 100 return_value = counter.increment() assert return_value[0] == 1 assert return_value[1] == 100 return_value = counter.increment() assert return_value[0] == 2 assert return_value[1] == 100