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