def test_create_fail_home(self):
user = factory.generate_user()
self.db.session.add(user)
self.db.session.commit()
home = factory.generate_home(user.id)
home.pv_modules = None
self.db.session.add(home)
self.assertRaises(IntegrityError, self.db.session.commit)
def test_sign_up_and_add_home(self):
testing_user = factory.generate_user()
signup_result = self._add_user(testing_user)
home = factory.generate_home(None)
addhome_result = self._add_home(home)
self.assertEqual(signup_result.status_code, 200)
self.assertEqual(addhome_result.status_code, 302)
def test_create_home(self):
user = factory.generate_user()
self.db.session.add(user)
self.db.session.commit()
home = factory.generate_home(user.id)
self.db.session.add(home)
self.assertEqual(self.db.session.query(Homes).count(), 1)
self.assertEqual(self.db.session.query(Homes).get(home.id).user, user)
def test_login_incorrect_password(self):
testing_user = factory.generate_user()
self._add_user(testing_user)
result = self.tester.post('/login',
data={
'form-username': testing_user.email,
'form-password': '******'
})
self.assertIn('Password fails. Try again', str(result.data))
def test_index_with_logged_in_and_not_home_added(self):
testing_user = factory.generate_user()
self._add_user(testing_user)
self.tester.post('/login',
data={
'form-username': testing_user.email,
'form-password': '******'
})
result = self.tester.get('/')
self.assertIn('Enter your home data', str(result.data))
self.assertEqual(result.status_code, 200)
def test_login_and_logout(self):
testing_user = factory.generate_user()
self._add_user(testing_user)
login_result = self.tester.post('/login',
data={
'form-username':
testing_user.email,
'form-password': '******'
})
logout_result = self.tester.get('/logout')
self.assertEqual(login_result.status_code, 302)
self.assertEqual(logout_result.status_code, 200)
def test_simulation_without_date(self):
testing_user = factory.generate_user()
testing_home = factory.generate_home(None)
self._add_user(testing_user)
self._add_home(testing_home)
self.tester.post('/login',
data={
'form-username': testing_user.email,
'form-password': '******'
})
result = self.tester.post('/simulation', data={'form-start-date': ''})
self.assertIn('You must select a simulation date', str(result.data))
def test_simulation_success(self):
simulation_date = dt.datetime.now().strftime("%Y-%m-%d")
testing_user = factory.generate_user()
testing_home = factory.generate_home(None)
self._add_user(testing_user)
self._add_home(testing_home)
self.tester.post('/login',
data={
'form-username': testing_user.email,
'form-password': '******'
})
result = self.tester.post('/simulation',
data={'form-start-date': simulation_date})
self.assertEqual(result.status_code, 200)
def test_simulation_without_file(self):
simulation_date = Faker().date_this_year(before_today=True,
after_today=False)
simulation_date = simulation_date.strftime("%Y-%m-%d")
testing_user = factory.generate_user()
testing_home = factory.generate_home(None)
self._add_user(testing_user)
self._add_home(testing_home)
self.tester.post('/login',
data={
'form-username': testing_user.email,
'form-password': '******'
})
result = self.tester.post('/simulation',
data={'form-start-date': simulation_date})
self.assertIn('You must upload your Endesa consumption',
str(result.data))
def test_create_fail_user(self):
user = factory.generate_user()
user.email = None
self.db.session.add(user)
self.assertRaises(IntegrityError, self.db.session.commit)
def test_create_user(self):
user = factory.generate_user()
self.db.session.add(user)
self.db.session.commit()
self.assertEqual(self.db.session.query(Users).count(), 1)
