WIP
This commit is contained in:
parent
75829eb4a1
commit
9132cd142e
7 changed files with 194 additions and 1 deletions
1
.env
Normal file
1
.env
Normal file
|
|
@ -0,0 +1 @@
|
|||
MANIFEST_URL='https://storage.googleapis.com/ccdb-db-files/manifest.json'
|
||||
36
ccdb/projects/factories.py
Normal file
36
ccdb/projects/factories.py
Normal file
|
|
@ -0,0 +1,36 @@
|
|||
import factory
|
||||
|
||||
from .models import Project, Grant, GrantReport
|
||||
|
||||
|
||||
class ProjectFactory(factory.DjangoModelFactory):
|
||||
class Meta:
|
||||
model = Project
|
||||
|
||||
name = factory.Sequence(lambda n: 'project{}'.format(n))
|
||||
code = factory.Sequence(lambda n: 'p{}'.format(n))
|
||||
iacuc_number = 'abc'
|
||||
description = 'lorem ipsum'
|
||||
sort_order = factory.Sequence(lambda n: n)
|
||||
|
||||
|
||||
class GrantFactory(factory.DjangoModelFactory):
|
||||
class Meta:
|
||||
model = Grant
|
||||
|
||||
title = factory.Sequence(lambda n: 'grant{}'.format(n))
|
||||
code = factory.Sequence(lambda n: 'g{}'.format(n))
|
||||
description = 'lorem ipsum'
|
||||
sort_order = factory.Sequence(lambda n: n)
|
||||
|
||||
|
||||
class GrantReportFactory(factory.DjangoModelFactory):
|
||||
class Meta:
|
||||
model = GrantReport
|
||||
|
||||
grant = 0
|
||||
title = factory.Sequence(lambda n: 'grant{}'.format(n))
|
||||
code = factory.Sequence(lambda n: 'g{}'.format(n))
|
||||
description = 'lorem ipsum'
|
||||
sort_order = factory.Sequence(lambda n: n)
|
||||
|
||||
83
ccdb/projects/migrations/0005_DATA_initial.py
Normal file
83
ccdb/projects/migrations/0005_DATA_initial.py
Normal file
|
|
@ -0,0 +1,83 @@
|
|||
from django.db import migrations
|
||||
from django.forms import modelform_factory
|
||||
|
||||
from ccdb.utils.data import get_data_sources
|
||||
|
||||
|
||||
class Migration(migrations.Migration):
|
||||
def migrate(apps, schema_editor):
|
||||
sources = get_data_sources()
|
||||
if not sources:
|
||||
print('no sources')
|
||||
return
|
||||
|
||||
c = sources['db0']
|
||||
|
||||
Project = apps.get_model('projects', 'Project')
|
||||
Grant = apps.get_model('projects', 'Grant')
|
||||
GrantReport = apps.get_model('projects', 'GrantReport')
|
||||
|
||||
Project.objects.all().delete()
|
||||
Grant.objects.all().delete()
|
||||
GrantReport.objects.all().delete()
|
||||
|
||||
ProjectForm = modelform_factory(Project, fields=('name', 'code',
|
||||
'iacuc_number',
|
||||
'description',
|
||||
'sort_order'))
|
||||
GrantForm = modelform_factory(Grant, fields=('title', 'code',
|
||||
'description', 'sort_order'))
|
||||
GrantReportForm = modelform_factory(GrantReport, fields=('grant', 'title',
|
||||
'report_type',
|
||||
'description',
|
||||
'due_date',
|
||||
'submitted_date',
|
||||
'sort_order'))
|
||||
|
||||
for r in c.execute('SELECT * FROM tbl_lu_projects;'):
|
||||
form = ProjectForm(dict(name=r[1], code=r[2],
|
||||
iacuc_number=r[3], description=r[4],
|
||||
sort_order=int(r[5])))
|
||||
if form.is_valid():
|
||||
Project.objects.create(id=r[0], **form.cleaned_data)
|
||||
else:
|
||||
print('project', r[0:], form.errors.as_data())
|
||||
|
||||
for r in c.execute('SELECT * FROM tbl_lu_grants;'):
|
||||
form = GrantForm(dict(title=r[1], code=r[2], description=r[3], sort_order=r[4]))
|
||||
if form.is_valid():
|
||||
Grant.objects.create(id=r[0], **form.cleaned_data)
|
||||
else:
|
||||
print('grant', r[0:], form.errors.as_data())
|
||||
|
||||
for r in c.execute('SELECT * FROM tbl_hash_project_grants;'):
|
||||
p = Project.objects.get(id=r[0])
|
||||
g = Grant.objects.get(id=r[1])
|
||||
p.grants.add(g)
|
||||
p.save()
|
||||
|
||||
for r in c.execute('SELECT *, report_due_date AS "due_date [dtdt]" FROM tbl_lu_grant_reports;'):
|
||||
form = GrantReportForm(dict(grant=r[0], title=r[1], report_type=r[2],
|
||||
description=r[3], due_date=r[8],
|
||||
submitted_date=r[5], sort_order=r[7]))
|
||||
if form.is_valid():
|
||||
form.save() # No PK field in Andre's file
|
||||
else:
|
||||
print('grant report', r[0:], form.errors.as_data())
|
||||
|
||||
|
||||
def rollback(apps, schema_editor):
|
||||
Project = apps.get_model('projects', 'Project')
|
||||
Grant = apps.get_model('projects', 'Grant')
|
||||
GrantReport = apps.get_model('projects', 'GrantReport')
|
||||
|
||||
for model in [Project, Grant, GrantReport]:
|
||||
model.objects.all().delete()
|
||||
|
||||
dependencies = [
|
||||
('projects', '0004_initial_grantreport'),
|
||||
]
|
||||
|
||||
operations = [
|
||||
migrations.RunPython(migrate, rollback),
|
||||
]
|
||||
0
ccdb/projects/tests/__init__.py
Normal file
0
ccdb/projects/tests/__init__.py
Normal file
67
ccdb/projects/tests/test_models.py
Normal file
67
ccdb/projects/tests/test_models.py
Normal file
|
|
@ -0,0 +1,67 @@
|
|||
from datetime import datetime
|
||||
|
||||
from django.test import TestCase
|
||||
from django.db import IntegrityError, transaction
|
||||
|
||||
from ..models import Project, Grant, GrantReport
|
||||
|
||||
|
||||
def _project(name='project', code='p', iacuc_number='xyz', description='lorem ipsum',
|
||||
sort_order=1):
|
||||
return Project.objects.create(name=name, code=code, iacuc_number=iacuc_number,
|
||||
description=description, sort_order=sort_order)
|
||||
|
||||
def _grant(title='grant', code='g', description='lorem ipsum', sort_order=1):
|
||||
return Grant.objects.create(title=title, code=code, description=description,
|
||||
sort_order=sort_order)
|
||||
|
||||
|
||||
def _grant_report(title='grant report', grant=None, report_type='g',
|
||||
description='lorem ipsum', due_date=datetime.now(), sort_order=1):
|
||||
if not grant:
|
||||
grant = _grant()
|
||||
return GrantReport.objects.create(grant=grant, title=title, report_type=report_type,
|
||||
description=description, due_date=due_date,
|
||||
sort_order=sort_order)
|
||||
|
||||
|
||||
class ProjectTestCase(TestCase):
|
||||
def test_creation(self):
|
||||
p = _project()
|
||||
self.assertTrue(isinstance(p, Project))
|
||||
self.assertEqual(p.__str__(), p.name)
|
||||
|
||||
def test_uniqueness(self):
|
||||
p1 = _project()
|
||||
with transaction.atomic(), self.assertRaises(IntegrityError):
|
||||
p2 = _project()
|
||||
p3 = _project(name='abc')
|
||||
self.assertTrue(isinstance(p3, Project))
|
||||
|
||||
|
||||
class GrantTestCase(TestCase):
|
||||
def test_creation(self):
|
||||
g = _grant()
|
||||
self.assertTrue(isinstance(g, Grant))
|
||||
self.assertEqual(g.__str__(), g.title)
|
||||
|
||||
def test_uniqueness(self):
|
||||
g1 = _grant()
|
||||
with transaction.atomic(), self.assertRaises(IntegrityError):
|
||||
g2 = _grant()
|
||||
g3 = _grant(title='abc')
|
||||
self.assertTrue(isinstance(g3, Grant))
|
||||
|
||||
|
||||
class GrantReportTestCase(TestCase):
|
||||
def test_creation(self):
|
||||
g = _grant_report()
|
||||
self.assertTrue(isinstance(g, GrantReport))
|
||||
self.assertEqual(g.__str__(), g.title)
|
||||
|
||||
def test_uniqueness(self):
|
||||
g1 = _grant_report()
|
||||
with transaction.atomic(), self.assertRaises(IntegrityError):
|
||||
g2 = _grant_report()
|
||||
g3 = _grant_report(title='abc')
|
||||
self.assertTrue(isinstance(g3, GrantReport))
|
||||
|
|
@ -45,7 +45,7 @@ def _write_url(url, filename):
|
|||
|
||||
|
||||
def _get_db0():
|
||||
dbfile = 'data/Replica_Hibernators_Back_UAF_Laptop_29_June_2015.sqlite'
|
||||
dbfile = 'data/CC_Database_020216.sqlite'
|
||||
return setup_sqlite(dbfile)
|
||||
|
||||
|
||||
|
|
|
|||
|
|
@ -41,6 +41,10 @@ EMAIL_BACKEND = env('DJANGO_EMAIL_BACKEND',
|
|||
MIDDLEWARE_CLASSES += ('debug_toolbar.middleware.DebugToolbarMiddleware',)
|
||||
INSTALLED_APPS += ('debug_toolbar', )
|
||||
|
||||
# Testing
|
||||
# ------------------------------------------------------------------------------
|
||||
INSTALLED_APPS += ('django_extensions', 'test_without_migrations')
|
||||
|
||||
INTERNAL_IPS = ('127.0.0.1', )
|
||||
|
||||
DEBUG_TOOLBAR_CONFIG = {
|
||||
|
|
@ -54,6 +58,8 @@ DEBUG_TOOLBAR_CONFIG = {
|
|||
# ------------------------------------------------------------------------------
|
||||
TEST_RUNNER = 'django.test.runner.DiscoverRunner'
|
||||
|
||||
MANIFEST_URL = env('MANIFEST_URL', default=None)
|
||||
CORS_ORIGIN_ALLOW_ALL = True
|
||||
|
||||
DJOSER = {
|
||||
'SITE_NAME': 'CCDB (test)',
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue