def setUp(self):
import datetime
today = datetime.date.today()
self.u = User(username='******')
self.u.save()
self.baron = Plane(tailnumber="N1234", type='BE-55')
self.baron.save()
self.seaplane = Plane(tailnumber="N5678", cat_class=3)
self.seaplane.save()
self.local_route = Route.from_string('kvta kvta')
self.more50nm_route = Route.from_string('kvta kluk')
self.no_land_more50nm_route = Route.from_string('kvta @kluk kvta')
self.less50nm_route = Route.from_string('kvta kcmh')
self.no_land_less50nm_route = Route.from_string('kvta @kcmh kvta')
self.f = Flight(total=11.0,
pic=10.0,
date=today,
route=self.local_route)
def handle_flight(self, line):
from forms import ImportFlightForm
if not line.get("tailnumber") == "":
kwargs = {"tailnumber": line.get("tailnumber"), "user": self.user}
if line.get("type"):
kwargs.update({"type": line.get("type")})
try:
plane, created = Plane.objects.get_or_create(**kwargs)
except Plane.MultipleObjectsReturned:
plane = Plane.objects.filter(**kwargs)[0]
else:
## 90 = the unknown plane
plane = Plane(pk=90) ## FIXME should be something more intuitive
line.update({"plane": plane.pk})
flight = Flight(user=self.user)
form = ImportFlightForm(line, instance=flight, user=self.user)
if form.is_valid():
ff = form.save(commit=False)
ff.save(no_badges=True)
message = 'good'
else:
message = form.errors
return status_decorator(
super(DatabaseImport, self).handle_flight, line, message)
def setUp(self):
self.p = Plane(tailnumber="N444444", cat_class=4, type="TYPE")
self.p.save()
self.u = User(username='******')
self.u.save()
self.f = Flight(
plane=self.p,
route=Route.from_string('mer-lga'),
user=self.u,
date='2009-01-05',
total=10.0,
)
self.f.save()
def setUp(self):
level = Level(flavor="Plain",
plane_life_span=3,
max_today_write=3,
max_today_reply=3,
next_level_likes=10)
level.save()
user = user_model.User.objects.create_user(username='******',
password='******')
user.save()
reply_user = user_model.User.objects.create_user(
username='******', password='******')
reply_user.save()
self.user = User(user=user,
today_write_count=10,
today_reply_count=10,
total_likes=10,
level=level)
self.user.save()
self.reply_user = User(user=reply_user,
today_write_count=10,
today_reply_count=10,
total_likes=10,
level=level)
self.reply_user.save()
self.plane = Plane(author=self.user, content='content', tag='tag')
self.plane.save()
self.reply = Reply(plane_author=self.user,
reply_author=self.reply_user,
original_content='orig',
original_tag='orig_tag',
content='cont')
self.reply.save()
self.client = Client()
