def __init__(self):
self.request_controller = RequestController
try:
Launch.create_table()
LaunchSite.create_table()
Rocket.create_table()
except peewee.OperationalError as err:
print(err)
def test_next_launch_response(self):
launch = self.db.get_next_launch()
api_response = requests.get(self.api_url + "next")
api_response = json.loads(api_response.text)
self.assertEqual(launch.mission, api_response['mission_name'])
self.assertNotEqual(Launch.select().count(), 0)
def get_latest_launch(self):
try:
launch = Launch.get(Launch.latest == 1)
return launch
except Launch.DoesNotExist:
return self.populate_latest_launch()
def get_past_launches(self):
launches = Launch.select().where(Launch.past == 1)
if len(launches) == 0:
print("buscando lançamentos")
return self.populate_past_launches()
print("mostrando lançamentos")
return launches
def save_launch(self, data, launch_site, rocket, status):
launch = Launch.create(
mission=data['mission_name'],
flight_number=data['flight_number'],
date=data['launch_date_local'],
rocket=rocket,
launch_site=launch_site,
)
if('next' in status):
launch.next = 1
if('upcoming' in status):
launch.upcoming = 1
if('past' in status):
launch.past = 1
if('latest' in status):
launch.latest = 1
launch.save()
return launch
def test_past_launches(self):
self.db.get_past_launches()
self.assertNotEqual(Launch.select().where(Launch.past == 1).count(), 0)
def test_upcoming_launches(self):
self.db.get_upcoming_launches()
self.assertNotEqual(
Launch.select().where(Launch.upcoming == 1).count(), 0)
def test_next_launch(self):
self.db.get_next_launch()
self.assertNotEqual(Launch.select().where(Launch.next == 1).count(), 0)