63 lines
2.6 KiB
Python
63 lines
2.6 KiB
Python
from django.test import TestCase
|
|
from django.contrib.auth.models import User
|
|
from inventory.models import Lift
|
|
from lifting.models import Set
|
|
from .importer import import_fitnotes_db
|
|
|
|
|
|
class TestFitnotesImport(TestCase):
|
|
# fitnotes.db has:
|
|
# April 1
|
|
# bench press 10 @ 45
|
|
# bench press 5 @ 95
|
|
# bench press 3 @ 135
|
|
# bench press 5 @ 155
|
|
# April 3
|
|
# squat 10 @ 45
|
|
# squat 5 @ 95
|
|
# squat 3 @ 135
|
|
# squat 2 @ 185
|
|
# squat 5 @ 225
|
|
|
|
def setUp(self):
|
|
self.user = User.objects.create_user('default', 'default@example.com', 'default')
|
|
self.bench = Lift.objects.create(name='bench press')
|
|
self.squat = Lift.objects.create(name='squat')
|
|
|
|
self.good_mapping = {'flat barbell bench press': 'bench press',
|
|
'barbell squat': 'squat'
|
|
}
|
|
self.bad_mapping = {'flat barbell bench press': 'bench press' }
|
|
|
|
def test_basic_import(self):
|
|
# ensure that the data comes in
|
|
import_fitnotes_db('fitnotes/testdata/example.fitnotes', self.user, self.good_mapping)
|
|
assert Set.objects.filter(lift=self.bench).count() == 4
|
|
assert Set.objects.filter(lift=self.squat).count() == 5
|
|
|
|
def test_double_import(self):
|
|
# two identical dbs, should be idempotent
|
|
import_fitnotes_db('fitnotes/testdata/example.fitnotes', self.user, self.good_mapping)
|
|
import_fitnotes_db('fitnotes/testdata/example.fitnotes', self.user, self.good_mapping)
|
|
assert Set.objects.filter(lift=self.bench).count() == 4
|
|
assert Set.objects.filter(lift=self.squat).count() == 5
|
|
|
|
def test_import_with_other_data(self):
|
|
Set.objects.create(lift=self.bench, weight_kg=100, reps=10, date='2014-01-01',
|
|
user=self.user)
|
|
import_fitnotes_db('fitnotes/testdata/example.fitnotes', self.user, self.good_mapping)
|
|
assert Set.objects.filter(lift=self.bench).count() == 5
|
|
|
|
def test_import_with_bad_mapping(self):
|
|
with self.assertRaises(ValueError):
|
|
import_fitnotes_db('fitnotes/testdata/example.fitnotes', self.user, self.bad_mapping)
|
|
assert Set.objects.filter(lift=self.bench).count() == 0
|
|
|
|
def test_bad_data_import(self):
|
|
# good db then bad db, should fail without screwing up existing data
|
|
import_fitnotes_db('fitnotes/testdata/example.fitnotes', self.user, self.good_mapping)
|
|
with self.assertRaises(Exception):
|
|
# baddata.fitnotes has all lift ids set to 9999
|
|
import_fitnotes_db('fitnotes/testdata/baddata.fitnotes', self.user, self.good_mapping)
|
|
assert Set.objects.count() == 9
|