Add a script to create mock data

It's convenient to have a script to easily add mock data to test the
API and model with. Let's do that.

Change-Id: I0f108b19d35280faa472850bdeb9ac83f3faa505
This commit is contained in:
David Moreau Simard 2018-03-10 12:24:56 -05:00
parent e6a39c9663
commit 23fbf95672
No known key found for this signature in database
GPG Key ID: 33A07694CBB71ECC
4 changed files with 73 additions and 14 deletions

View File

@ -33,6 +33,10 @@ This is python3 only right now.
# Run test server -> http://127.0.0.1:8000/api/v1/
tox -e runserver
# Create mock data
source .tox/runserver/bin/activate
python standalone/mockdata.py
# Run actual tests or get coverage
tox -e pep8
tox -e py35

View File

@ -1,4 +1,4 @@
# Generated by Django 2.0.3 on 2018-03-10 16:49
# Generated by Django 2.0.3 on 2018-03-10 17:18
from django.db import migrations, models
import django.db.models.deletion

68
standalone/mockdata.py Normal file
View File

@ -0,0 +1,68 @@
#!/usr/bin/env python
# Copyright (c) 2018 Red Hat, Inc.
#
# This file is part of ARA Records Ansible.
#
# ARA is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# ARA is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with ARA. If not, see <http://www.gnu.org/licenses/>.
# Creates fake data in the database, bypassing the API.
import django
import hashlib
import os
import sys
from django.core import serializers
parent_directory = os.path.dirname(os.path.dirname(os.path.realpath(__file__)))
sys.path.append(parent_directory)
os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'ara.settings')
django.setup()
from api import models
playbook, _ = models.Playbook.objects.get_or_create(
started='2016-05-06T17:20:25.749489-04:00',
path='/tmp/test.yml',
ansible_version='2.3.4',
completed=False,
)
print(serializers.serialize('json',
models.Playbook.objects.all(),
indent=2))
play, _ = models.Play.objects.get_or_create(
started='2016-05-06T17:20:25.749489-04:00',
name='Test play',
playbook=playbook,
)
print(serializers.serialize('json',
models.Play.objects.all(),
indent=2))
content = 'foo'.encode('utf8')
filecontent, _ = models.FileContent.objects.get_or_create(
contents=content,
sha1=hashlib.sha1(content).hexdigest()
)
print(serializers.serialize('json',
models.FileContent.objects.all(),
indent=2))
file, _ = models.File.objects.get_or_create(
playbook=playbook,
content=filecontent,
path='/tmp/anothertest.yml'
)
print(serializers.serialize('json',
models.File.objects.all(),
indent=2))

View File

@ -1,13 +0,0 @@
import os
import sys
import django
parent_directory = os.path.dirname(os.path.dirname(os.path.realpath(__file__)))
sys.path.append(parent_directory)
os.environ.setdefault("DJANGO_SETTINGS_MODULE", "ara.settings")
django.setup()
from api import models
print(models.Playbook.objects.all())