def __init__(self):
self.publisher = CookiePublisher()
logging.info('waiting for CookiePublisher to settle')
time.sleep(5)
self.last_cookie = None
self.visits = list()
self.ancestors = [Ancestors(), Ancestors()]
self.lock = threading.Lock()
def next(self, taken_cookie):
self.lock.acquire()
new_ancestor = None
for i, anc in enumerate(self.ancestors):
if taken_cookie is anc.last():
if anc.discontinued:
new_cookie = RecombinationCookie(
(self.last_cookie, taken_cookie))
else:
new_cookie = MutantCookie(taken_cookie)
anc.add(new_cookie)
else:
if anc.discontinued:
# start a new ancestor line
new_ancestor = i
self.ancestors[i] = Ancestors()
else:
# last chance for ancestor line
anc.discontinue()
self.last_cookie = taken_cookie
for func in self.visits:
try:
if new_ancestor != None:
func(new_cookie, self.ancestors[new_ancestor].last())
else:
func(new_cookie)
except:
pass
self.lock.release()
class TestAncestores:
def setup_method(self, method):
self.ancestors = Ancestors()
@mock.patch('random.randint')
def test_init(self, randint):
randint.return_value = 50
assert isinstance(self.ancestors, Ancestors)
assert isinstance(self.ancestors.last(), Cookie)
def test_add_cookie(self):
cookie = Cookie([30, 40])
self.ancestors.add(cookie)
assert self.ancestors.last() == cookie
def test_unset_discontinued(self):
self.ancestors.discontinued = True
cookie = Cookie([30, 40])
self.ancestors.add(cookie)
assert self.ancestors.discontinued == False
class TestAncestores:
def setup_method(self, method):
self.ancestors = Ancestors()
@mock.patch('random.randint')
def test_init(self, randint):
randint.return_value = 50
assert isinstance(self.ancestors, Ancestors)
assert isinstance(self.ancestors.last(), Cookie)
def test_add_cookie(self):
cookie = Cookie([30, 40])
self.ancestors.add(cookie)
assert self.ancestors.last() == cookie
def test_unset_discontinued(self):
self.ancestors.discontinued = True
cookie = Cookie([30, 40])
self.ancestors.add(cookie)
assert self.ancestors.discontinued == False
def setup_method(self, method):
self.ancestors = Ancestors()
def setup_method(self, method):
self.ancestors = Ancestors()
