def test_graph_isomorphic(self):
from Goap.Planner import Node
from Goap.Planner import Edge
acts = Actions()
acts.add(name='CreateTmpDir',
pre_conditions={
'tmp_dir_state': False,
'tmp_dir_content': False
},
effects={
'tmp_dir_state': True,
'tmp_dir_content': False
},
shell='mkdir -p /tmp/goap_tmp')
acts.add(name='CreateToken',
pre_conditions={
'tmp_dir_state': True,
'tmp_dir_content': False
},
effects={
'tmp_dir_state': True,
'tmp_dir_content': True
},
shell='touch /tmp/goap_tmp/.token')
node1 = Node(attributes={
'tmp_dir_state': False,
'tmp_dir_content': False
})
node2 = Node(attributes={
'tmp_dir_state': True,
'tmp_dir_content': False
})
node3 = Node(attributes={
'tmp_dir_state': True,
'tmp_dir_content': True
})
edge1 = Edge(name='CreateTmpDir',
predecessor=node1,
successor=node2,
obj=acts.get('CreateTmpDir'))
edge2 = Edge(name='CreateToken',
predecessor=node2,
successor=node3,
obj=acts.get('CreateToken'))
g1 = nx.DiGraph(nodes=[node1, node2, node3], edges=[edge1, edge2])
g2 = self.planner.graph.directed
assert nx.is_isomorphic(g1, g2) is True
class AutomatonTest(unittest.TestCase):
@staticmethod
def __reset_environment():
if path.isdir('/tmp/goap_tmp'):
subprocess.call(['rm', '-rf', '/tmp/goap_tmp'])
def __print(self):
self.print.pprint(
'Acknowledge world: {}, Action Plan: {}, Result: {}'.format(
self.automaton.world_state, self.automaton.action_plan,
self.automaton.actions_response))
def setUp(self):
self.print = PrettyPrinter(indent=4)
self.goal = {
"tmp_dir_state": "exist",
"tmp_dir_content": "token_found",
}
self.actions = Actions()
self.sensors = Sensors()
self.sensors.add(
name='SenseTmpDirState',
shell=
'if [ -d "/tmp/goap_tmp" ]; then echo -n "exist"; else echo -n "not_exist"; fi',
binding='tmp_dir_state')
self.sensors.add(
name='SenseTmpDirContent',
shell=
'[ -f /tmp/goap_tmp/.token ] && echo -n "token_found" || echo -n "token_not_found"',
binding='tmp_dir_content')
self.actions.add(name='CreateTmpDir',
pre_conditions={
'tmp_dir_state': 'not_exist',
'tmp_dir_content': 'token_not_found'
},
effects={
'tmp_dir_state': 'exist',
'tmp_dir_content': 'token_not_found'
},
shell='mkdir -p /tmp/goap_tmp')
self.actions.add(name='CreateToken',
pre_conditions={
'tmp_dir_state': 'exist',
'tmp_dir_content': 'token_not_found'
},
effects={
'tmp_dir_state': 'exist',
'tmp_dir_content': 'token_found'
},
shell='touch /tmp/goap_tmp/.token')
world_state_matrix = {
"tmp_dir_state": 'Unknown',
"tmp_dir_content": 'Unknown',
}
self.automaton = Automaton(name='directory_watcher',
actions=self.actions,
sensors=self.sensors,
world_state=world_state_matrix)
def test_sensing(self):
self.__reset_environment()
self.automaton.input_goal(self.goal)
self.automaton.sense()
assert self.automaton.world_state == {
'tmp_dir_state': 'not_exist',
'tmp_dir_content': 'token_not_found'
}
def test_planning(self):
create_tmp_dir = self.actions.get('CreateTmpDir')
create_token = self.actions.get('CreateToken')
self.__reset_environment()
self.automaton.input_goal(self.goal)
self.automaton.sense()
self.automaton.plan()
action_plan = [
action[2]['object'] for action in self.automaton.action_plan
]
assert action_plan == [create_tmp_dir, create_token]
def test_acting(self):
self.__reset_environment()
self.automaton.input_goal(self.goal)
self.automaton.sense()
self.automaton.plan()
self.automaton.act()
assert path.isdir('/tmp/goap_tmp') and path.isfile(
'/tmp/goap_tmp/.token')
class ActionTest(unittest.TestCase):
def setUp(self):
# ACTIONS
self.actions = Actions()
def test_add_action_success(self):
self.actions.add(name='CreateVPC',
pre_conditions={
'vpc': False,
'db': False,
'app': False
},
effects={
'vpc': True,
'db': False,
'app': False
},
shell='awscli vpc create')
self.actions.add(name='CreateDB',
pre_conditions={
'vpc': True,
'db': False,
'app': False
},
effects={
'vpc': True,
'db': True,
'app': False
},
shell='awscli vpc create')
assert "CreateVPC" == str(self.actions.get(name='CreateVPC'))
assert "CreateDB" == str(self.actions.get(name='CreateDB'))
def test_remove_action_success(self):
self.actions.add(name='CreateVPC',
pre_conditions={
'vpc': False,
'db': False,
'app': False
},
effects={
'vpc': True,
'db': False,
'app': False
},
shell='awscli vpc create')
self.actions.add(name='CreateDB',
pre_conditions={
'vpc': True,
'db': False,
'app': False
},
effects={
'vpc': True,
'db': True,
'app': False
},
shell='awscli vpc create')
self.actions.remove(name='CreateVPC')
assert "CreateDB" == str(self.actions.get(name='CreateDB'))
def test_remove_action_error(self):
self.actions.add(name='CreateVPC',
pre_conditions={
'vpc': False,
'db': False,
'app': False
},
effects={
'vpc': True,
'db': False,
'app': False
},
shell='awscli vpc create')
self.actions.add(name='CreateDB',
pre_conditions={
'vpc': True,
'db': False,
'app': False
},
effects={
'vpc': True,
'db': True,
'app': False
},
shell='awscli vpc create')
self.actions.remove(name='CreateAPP')
assert 'None' == str(self.actions.get(name='CreateAPP'))