def test_duplicate_data(self):
data = m.Data(playbook=self.playbook,
key='test key',
value='another value')
m.db.session.add(data)
with self.assertRaises(Exception):
m.db.session.commit()
def v2_playbook_on_start(self, playbook):
path = os.path.abspath(playbook._file_name)
act = ansibleCallbackTools()
for _d in act.getDatas():
data = models.Data(playbook_id=current_app._cache['playbook'],
key=_d['key'],
value=_d['value'],
type=_d['type'])
db.session.add(data)
db.session.commit()
def setUp(self):
super(TestModels, self).setUp()
self.playbook = m.Playbook(path='testing.yml')
self.play = m.Play(
name='test play',
playbook=self.playbook,
)
self.task = m.Task(
name='test task',
play=self.play,
playbook=self.playbook,
)
self.data = m.Data(playbook=self.playbook,
key='test key',
value='test value')
self.host = m.Host(
name='localhost',
playbook=self.playbook,
)
self.host_facts = m.HostFacts(host=self.host,
values=json.dumps('{"fact": "value"}'))
self.task_result = m.TaskResult(
task=self.task,
status='ok',
host=self.host,
)
self.stats = m.Stats(
playbook=self.playbook,
host=self.host,
changed=0,
failed=0,
skipped=0,
unreachable=0,
ok=0,
)
for obj in [
self.playbook, self.play, self.task, self.data, self.host,
self.task_result, self.stats
]:
m.db.session.add(obj)
m.db.session.commit()
def create_or_update_key(self, playbook_id, key, value, type):
try:
data = (models.Data.query.filter_by(key=key).filter_by(
playbook_id=playbook_id).one())
data.value = value
data.type = type
except models.NoResultFound:
data = models.Data(playbook_id=playbook_id,
key=key,
value=value,
type=type)
db.session.add(data)
db.session.commit()
return data
def ansible_run(complete=True, gather_facts=True, ara_record=False):
'''Simulate a simple Ansible run by creating the
expected database objects. This roughly approximates the
following playbook:
- hosts: host-<int>
gather_facts: true
tasks:
- test-action:
when: not ara_record
- ara_record:
key: 'test key'
value: 'test value'
when: ara_record
Where `<int>` is a random integer generated each time this
function is called.
Set the `complete` parameter to `False` to simulate an
aborted Ansible run.
Set the `gathered_facts` parameter to `False` to simulate a run with no
facts gathered.
Set the `ara_record` parameter to `True` to simulate a run with an
ara_record task.
'''
playbook = m.Playbook(path='testing.yml')
play = m.Play(playbook=playbook, name='test play')
host = m.Host(name='host-%04d' % random.randint(0, 9999),
playbook=playbook)
if ara_record:
task = m.Task(play=play, playbook=playbook, action='ara_record')
msg = 'Data recorded in ARA for this playbook.'
else:
task = m.Task(play=play, playbook=playbook, action='test-action')
msg = 'This is a test'
result = m.TaskResult(task=task, status='ok', host=host, result=msg)
ctx = dict(playbook=playbook,
play=play,
task=task,
host=host,
result=result)
if gather_facts:
facts = m.HostFacts(host=host, values='{"fact": "value"}')
ctx['facts'] = facts
if ara_record:
data = m.Data(playbook=playbook, key='test key', value='test value')
ctx['data'] = data
for obj in ctx.values():
if hasattr(obj, 'start'):
obj.start()
db.session.add(obj)
db.session.commit()
if complete:
stats = m.Stats(playbook=playbook, host=host)
ctx['stats'] = stats
db.session.add(stats)
ctx['playbook'].complete = True
for obj in ctx.values():
if hasattr(obj, 'stop'):
obj.stop()
db.session.commit()
return ctx
def model(self):
return m.Data(playbook=self.playbook,
key=self.key,
value=self.value)
