def test_appendToUserAgents(self):
parser = UserAgentParser()
firefox = parser._parseUserAgents(parser.SOFTWARE['firefox'])
parser._appendToUserAgents(firefox)
self.assertListEqual(firefox, parser._userAgents)
opera = parser._parseUserAgents(parser.SOFTWARE['opera'])
parser._appendToUserAgents(opera)
for item in opera:
self.assertIn(item, parser._userAgents)
for item in firefox:
self.assertIn(item, parser._userAgents)
import json
import sys
import os
from flask import request, Response
skyenv = os.environ.get('SKYENV', '/home/skyenv/')
sys.path.append(skyenv)
from skyScanner.searchingEngine import app
from skyScanner.parsers.ProxyParser import ProxyParser
from skyScanner.parsers.UserAgentParser import UserAgentParser
from skyScanner.searchingEngine.Search import Search
userAgentParser = UserAgentParser()
proxyParser = ProxyParser()
searchingTasks = {}
@app.route('/search/<int:searchId>', methods=['GET'])
def getSearch(searchId):
try:
search = {
'sourceCity': searchingTasks[searchId].sourceCity,
'targetCity': searchingTasks[searchId].targetCity,
'date': searchingTasks[searchId].date
}
return Response(json.dumps(search),
status=200,
mimetype='application/json')
def test_createUserAgents(self):
parser = UserAgentParser()
parser._createUserAgents()
self.assertTrue(parser._hasUserAgents)
self.assertNotEqual([], parser._userAgents)
def test_parseUserAgentsRaisesExeption(self):
parser = UserAgentParser()
with self.assertRaises(ValueError):
parser._parseUserAgents('Bad string')
def test_getUserAgents(self):
parser = UserAgentParser()
userAgents = parser.getUserAgents()
self.assertListEqual(userAgents, parser._userAgents)
def test_getRandomUserAgentEvoke__createUserAgents(self):
parser = UserAgentParser()
parser.getRandomUserAgent()
self.assertTrue(parser._hasUserAgents)
self.assertNotEqual([], parser._userAgents)
def test_getRandomUserAgent(self):
parser = UserAgentParser()
randomChoice = parser.getRandomUserAgent()
self.assertIn(randomChoice, parser._userAgents)
def test_clearUserAgents(self):
parser = UserAgentParser()
parser.getUserAgents()
parser._clearUserAgents()
self.assertFalse(parser._hasUserAgents)
self.assertListEqual([], parser._userAgents)
def test_initialization(self):
parser = UserAgentParser()
self.assertFalse(parser._hasUserAgents)
self.assertListEqual([], parser._userAgents)