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)