def test_fluent_behavior_simple(self):
thing = ObjectWithBehaviors()
when(thing).test_request(anything()).then().increment_count()
thing.behaviors.test_request_behavior.run("A")
assertThat(thing.count).isEqualTo(1)
assertThat(thing.captured).isEqualTo(["A"])
def test_fluent_behavior__set_default_works(self):
thing = ObjectWithBehaviors()
thing.behaviors.test_request_behavior.set_default(
lambda obj: thing.increment_unhandled())
when(thing).test_request(anything()).then().increment_count()
thing.behaviors.test_request_behavior.run("A")
thing.behaviors.test_request_behavior.run("A")
assertThat(thing.unhandled_count).isEqualTo(1)
def test_fluent_behavior__always__captures_all(self):
thing = ObjectWithBehaviors()
when(thing).test_request(anything()).always().increment_count()
thing.behaviors.test_request_behavior.run("A")
thing.behaviors.test_request_behavior.run("B")
thing.behaviors.test_request_behavior.run("C")
assertThat(thing.count).isEqualTo(3)
assertThat(thing.captured).isEqualTo(["A", "B", "C"])
def test_fluent_behavior__wait_until_done_not_happened_core():
thing = ObjectWithBehaviors()
thing.behaviors.test_request_behavior.set_default_to_ignore()
when(thing).test_request(lambda obj: obj == "A").then().increment_count()
closure = lambda: thing.behaviors.test_request_behavior.run("B")
t = Timer(0.5, closure)
t.start()
assertThat(
wait_until(thing).test_request(lambda obj: obj == "A").times(
1)).isFalse()
def test_fluent_behavior__wait_until_done_anything_core():
thing = ObjectWithBehaviors()
when(thing).test_request(lambda obj: obj == "A").then().increment_count()
closure = lambda: thing.behaviors.test_request_behavior.run("A")
t = Timer(0.5, closure)
t.start()
wait_until(thing).test_request(anything()).times(1)
assertThat(thing.count).isEqualTo(1)
assertThat(thing.captured).isEqualTo(["A"])
def test_fluent_behavior__default_unhandled_crash(self):
thing = ObjectWithBehaviors()
when(thing).test_request(anything()).then().increment_count()
thing.behaviors.test_request_behavior.run("A")
try:
thing.behaviors.test_request_behavior.run("A")
except Exception as e:
logging.debug(e)
return True # Failed as expected
return False
def test_fluent_behavior__matcher__captures_relevant_core():
thing = ObjectWithBehaviors()
thing.behaviors.test_request_behavior.set_default_to_ignore()
when(thing).test_request(lambda obj: obj == "B").always().increment_count()
thing.behaviors.test_request_behavior.run("A")
thing.behaviors.test_request_behavior.run("B")
thing.behaviors.test_request_behavior.run("C")
assertThat(thing.count).isEqualTo(1)
assertThat(thing.captured).isEqualTo(["B"])
def test_fluent_behavior__wait_until_done_only_one_event_is_done(self):
thing = ObjectWithBehaviors()
when(thing).test_request(anything()).always().increment_count()
closure1 = lambda: thing.behaviors.test_request_behavior.run("A")
t1 = Timer(1, closure1)
t1.start()
closure2 = lambda: thing.behaviors.test_request_behavior.run("B")
t2 = Timer(3, closure2)
t2.start()
assertThat(
wait_until(thing).test_request(lambda obj: obj == "A").times(
2)).isFalse()
def test_fluent_behavior__then_repeated__captures_relevant(self):
thing = ObjectWithBehaviors()
thing.behaviors.test_request_behavior.set_default_to_ignore()
when(thing).test_request(
anything()).then().increment_count().increment_count()
thing.behaviors.test_request_behavior.run("A")
thing.behaviors.test_request_behavior.run("B")
thing.behaviors.test_request_behavior.run("A")
assertThat(thing.count).isEqualTo(2)
assertThat(thing.captured).isEqualTo(["A", "B"])
def test_fluent_behavior__then_single__captures_one_core():
thing = ObjectWithBehaviors()
thing.behaviors.test_request_behavior.set_default_to_ignore()
when(thing).test_request(anything()).then().increment_count()
thing.behaviors.test_request_behavior.run("A")
thing.behaviors.test_request_behavior.run("A")
thing.behaviors.test_request_behavior.run("A")
assertThat(thing.count).isEqualTo(1)
assertThat(thing.captured).isEqualTo(["A"])
def test_fluent_behavior__wait_until_done_two_events_AB(self):
thing = ObjectWithBehaviors()
when(thing).test_request(anything()).always().increment_count()
closure1 = lambda: thing.behaviors.test_request_behavior.run("A")
t1 = Timer(0.5, closure1)
t1.start()
closure2 = lambda: thing.behaviors.test_request_behavior.run("B")
t2 = Timer(1, closure2)
t2.start()
wait_until(thing).test_request(anything()).times(2)
assertThat(thing.count).isEqualTo(2)
assertThat(thing.captured).isEqualTo(["A", "B"])
def test_fluent_behavior__wait_until_done_two_events_AA_core():
thing = ObjectWithBehaviors()
when(thing).test_request(
lambda obj: obj == "A").then().increment_count().increment_count()
closure1 = lambda: thing.behaviors.test_request_behavior.run("A")
t1 = Timer(0.5, closure1)
t1.start()
closure2 = lambda: thing.behaviors.test_request_behavior.run("A")
t2 = Timer(0.5, closure2)
t2.start()
wait_until(thing).test_request(lambda obj: obj == "A").times(2)
assertThat(thing.count).isEqualTo(2)
assertThat(thing.captured).isEqualTo(["A", "A"])
