skynet_tests.py

"""
	Suit of tests for server of the SkyNet social network.
"""
import os
import sys
import unittest
import flask
import flask_testing

import skynet_server as ss
#Modulos necesario para usar SQLAlchemy y base de datos
from sqlalchemy import create_engine, text, and_, or_
from sqlalchemy.sql import exists
from sqlalchemy.orm import sessionmaker
from database_setup import Base, Usuario, Fotos, Publicacion, Amigos, Etiquetas, Chat, Mensaje

class SkynetTestCase(unittest.TestCase):
	def setUp(self):
		ss.app.config['TESTING'] = True
		self.app = ss.app.test_client()
		self.app.secret_key = 'super_secret_key'
		engine = create_engine("mysql+pymysql://root:12345@localhost/")
		engine.execute("USE skynet")
		Base.metadata.bind = engine
		#Crear sesion para comunicarse con la base de datos
		self.DBSession = sessionmaker(bind = engine)

	def tearDown(self):
		pass

	def login(self, email, password, route):
		"""For testing the log in"""
		return self.app.post(route, data = dict(email = email, password = password), follow_redirects = True)

	def test_index(self):
		"""
		Test the method index. If the method of the request is GET, just verify that the returned HTML
		is correctly formatted. Check if the strings 'Username', 'Password' and 'Login' are present in the file.
		If the methos is POST, there are many things to verify:
			-If there is no password or email on the request form, indicate that a field is missing.
			-If the format of the email is incorrect, indicate to the user that email format is wrong
			-If the user is not in the database or the password is incorrect: indicate it to the user.
			-If all went fine, check that the status of the user is correctly updated on the data base and the 
			returned template is the adecuate.
		"""
		session = self.DBSession()
		#Tests for GET method
		rv = self.app.get("/")
		assert "Username" in rv.data
		assert "Password" in rv.data
		assert "Login" in rv.data
		rv = self.app.get("/index")
		assert "Username" in rv.data
		assert "Password" in rv.data
		assert "Login" in rv.data

		#Tests for POST method
		#Testing log in
		rv = self.login("johns@correo.com", "", "/")
		assert "Missing field" in rv.data
		rv = self.login("", "12345", "/")
		assert "Missing field" in rv.data
		rv = self.login("johns", "12345", "/")
		assert "Incorrect email format" in rv.data
		rv = self.login("john@correo.com", "12345", "/")
		assert "User not registered or incorrect password" in rv.data
		rv = self.login("johns@correo.com", "1234", "/")
		assert "User not registered or incorrect password" in rv.data

		rv = self.login("johns@correo.com", "12345", "/")
		#Get the user from the database
		user = session.query(Usuario).filter(Usuario.email == "johns@correo.com").one()
		assert user.conectado == True
		assert user.disponibilidad == True
		#Test correct redirect
		assert "johns" in rv.data

	def register(self, name, lastname, email, password):
		return self.app.post("/register", data = dict(name = name, LastName = lastname, email = email, \
			password = password), follow_redirects = True)

	def test_register(self):
		"""
		Test the method register. If the method of the request is GET, just verify that the returned HTML
		is correctly formatted. Check if the strings 'First Name', 'Last Name', 'Your Email', 'Password'
		and 'Register' are present in the file.
		If the methos is POST, there are many things to verify:
			-If there is no first name, last name, email or password on the request form, indicate that a field is missing.
			-If the format of the email is incorrect, indicate to the user that email format is wrong
			-If the user is already in the database: indicate it to the user.
			-If all went fine, check that the user is correctly added to the database and the
			status of the user is correctly updated on the data base and the returned template is the adecuate.
		"""
		session = self.DBSession()
		#Test the GET method
		rv = self.app.get("/register")
		assert "First Name" in rv.data
		assert "Last Name" in rv.data
		assert "Your Email" in rv.data
		assert "Password" in rv.data
		assert "Register" in rv.data

		#Test the POST method
		#Check for missing fields
		rv = self.register("", "", "", "")
		assert "Missing field" in rv.data
		rv = self.register("Angry", "Bird", "blue_bird@angry.ab", "")
		assert "Missing field" in rv.data
		rv = self.register("Angry", "Bird", "", "angryme")
		assert "Missing field" in rv.data
		rv = self.register("Angry", "", "blue_bird@angry.ab", "angryme")
		assert "Missing field" in rv.data
		rv = self.register("", "Bird", "blue_bird@angry.ab", "angryme")
		assert "Missing field" in rv.data

		#Incorrect email format
		rv = self.register("Angry", "Bird", "blue_bird@angry", "angryme")
		assert "Incorrect email format" in rv.data

		#Already registered user
		rv = self.register("John", "Snow", "johns@correo.com", "12345")
		assert "User already register" in rv.data

		#Check correct add of user to database and update status
		rv = self.register("Angry", "Bird", "blue_bird@angry.ab", "angryme")
		q = session.query(exists().where(Usuario.username == "blue_bird")).scalar()
		assert q == True
		user = session.query(Usuario).filter(Usuario.email == "blue_bird@angry.ab").one()
		assert user.nombre == "Angry"
		assert user.apellido == "Bird"
		assert user.email == "blue_bird@angry.ab"
		assert user.contrasena == "angryme"
		assert user.username == "blue_bird"
		assert user.conectado == True
		assert user.disponibilidad == True

	def test_userlogoutMethod(self):
		"""
		Test the logout function. Only accepts the GET method.
		"""
		session = self.DBSession()
		rv = self.app.get("/johns/logout", follow_redirects = True)
		user = session.query(Usuario).filter(Usuario.username == "johns").one()
		assert user.conectado == False
		assert user.disponibilidad == False

		assert "Username" in rv.data
		assert "Password" in rv.data
		assert "Login" in rv.data

	def test_getImage(self):
		"""
		Test the function getImage. If the user has no image, returns None. Otherwise returns the url for it.
		"""
		#Get the image of user 5 (Jose Septien)
		img = ss.getImage(5, True)
		assert img == None
		img = ss.getImage(1, True)
		assert img == 'public/images/cnt.jpg'

	def test_getContactos(self):
		"""
		Test the function getContactos. Should return their information and ids.
		"""
		(contactos, ids) = ss.getContactos(1)
		assert ids[0] == 2
		assert ids[1] == 3
		assert ids[2] == 4
		
		cont = contactos[0]
		assert cont["name"] == 'John Snow'
		assert cont["img"] == 'public/images/cnt3.jpg'
		assert cont["username"] == 'johns'

		cont = contactos[1]
		assert cont["name"] == 'Sakura Card Captor'
		assert cont["img"] == 'public/images/cnt1.jpg'
		assert cont["username"] == 'sakura'

		cont = contactos[2]
		assert cont["name"] == 'Sofia Fiorelli'
		assert cont["img"] == 'public/images/cnt034.jpg'
		assert cont["username"] == 'sfiorelli'

	def test_getPublicaciones(self):
		"""
		Test the function getPublicaciones. Should return image of the user, name, text, date, username and idicate if
		the publication correspond to the main user.
		"""
		publicaciones = ss.getPublicaciones(5, True)

		p = publicaciones[0]
		assert p["name"] == "Jose Septien"
		assert p["text"] == "Hola a todos!"
		assert str(p["fecha"]) == "2016-12-06 15:04:21"
		assert p["username"] == "jaseptienh"
		assert p["user"] == True
		assert p["img"] == None

		publicaciones = ss.getPublicaciones(5, False)
		p = publicaciones[0]
		assert p["name"] == "Jose Septien"
		assert p["text"] == "Hola a todos!"
		assert str(p["fecha"]) == "2016-12-06 15:04:21"
		assert p["username"] == "jaseptienh"
		assert p["user"] == False
		assert p["img"] == None

		publicaciones = ss.getPublicaciones(12, True)
		assert publicaciones == None

	def test_home(self):
		"""
		Test the method home. Only accepts the GET method.
			-If the user does not exits on database, it should indicate it.
			-If the user is not connected, it should return to the index page.
		"""
		rv = self.app.get("/juan/home")
		assert "User not registered" in rv.data

		rv = self.app.get("/sakura/home", follow_redirects = True)
		assert "Username" in rv.data
		assert "Password" in rv.data
		assert "Login" in rv.data

		rv = self.app.get("/johns/home")
		assert "johns" in rv.data

	def test_profile(self):
		"""
		Test the method profile. Accepts methods GET and POST.
		If the method is GET, check if the user is connected. If it is, return the profile template.
			If not, redirect to the index page.
		If the method is POST, check first if the user is not connected. redirect to indes.
			If it is connected, check if the message sent is empty, in such, indicate it to the user.
			If not, add it to the database.
		"""
		rv = self.app.get("/sakura/profile", follow_redirects = True)
		assert "Username" in rv.data
		assert "Password" in rv.data
		assert "Login" in rv.data

		rv = self.app.get("/johns/profile")
		assert "John Snow" in rv.data
		assert "johns" in rv.data

		rv = self.app.post("/sakura/profile", data = dict(publish = "jajaja"), follow_redirects = True)
		assert "Username" in rv.data
		assert "Password" in rv.data
		assert "Login" in rv.data

		rv = self.app.post("/johns/profile", data = dict(publish = ""), follow_redirects = True)
		assert "No text added" in rv.data

		rv = self.app.post("/johns/profile", data = dict(publish = "Publication test."), follow_redirects = True)
		assert "Publication test." in rv.data

	def test_friend(self):
		"""
		Test the friend function. Accept POST and GET method. If the method is GET, check for the user to be connected.
		If so, returns to home page. If not, returns to index.
		If method is POST:
			-Check if something was introduced, if not, it indicates to the user.
			-Check if the user exists on database, if not, it tells to the user.
			-Get all the user from database with a string like the one introduced on their name, last name or username.
		"""
		rv = self.app.get("/sakura/friend", follow_redirects = True)
		assert "Username" in rv.data
		assert "Password" in rv.data
		assert "Login" in rv.data

		rv = self.app.get("/johns/friend")
		assert "HOME" in rv.data

		rv = self.app.post("/sakura/friend", data = dict(user = "s"), follow_redirects = True)
		assert "Username" in rv.data
		assert "Password" in rv.data
		assert "Login" in rv.data

		rv = self.app.post("johns/friend", data = dict(user = ""), follow_redirects = True)
		assert "User not introduced" in rv.data

		rv = self.app.post("johns/friend", data = dict(user = "pikachu"), follow_redirects = True)
		assert "User not found" in rv.data

		rv = self.app.post("johns/friend", data = dict(user = "s"), follow_redirects = True)
		assert "Luis Perez" in rv.data
		assert "Sakura Card Captor" in rv.data
		assert "Sofia Fiorelli" in rv.data

	def test_contact(self):
		"""
		Function to test the method contact. Accept only GET method.
		"""
		rv = self.app.get("/sakura/friend/johns", follow_redirects = True)
		assert "Username" in rv.data
		assert "Password" in rv.data
		assert "Login" in rv.data

		rv = self.app.get("/johns/friend/pikachu", follow_redirects = True)
		assert "Friend not existing" in rv.data

		rv = self.app.get("/johns/friend/sakura", follow_redirects = True)
		assert "Sakura Card Captor" in rv.data

if __name__ == '__main__':
	unittest.main()