def setUp(self):
super(MultipleJumpsFromNormalStateTests, self).setUp()
self.restriction = restrictions.MultipleJumpsFromNormalState()
self.kb += [
facts.Start(uid='start', type='test', nesting=0),
facts.Choice(uid='state_1'),
facts.State(uid='state_2'),
facts.Question(uid='state_3', condition=()),
facts.Jump(state_from='state_2', state_to='state_3'),
facts.Finish(start='start', uid='finish_1', results={}, nesting=0),
facts.Finish(start='start', uid='finish_2', results={}, nesting=0),
facts.Jump(state_from='start', state_to='state_1'),
facts.Option(state_from='state_1',
state_to='state_2',
type='opt_1',
markers=()),
facts.Option(state_from='state_1',
state_to='finish_1',
type='opt_2',
markers=()),
facts.Answer(state_from='state_3',
state_to='finish_1',
condition=True),
facts.Answer(state_from='state_3',
state_to='finish_2',
condition=False)
]
def test_do_step__next_jump(self):
jump_1 = facts.Jump(state_from=self.start.uid, state_to=self.state_1.uid)
jump_2 = facts.Jump(state_from=self.state_1.uid, state_to=self.state_2.uid)
self.kb += [jump_1, jump_2]
self.machine.step()
calls_manager = mock.MagicMock()
pointer = self.machine.pointer
self.assertEqual(pointer.state, self.start.uid)
self.assertEqual(pointer.jump, None)
with mock.patch.object(self.machine.interpreter, 'on_state__before_actions') as on_state__before_actions:
with mock.patch.object(self.machine.interpreter, 'on_state__after_actions') as on_state__after_actions:
with mock.patch.object(self.machine.interpreter, 'on_jump_start__before_actions') as on_jump_start__before_actions:
with mock.patch.object(self.machine.interpreter, 'on_jump_start__after_actions') as on_jump_start__after_actions:
with mock.patch.object(self.machine.interpreter, 'on_jump_end__before_actions') as on_jump_end__before_actions:
with mock.patch.object(self.machine.interpreter, 'on_jump_end__after_actions') as on_jump_end__after_actions:
calls_manager.attach_mock(on_state__before_actions, 'on_state__before_actions')
calls_manager.attach_mock(on_state__after_actions, 'on_state__after_actions')
calls_manager.attach_mock(on_jump_start__before_actions, 'on_jump_start__before_actions')
calls_manager.attach_mock(on_jump_start__after_actions, 'on_jump_start__after_actions')
calls_manager.attach_mock(on_jump_end__before_actions, 'on_jump_end__before_actions')
calls_manager.attach_mock(on_jump_end__after_actions, 'on_jump_end__after_actions')
self.machine.step()
self.assertEqual(calls_manager.mock_calls, [mock.call.on_jump_start__before_actions(jump=jump_1),
mock.call.on_jump_start__after_actions(jump=jump_1)])
pointer = self.machine.pointer
self.assertEqual(pointer.state, self.start.uid)
self.assertEqual(pointer.jump, jump_1.uid)
def test_get_available_jumps__all_jumps(self):
jump_1 = facts.Jump(state_from=self.start.uid, state_to=self.state_1.uid)
jump_2 = facts.Jump(state_from=self.start.uid, state_to=self.state_2.uid)
self.kb += [jump_1, jump_2]
self.assertEqual(set(jump.uid for jump in self.machine.get_available_jumps(self.start)),
set([jump_1.uid, jump_2.uid]))
def construct(cls, nesting, selector, initiator, initiator_position, receiver, receiver_position):
hero = selector.heroes()[0]
ns = selector._kb.get_next_ns()
start = facts.Start(uid=ns+'start',
type=cls.TYPE,
nesting=nesting,
description='Начало: помочь знакомому',
require=[requirements.LocatedIn(object=hero.uid, place=initiator_position.uid),
requirements.LocatedIn(object=receiver.uid, place=receiver_position.uid)],
actions=[actions.Message(type='intro')])
participants = [facts.QuestParticipant(start=start.uid, participant=initiator.uid, role=ROLES.INITIATOR),
facts.QuestParticipant(start=start.uid, participant=receiver.uid, role=ROLES.RECEIVER) ]
finish_successed = facts.Finish(uid=ns+'finish_successed',
start=start.uid,
results={ initiator.uid: RESULTS.SUCCESSED,
receiver.uid: RESULTS.SUCCESSED},
nesting=nesting,
description='помощь оказана',
require=[requirements.LocatedIn(object=hero.uid, place=initiator_position.uid)],
actions=[actions.GiveReward(object=hero.uid, type='finish_successed')])
finish_failed = facts.Finish(uid=ns+'finish_failed',
start=start.uid,
results={ initiator.uid: RESULTS.FAILED,
receiver.uid: RESULTS.FAILED},
nesting=nesting,
description='не удалось помочь',
actions=[actions.GiveReward(object=hero.uid, type='finish_failed')])
help_quest = selector.create_quest_from_person(nesting=nesting+1, initiator=receiver, tags=('can_continue',))
help_extra = []
for help_fact in logic.filter_subquest(help_quest, nesting):
if isinstance(help_fact, facts.Start):
help_extra.append(facts.Jump(state_from=start.uid, state_to=help_fact.uid, start_actions=[actions.Message(type='before_help')]))
elif isinstance(help_fact, facts.Finish):
if help_fact.results[receiver.uid] == RESULTS.SUCCESSED:
help_extra.append(facts.Jump(state_from=help_fact.uid, state_to=finish_successed.uid, start_actions=[actions.Message(type='after_successed_help')]))
else:
help_extra.append(facts.Jump(state_from=help_fact.uid, state_to=finish_failed.uid))
subquest = facts.SubQuest(uid=ns+'help_subquest', members=logic.get_subquest_members(help_quest))
line = [ start,
finish_successed,
finish_failed,
subquest ]
line.extend(participants)
line.extend(help_quest)
line.extend(help_extra)
return line
def construct(cls, nesting, selector, initiator, initiator_position, receiver, receiver_position):
hero = selector.heroes()[0]
ns = selector._kb.get_next_ns()
start = facts.Start(uid=ns+'start',
type=cls.TYPE,
nesting=nesting,
description=u'Начало: посетить родной города',
require=[requirements.LocatedIn(object=hero.uid, place=initiator_position.uid)],
actions=[actions.Message(type='intro')])
participants = [facts.QuestParticipant(start=start.uid, participant=receiver_position.uid, role=ROLES.RECEIVER_POSITION) ]
arriving = facts.State(uid=ns+'arriving',
description=u'Прибытие в город',
require=[requirements.LocatedIn(object=hero.uid, place=receiver_position.uid)])
action_choices = [facts.State(uid=ns+'drunk_song', description=u'спеть пьяную песню', actions=[actions.Message(type='drunk_song'),
actions.DoNothing(type='drunk_song')]),
facts.State(uid=ns+'stagger_streets', description=u'пошататься по улицам', actions=[actions.Message(type='stagger_streets'),
actions.DoNothing(type='stagger_streets')]),
facts.State(uid=ns+'chatting', description=u'пообщаться с друзьями', actions=[actions.Message(type='chatting'),
actions.DoNothing(type='chatting')]),
facts.State(uid=ns+'search_old_friends', description=u'искать старого друга', actions=[actions.Message(type='search_old_friends'),
actions.DoNothing(type='search_old_friends')]),
facts.State(uid=ns+'remember_names', description=u'вспоминать имена друзей', actions=[actions.Message(type='remember_names'),
actions.DoNothing(type='remember_names')])]
home_actions = random.sample(action_choices, 3)
finish = facts.Finish(uid=ns+'finish',
start=start.uid,
results={ receiver_position.uid: RESULTS.SUCCESSED},
nesting=nesting,
description=u'завершить посещение города',
actions=[actions.GiveReward(object=hero.uid, type='finish')])
line = [ start,
facts.Jump(state_from=start.uid, state_to=arriving.uid),
arriving,
facts.Jump(state_from=arriving.uid, state_to=home_actions[0].uid),
facts.Jump(state_from=home_actions[0].uid, state_to=home_actions[1].uid),
facts.Jump(state_from=home_actions[1].uid, state_to=home_actions[2].uid),
facts.Jump(state_from=home_actions[2].uid, state_to=finish.uid),
finish
]
line.extend(home_actions)
line.extend(participants)
return line
def construct(cls, nesting, selector, initiator, initiator_position, receiver, receiver_position):
hero = selector.heroes()[0]
ns = selector._kb.get_next_ns()
start = facts.Start(uid=ns+'start',
type=cls.TYPE,
nesting=nesting,
description=u'Начало: навестить соратника',
require=[requirements.LocatedIn(object=hero.uid, place=initiator_position.uid)],
actions=[actions.Message(type='intro')])
participants = [facts.QuestParticipant(start=start.uid, participant=receiver.uid, role=ROLES.RECEIVER) ]
meeting = facts.State(uid=ns+'meeting',
description=u'встреча с соратником',
require=[requirements.LocatedIn(object=hero.uid, place=receiver_position.uid)])
finish_meeting = facts.Finish(uid=ns+'finish_meeting',
start=start.uid,
results={receiver.uid: RESULTS.SUCCESSED},
nesting=nesting,
description=u'соратнику оказана помощь',
require=[requirements.LocatedIn(object=hero.uid, place=receiver_position.uid)],
actions=[actions.GiveReward(object=hero.uid, type='finish_meeting'),
actions.GivePower(object=receiver.uid, power=1)])
help_quest = selector.create_quest_from_person(nesting=nesting+1, initiator=receiver, tags=('can_continue',))
help_extra = []
for help_fact in logic.filter_subquest(help_quest, nesting):
if isinstance(help_fact, facts.Start):
help_extra.append(facts.Jump(state_from=meeting.uid, state_to=help_fact.uid, start_actions=[actions.Message(type='before_help')]))
elif isinstance(help_fact, facts.Finish):
if help_fact.results[receiver.uid] == RESULTS.SUCCESSED:
help_extra.append(facts.Jump(state_from=help_fact.uid, state_to=finish_meeting.uid, start_actions=[actions.Message(type='after_help')]))
subquest = facts.SubQuest(uid=ns+'help_subquest', members=logic.get_subquest_members(help_quest))
line = [ start,
facts.Jump(state_from=start.uid, state_to=meeting.uid),
meeting,
finish_meeting,
subquest
]
line.extend(participants)
line.extend(help_quest)
line.extend(help_extra)
return line
def setUp(self):
super(AllStatesHasJumpsTests, self).setUp()
self.kb += [
facts.Start(uid='start', type='test', nesting=0),
facts.State(uid='state_1'),
facts.Finish(start='start', uid='finish_1', results={}, nesting=0),
facts.Jump(state_from='start', state_to='state_1'),
facts.Jump(state_from='state_1', state_to='finish_1')
]
self.restriction = restrictions.AllStatesHasJumps()
def test_get_next_jump__require_one_jump__exception(self):
jump_1 = facts.Jump(state_from=self.start.uid,
state_to=self.state_1.uid)
jump_2 = facts.Jump(state_from=self.start.uid,
state_to=self.state_2.uid)
self.kb += [jump_1, jump_2]
self.assertRaises(exceptions.MoreThenOneJumpsAvailableError,
self.machine.get_next_jump,
self.start,
single=True)
def test_get_next_jump__all_jumps(self):
jump_1 = facts.Jump(state_from=self.start.uid, state_to=self.state_1.uid)
jump_2 = facts.Jump(state_from=self.start.uid, state_to=self.state_2.uid)
self.kb += [jump_1, jump_2]
jumps = set()
for i in xrange(100):
jumps.add(self.machine.get_next_jump(self.start, single=False).uid)
self.assertEqual(jumps, set([jump_1.uid, jump_2.uid]))
def test_get_start_state(self):
start_2 = facts.Start(uid='s2', type='test', nesting=0)
start_3 = facts.Start(uid='start_3', type='test', nesting=0)
self.kb += [ start_2,
start_3,
facts.Jump(state_from=self.start.uid, state_to=start_2.uid),
facts.Jump(state_from=start_3.uid, state_to=self.start.uid) ]
self.assertEqual(self.machine.get_start_state().uid, start_3.uid)
def construct(cls, nesting, selector, initiator, initiator_position,
receiver, receiver_position):
hero = selector.heroes()[0]
ns = selector._kb.get_next_ns()
start = facts.Start(uid=ns + 'start',
type=cls.TYPE,
nesting=nesting,
description=u'Начало: простейшее задание',
require=[
requirements.LocatedIn(
object=hero.uid,
place=initiator_position.uid)
],
actions=[actions.Message(type='intro')])
participants = [
facts.QuestParticipant(start=start.uid,
participant=receiver_position.uid,
role=ROLES.RECEIVER_POSITION)
]
arriving = facts.State(uid=ns + 'arriving',
description=u'Прибытие в другой город',
require=[
requirements.LocatedIn(
object=hero.uid,
place=receiver_position.uid)
])
facts.State(uid=ns + 'any_action',
description=u'выполнить какое-то действие',
actions=[actions.Message(type='do smth')])
finish = facts.Finish(
uid=ns + 'finish',
start=start.uid,
results={receiver_position.uid: RESULTS.SUCCESSED},
nesting=nesting,
description=u'завершить задание',
actions=[actions.GiveReward(object=hero.uid, type='finish')])
line = [
start,
facts.Jump(state_from=start.uid, state_to=arriving.uid), arriving,
facts.Jump(state_from=arriving.uid, state_to=finish.uid), finish
]
line.extend(participants)
return line
def test_get_nearest_choice__choice_after_finish(self):
choice = facts.Choice(uid='choice')
option_1 = facts.Option(state_from=choice.uid, state_to=self.state_1.uid, type='opt_1', markers=())
option_2 = facts.Option(state_from=choice.uid, state_to=self.state_2.uid, type='opt_2', markers=())
path = facts.ChoicePath(choice=choice.uid, option=option_2.uid, default=True)
self.kb += (choice,
option_1,
option_2,
path,
facts.Jump(state_from=self.start.uid, state_to=self.finish_1.uid),
facts.Jump(state_from=self.finish_1.uid, state_to=choice.uid))
self.assertEqual(self.machine.get_nearest_choice(), (None, None, None))
def construct(cls, nesting, selector, initiator, initiator_position, receiver, receiver_position):
hero = selector.heroes()[0]
ns = selector._kb.get_next_ns()
start = facts.Start(uid=ns+'start',
type=cls.TYPE,
nesting=nesting,
description=u'Начало: посетить святой город',
require=[requirements.LocatedIn(object=hero.uid, place=initiator_position.uid)],
actions=[actions.Message(type='intro')])
participants = [facts.QuestParticipant(start=start.uid, participant=receiver_position.uid, role=ROLES.RECEIVER_POSITION) ]
arriving = facts.State(uid=ns+'arriving',
description=u'Прибытие в город',
require=[requirements.LocatedIn(object=hero.uid, place=receiver_position.uid)])
action_choices = [facts.State(uid=ns+'speak_with_guru', description=u'поговорить с гуру', actions=[actions.Message(type='speak_with_guru'),
actions.DoNothing(type='speak_with_guru')]),
facts.State(uid=ns+'stagger_holy_streets', description=u'пошататься по улицам', actions=[actions.Message(type='stagger_holy_streets'),
actions.DoNothing(type='stagger_holy_streets')])]
holy_actions = random.sample(action_choices, 1)
finish = facts.Finish(uid=ns+'finish',
start=start.uid,
results={ receiver_position.uid: RESULTS.SUCCESSED},
nesting=nesting,
description=u'завершить посещение города',
actions=[actions.GiveReward(object=hero.uid, type='finish'),
actions.GivePower(object=receiver_position.uid, power=1)])
line = [ start,
facts.Jump(state_from=start.uid, state_to=arriving.uid),
arriving,
facts.Jump(state_from=arriving.uid, state_to=holy_actions[0].uid),
facts.Jump(state_from=holy_actions[0].uid, state_to=finish.uid),
finish
]
line.extend(holy_actions)
line.extend(participants)
return line
def construct(cls, nesting, selector, initiator, initiator_position, receiver, receiver_position):
hero = selector.heroes()[0]
ns = selector._kb.get_next_ns()
antagonist = selector.new_person(first_initiator=False)
antagonist_position = selector.place_for(objects=(antagonist.uid,))
start = facts.Start(uid=ns+'start',
type=cls.TYPE,
nesting=nesting,
description=u'Начало: навредить противнику',
require=[requirements.LocatedIn(object=hero.uid, place=initiator_position.uid)],
actions=[actions.Message(type='intro')])
participants = [facts.QuestParticipant(start=start.uid, participant=receiver.uid, role=ROLES.RECEIVER),
facts.QuestParticipant(start=start.uid, participant=antagonist_position.uid, role=ROLES.ANTAGONIST_POSITION) ]
finish = facts.Finish(uid=ns+'finish',
start=start.uid,
results={ receiver.uid: RESULTS.FAILED,
antagonist_position.uid: RESULTS.NEUTRAL},
nesting=nesting,
description=u'навредили противнику',
actions=[actions.GiveReward(object=hero.uid, type='finish')])
help_quest = selector.create_quest_between_2(nesting=nesting+1, initiator=antagonist, receiver=receiver, tags=('can_continue',))
help_extra = []
for help_fact in logic.filter_subquest(help_quest, nesting):
if isinstance(help_fact, facts.Start):
help_extra.append(facts.Jump(state_from=start.uid, state_to=help_fact.uid))
elif isinstance(help_fact, facts.Finish):
if help_fact.results[receiver.uid] == RESULTS.FAILED:
help_extra.append(facts.Jump(state_from=help_fact.uid, state_to=finish.uid, start_actions=[actions.Message(type='after_interfere')]))
subquest = facts.SubQuest(uid=ns+'interfere_subquest', members=logic.get_subquest_members(help_quest))
line = [ start,
finish,
subquest
]
line.extend(participants)
line.extend(help_quest)
line.extend(help_extra)
return line
def setUp(self):
super(NoCirclesInStateJumpGraphTests, self).setUp()
self.restriction = restrictions.NoCirclesInStateJumpGraph()
self.kb += [
facts.Start(uid='start', type='test', nesting=0),
facts.State(uid='state_1'),
facts.State(uid='state_2'),
facts.Start(uid='start_3', type='test', nesting=0),
facts.Finish(start='start', uid='finish_1', results={}, nesting=0),
facts.Jump(state_from='start', state_to='state_1'),
facts.Jump(state_from='state_1', state_to='state_2'),
facts.Jump(state_from='state_2', state_to='start_3'),
facts.Jump(state_from='start_3', state_to='finish_1')
]
def test_error(self):
self.kb += [
facts.State(uid='state_2'),
facts.Jump(state_from='start', state_to='state_2')
]
self.assertRaises(self.restriction.Error, self.restriction.validate,
self.kb)
def test_get_nearest_choice__2_choices__second_choice(self):
choice_1 = facts.Choice(uid='choice_1')
choice_2 = facts.Choice(uid='choice_2')
option_2_1 = facts.Option(state_from=choice_2.uid, state_to=self.state_1.uid, type='opt_2_1', markers=())
option_2_2 = facts.Option(state_from=choice_2.uid, state_to=self.state_2.uid, type='opt_2_2', markers=())
path_2 = facts.ChoicePath(choice=choice_2.uid, option=option_2_2.uid, default=True)
option_1_1 = facts.Option(state_from=choice_1.uid, state_to=self.state_1.uid, type='opt_1_1', markers=())
option_1_2 = facts.Option(state_from=choice_1.uid, state_to=choice_2.uid, type='opt_1_2', markers=())
path_1 = facts.ChoicePath(choice=choice_1.uid, option=option_1_2.uid, default=True)
self.kb += (choice_1,
option_1_1,
option_1_2,
path_1,
choice_2,
option_2_1,
option_2_2,
path_2,
facts.Jump(state_from=self.start.uid, state_to=choice_1.uid),
facts.Pointer(state=choice_2.uid)
)
choice_, options_, path_ = self.machine.get_nearest_choice()
self.assertEqual(choice_.uid, choice_2.uid)
self.assertEqual(set(o.uid for o in options_), set([option_2_1.uid, option_2_2.uid]))
self.assertEqual([p.uid for p in path_], [path_2.uid])
def construct_from_place(cls, nesting, selector, start_place):
from questgen.quests.base_quest import ROLES
initiator = selector.new_person(first_initiator=(nesting==0), restrict_places=False, places=(start_place.uid, ))
receiver = selector.new_person(first_initiator=False)
initiator_position = selector.place_for(objects=(initiator.uid,))
receiver_position = selector.place_for(objects=(receiver.uid,))
ns = selector._kb.get_next_ns()
start = facts.Start(uid=ns+'start', type=cls.TYPE, nesting=nesting)
choice_1 = facts.Choice(uid=ns+'choice_1')
choice_2 = facts.Choice(uid=ns+'choice_2')
finish_1_1 = facts.Finish(uid=ns+'finish_1_1',
start=start.uid,
results={initiator.uid: RESULTS.SUCCESSED,
initiator_position.uid: RESULTS.FAILED,
receiver.uid: RESULTS.SUCCESSED,
receiver_position.uid: RESULTS.SUCCESSED},
nesting=nesting)
finish_1_2 = facts.Finish(uid=ns+'finish_1_2',
start=start.uid,
results={initiator.uid: RESULTS.FAILED,
initiator_position.uid: RESULTS.FAILED,
receiver.uid: RESULTS.FAILED,
receiver_position.uid: RESULTS.FAILED},
nesting=nesting)
finish_2 = facts.Finish(uid=ns+'finish_2',
start=start.uid,
results={initiator.uid: RESULTS.SUCCESSED,
initiator_position.uid: RESULTS.SUCCESSED,
receiver.uid: RESULTS.FAILED,
receiver_position.uid: RESULTS.FAILED},
nesting=nesting)
participants = [facts.QuestParticipant(start=start.uid, participant=initiator.uid, role=ROLES.INITIATOR),
facts.QuestParticipant(start=start.uid, participant=initiator_position.uid, role=ROLES.INITIATOR_POSITION),
facts.QuestParticipant(start=start.uid, participant=receiver.uid, role=ROLES.RECEIVER),
facts.QuestParticipant(start=start.uid, participant=receiver_position.uid, role=ROLES.RECEIVER_POSITION) ]
quest_facts = [ start,
choice_1,
choice_2,
finish_1_1,
finish_1_2,
finish_2,
facts.Jump(state_from=start.uid, state_to=choice_1.uid),
facts.Option(state_from=choice_1.uid, state_to=finish_2.uid, type='opt_1', markers=()),
facts.Option(state_from=choice_1.uid, state_to=choice_2.uid, type='opt_2', markers=()),
facts.Option(state_from=choice_2.uid, state_to=finish_1_1.uid, type='opt_2_1', markers=()),
facts.Option(state_from=choice_2.uid, state_to=finish_1_2.uid, type='opt_2_2', markers=())
]
return participants + quest_facts
def test_linked_choices__linked_option_with_processed_choice(self):
is_raised = False
for i in xrange(100):
kb = KnowledgeBase()
start = facts.Start(uid='start', type='test', nesting=0)
choice_1 = facts.Choice(uid='choice_1')
choice_2 = facts.Choice(uid='choice_2')
finish_1 = facts.Finish(uid='finish_1',
results={},
nesting=0,
start='start')
finish_2 = facts.Finish(uid='finish_2',
results={},
nesting=0,
start='start')
option_1 = facts.Option(state_from=choice_1.uid,
state_to=finish_1.uid,
type='opt_1',
markers=())
option_2 = facts.Option(state_from=choice_1.uid,
state_to=choice_2.uid,
type='opt_2',
markers=())
option_2_1 = facts.Option(state_from=choice_2.uid,
state_to=finish_1.uid,
type='opt_2_1',
markers=())
option_2_2 = facts.Option(state_from=choice_2.uid,
state_to=finish_2.uid,
type='opt_2_2',
markers=())
facts_list = [
start,
facts.Jump(state_from=start.uid,
state_to=choice_1.uid), choice_1, choice_2,
finish_1, finish_2, option_1, option_2, option_2_1, option_2_2,
facts.OptionsLink(options=(option_2.uid, option_2_2.uid))
]
kb += facts_list
try:
transformators.determine_default_choices(kb)
except exceptions.LinkedOptionWithProcessedChoiceError:
is_raised = True
self.assertEqual(len(list(kb.filter(facts.ChoicePath))), 1)
self.assertEqual(
[path.choice for path in kb.filter(facts.ChoicePath)],
[choice_2.uid])
self.assertEqual(
[path.option for path in kb.filter(facts.ChoicePath)],
[option_2_1.uid])
break
self.assertFalse(is_raised)
def test_state_not_reached(self):
self.kb += [
facts.State(uid='state_3'),
facts.State(uid='state_4'),
facts.Jump(state_from='state_3', state_to='state_4')
]
self.assertRaises(self.restriction.Error, self.restriction.validate,
self.kb)
def test_wrong_end_action(self):
self.kb += [
facts.Jump(state_from='start',
state_to='state_1',
end_actions=[facts.Place(uid='wrong action')])
]
self.assertRaises(self.restriction.Error, self.restriction.validate,
self.kb)
def test_do_step__step_after_finish(self):
jump_1 = facts.Jump(state_from=self.start.uid, state_to=self.finish_1.uid)
self.kb += jump_1
self.machine.step()
self.machine.step()
self.machine.step()
self.assertRaises(exceptions.NoJumpsFromLastStateError, self.machine.step)
def construct(cls, nesting, selector, initiator, initiator_position, receiver, receiver_position):
ns = selector._kb.get_next_ns()
start = facts.Start(uid=ns+'start',
type=cls.TYPE,
nesting=nesting,
description=u'Начало: самый простой квест')
participants = [facts.QuestParticipant(start=start.uid, participant=initiator.uid, role=ROLES.INITIATOR),
facts.QuestParticipant(start=start.uid, participant=initiator_position.uid, role=ROLES.INITIATOR_POSITION),
facts.QuestParticipant(start=start.uid, participant=receiver.uid, role=ROLES.RECEIVER),
facts.QuestParticipant(start=start.uid, participant=receiver_position.uid, role=ROLES.RECEIVER_POSITION) ]
finish_successed = facts.Finish(uid=ns+'finish_successed',
start=start.uid,
results={initiator.uid: RESULTS.SUCCESSED,
initiator_position.uid: RESULTS.SUCCESSED,
receiver.uid: RESULTS.SUCCESSED,
receiver_position.uid: RESULTS.SUCCESSED},
nesting=nesting,
description=u'завершить задание удачно')
finish_failed = facts.Finish(uid=ns+'finish_failed',
start=start.uid,
results={initiator.uid: RESULTS.FAILED,
initiator_position.uid: RESULTS.FAILED,
receiver.uid: RESULTS.FAILED,
receiver_position.uid: RESULTS.FAILED},
nesting=nesting,
description=u'завершить задание плохо')
event = facts.Event(uid=ns+'event', members=(finish_successed.uid, finish_failed.uid))
line = [ start,
finish_successed,
finish_failed,
event,
facts.Jump(state_from=start.uid, state_to=finish_successed.uid),
facts.Jump(state_from=start.uid, state_to=finish_failed.uid) ]
line.extend(participants)
return line
def test_no_restricted_states(self):
facts_list = [
facts.Start(uid='start', type='test', nesting=0),
facts.Finish(uid='st_finish', results={}, nesting=0,
start='start'),
facts.Jump(state_from='start', state_to='st_finish')
]
self.kb += facts_list
transformators.remove_restricted_states(self.kb)
self.check_in_knowledge_base(self.kb, facts_list)
def test_can_do_step__requirement_failed(self):
state_3 = facts.State(uid='state_3', require=[requirements.IsAlive(object='hero')])
jump_3 = facts.Jump(state_from=self.start.uid, state_to=state_3.uid)
self.kb += [ state_3, jump_3]
pointer = self.machine.pointer
self.kb -= pointer
self.kb += pointer.change(state=self.start.uid, jump=jump_3.uid)
self.assertFalse(self.machine.can_do_step())
def test_no_choices(self):
facts_list = [
facts.Start(uid='start', type='test', nesting=0),
facts.Finish(uid='st_finish', results={}, nesting=0,
start='start'),
facts.Jump(state_from='start', state_to='st_finish')
]
self.kb += facts_list
transformators.determine_default_choices(self.kb)
self.check_in_knowledge_base(self.kb, facts_list)
self.assertEqual(len(list(self.kb.filter(facts.OptionsLink))), 0)
def test_simple_tagged_jump(self):
facts_list = [
facts.Start(uid='start', type='test', nesting=0),
facts.Finish(uid='st_finish', results={}, nesting=0,
start='start'),
facts.Jump(state_from='start', state_to='st_finish'),
facts.Event(uid='event_tag', members=('st_finish', ))
]
self.kb += facts_list
transformators.activate_events(self.kb)
self.check_in_knowledge_base(self.kb, facts_list)
def setUp(self):
super(RemoveBrokenQuestionStatesTests, self).setUp()
self.question = facts.Question(uid='state_1', condition=())
self.answer_1 = facts.Answer(state_from='state_1',
state_to='state_2',
condition=True)
self.answer_2 = facts.Answer(state_from='state_1',
state_to='finish_1',
condition=False)
self.kb += [
facts.Start(uid='start', type='test', nesting=0), self.question,
facts.State(uid='state_2'),
facts.Finish(start='start', uid='finish_1', results={}, nesting=0),
facts.Finish(start='start', uid='finish_2', results={}, nesting=0),
facts.Jump(state_from='start', state_to='state_1'), self.answer_1,
self.answer_2,
facts.Jump(state_from='state_2', state_to='finish_2')
]
def test_finish_at_not_finish_state(self):
facts_list = [
facts.Start(uid='start', type='test', nesting=0),
facts.Finish(uid='st_finish', results={}, nesting=0,
start='start'),
facts.Jump(state_from='start', state_to='st_finish')
]
self.kb += facts_list
broken_path = [
facts.State(uid='st_broken_1'),
facts.Jump(state_from='start', state_to='st_broken_1')
]
self.kb += broken_path
transformators.remove_broken_states(self.kb)
self.check_in_knowledge_base(self.kb, facts_list)
self.check_not_in_knowledge_base(self.kb, broken_path)
def test_can_do_step__success(self):
state_3 = facts.State(uid='state_3', require=[requirements.IsAlive(object='hero')])
jump_3 = facts.Jump(state_from=self.start.uid, state_to=state_3.uid)
self.kb += [ state_3, jump_3]
pointer = self.machine.pointer
self.kb -= pointer
self.kb += pointer.change(state=self.start.uid, jump=jump_3.uid)
with mock.patch('questgen.machine.Machine.interpreter', FakeInterpreter(check_is_alive=True)):
self.assertTrue(self.machine.can_do_step())
