def testGetPlanetByName(self):
randNumber = str(randrange(0, 10000))
name = 'name_test' + randNumber
weather = 'weather_test' + randNumber
terrain = 'terrain_test' + randNumber
#Inserindo o planeta
Dal = Mongo(Planet.collectionName)
self.Planet = Planet(Dal)
self.Planet.setName(name)
self.Planet.setWeather(weather)
self.Planet.setTerrain(terrain)
storedPlanetId = self.Planet.store()
planetRetrieved = self.Planet.getByName()
self.assertEqual(storedPlanetId, planetRetrieved[0]['_id'])
def testRemovePlanet(self):
randNumber = str(randrange(0, 10000))
name = 'name_test' + randNumber
weather = 'weather_test' + randNumber
terrain = 'terrain_test' + randNumber
#Inserindo o planeta
Dal = Mongo(Planet.collectionName)
self.Planet = Planet(Dal)
self.Planet.setName(name)
self.Planet.setWeather(weather)
self.Planet.setTerrain(terrain)
storedPlanetId = self.Planet.store()
self.Planet.setId(str(storedPlanetId))
quantityPlanetsRemoved = self.Planet.remove()
self.assertEqual(1, quantityPlanetsRemoved)
def testTimesPlanetAppeared(self):
randNumber = str(randrange(0, 10000))
name = 'Naboo'
weather = 'weather_test' + randNumber
terrain = 'terrain_test' + randNumber
#Inserindo o planeta
Dal = Mongo(Planet.collectionName)
self.Planet = Planet(Dal)
self.Planet.setName(name)
self.Planet.setWeather(weather)
self.Planet.setTerrain(terrain)
storedPlanetId = self.Planet.store()
#Buscando o planeta
self.Planet.setId(str(storedPlanetId))
planetData = self.Planet.getById()
#Removendo o planeta
self.Planet.remove()
self.assertEqual(4, planetData['timesAppeared'])
def testInsertPlanet(self):
randNumber = str(randrange(0, 10000))
name = 'name_test' + randNumber
weather = 'weather_test' + randNumber
terrain = 'terrain_test' + randNumber
#Inserindo o planeta
Dal = Mongo(Planet.collectionName)
self.Planet = Planet(Dal)
self.Planet.setName(name)
self.Planet.setWeather(weather)
self.Planet.setTerrain(terrain)
storedPlanetId = self.Planet.store()
#Buscando o planeta
self.Planet.setId(str(storedPlanetId))
planetData = self.Planet.getById()
#Comparando o planeta inserido com o buscado
self.assertEqual(storedPlanetId, planetData['_id'])
self.assertEqual(name, planetData['name'])
self.assertEqual(weather, planetData['weather'])
self.assertEqual(terrain, planetData['terrain'])
def testListPlanets(self):
'''
Inserindo 10 planetas para garantir
uma massa de testes
'''
Dal = Mongo(Planet.collectionName)
self.Planet = Planet(Dal)
for x in range(0, 10):
randNumber = str(randrange(0, 10000))
name = 'name_test' + randNumber
weather = 'weather_test' + randNumber
terrain = 'terrain_test' + randNumber
self.Planet.setName(name)
self.Planet.setWeather(weather)
self.Planet.setTerrain(terrain)
storedPlanetId = self.Planet.store()
#Testando 10 limits diferentes
for y in range(0, 10):
limit = random.randrange(1, 10)
planets = self.Planet.getAll(0, limit)
self.assertEqual(limit, len(planets))
#import pymysql
from flask import Flask, request, jsonify
from flask_restful import Resource, Api
from models.planet import Planet
from models.dal.mongo import Mongo
app = Flask(__name__)
Dal = Mongo(Planet.collectionName)
Planet = Planet(Dal)
@app.route('/planets/store', methods=['POST'])
def store():
try:
Planet.setName(request.form['name'])
Planet.setWeather(request.form['weather'])
Planet.setTerrain(request.form['terrain'])
data = Planet.store()
return formatResponse(data)
except Exception as ex:
return formatResponse(ex, False)
@app.route('/planets/getAll/<skip>/<limit>')
def getAll(skip, limit):
try:
data = Planet.getAll(skip, limit)
return formatResponse(data)
except Exception as ex:
def tearDown(self):
Dal = Mongo(self.log.collectionName)
Log = self.log(Dal)
Log.removeAll()
def setUp(self):
Dal = Mongo(self.log.collectionName)
Log = self.log(Dal)
Log.removeAll()
def testInsertLogSuccess(self):
Dal = Mongo(self.log.collectionName)
Log = self.log(Dal)
self.assertEqual(len(str(Log.process('/app/fileToSend/log.txt'))), 24)
def testInsertLogFail(self):
Dal = Mongo(self.log.collectionName)
Log = self.log(Dal)
with self.assertRaises(OSError):
Log.process('/app/fileToSend/arquivoinexistente.txt')
from flask import Flask, request, jsonify, render_template
from flask_restful import Resource, Api
from werkzeug.utils import secure_filename
from models.log import Log
from models.dal.mongo import Mongo
import json
#Ajustar
ALLOWED_EXTENSIONS = set(['txt'])
UPLOAD_FOLDER = '/app/filesReceived'
app = Flask(__name__)
app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
app.config['MAX_CONTENT_LENGTH'] = 16 * 1024 * 1024
Dal = Mongo(Log.collectionName)
Log = Log(Dal)
@app.route('/', methods=['GET'])
def index():
return render_template('index.html')
@app.route('/log/parse', methods=['POST'])
def parse():
f = request.files['file']
if allowed_extensions(f.filename):
filename = secure_filename(f.filename)
f.save(os.path.join(app.config['UPLOAD_FOLDER'], filename))
response = Log.process(app.config['UPLOAD_FOLDER'] + "/" + f.filename)