def get_or_create_file(self, path):
try:
if self.playbook.id:
file_ = (models.File.query
.filter_by(path=path)
.filter_by(playbook_id=self.playbook.id)
.one())
return file_
except models.NoResultFound:
pass
file_ = models.File(path=path, playbook=self.playbook)
db.session.add(file_)
db.session.commit()
try:
with open(path, 'r') as fd:
data = fd.read()
sha1 = models.content_sha1(data)
content = models.FileContent.query.get(sha1)
if content is None:
content = models.FileContent(content=data)
file_.content = content
except IOError:
log.warning('failed to open %s for reading', path)
return file_
def model(self):
return m.File(is_playbook=self.is_playbook,
path=self.path,
playbook=self.playbook)
def ansible_run(complete=True,
failed=False,
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 `failed` parameter to `True` to simulate a
failed 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')
playbook_file = m.File(path=playbook.path,
playbook=playbook,
is_playbook=True)
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)
task_skipped = m.Task(play=play, playbook=playbook, action='foo')
result_skipped = m.TaskResult(task=task_skipped,
status='skipped',
host=host,
result='Conditional check failed')
task_failed = m.Task(play=play, playbook=playbook, action='bar')
result_failed = m.TaskResult(task=task_failed,
status='failed',
host=host,
result='Failed to do thing')
ctx = dict(playbook=playbook,
play=play,
task=task,
result=result,
file=playbook_file,
host=host)
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)
extra_tasks = [task_skipped, result_skipped]
if failed:
extra_tasks.append(task_failed)
extra_tasks.append(result_failed)
for obj in extra_tasks:
db.session.add(obj)
db.session.commit()
if complete:
stats = m.Stats(playbook=playbook,
host=host,
ok=1,
skipped=1,
failed=int(failed))
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