def test_term_search(self):
"""
Test twitter term search
"""
lbt = lbtwitter.Twitter(debug=True, term='crime')
status = lbt.search()
assert len(status) > 0
def test_twitter_hastag(self):
"""
Test reading twitter hashtag
"""
lbt = lbtwitter.Twitter(debug=True, term='#tvbrasilia')
hashtag = lbt.hashtag
assert isinstance(hashtag, twitter.Hashtag)
def test_twitter_config(self):
"""
Test loading twitter configuration
"""
lbt = lbtwitter.Twitter(debug=True)
api = lbt.api
results = api.VerifyCredentials()
assert isinstance(results, twitter.User)
def test_json_results(self):
"""
Test convert results to JSON format
"""
lbt = lbtwitter.Twitter(debug=True, term='crime')
status = lbt.search()
json_status = lbt.status_to_json(status)
fd = open('/tmp/status.json', 'w+')
fd.write(json_status)
fd.close()
assert json.loads(json_status)
def __init__(self, name):
"""
Constructor method
"""
super(TwitterCommands, self).__init__(name)
# Set base to production
self.status_base = lbstatus.StatusBase(status_name='status',
dic_name='dictionary')
self.dictionary_base = dicbase.DictionaryBase(dic_base='dictionary')
self.lbt = lbtwitter.Twitter(debug=False,
term='crime',
status_base=self.status_base,
dictionary_base=self.dictionary_base)
def setUp(self):
"""
Load test data
"""
test_twitter_import.TwitterImportTestCase.setUp(self)
self.lbs = lbsociam.LBSociam()
self.baserest = lbrest.BaseREST(rest_url=self.lbs.lbgenerator_rest_url, response_object=True)
self.lbt = lbtwitter.Twitter(debug=False, term='crime')
self.status_base = lbstatus.StatusBase()
self.tw_status = self.lbt.search()
# Debug
fd = open('/tmp/status_base.json', 'w+')
fd.write(self.status_base.lbbase.json)
fd.close()
# Cria base
self.lbbase = self.status_base.create_base()
pass
def setUp(self):
"""
Ajusta parâmetros iniciais
"""
self.lbs = lbsociam.LBSociam()
self.baserest = lbrest.BaseREST(rest_url=self.lbs.lbgenerator_rest_url,
response_object=True)
self.lbt = lbtwitter.Twitter(debug=False, term='crime')
self.status_base = lbstatus.StatusBase()
self.tw_status = self.lbt.search()
self.data_dir = os.path.join(lbs.lbsociam_data_dir, 'tests')
if not os.path.isdir(self.data_dir):
os.mkdir(self.data_dir)
# Cria base
self.lbbase = self.status_base.create_base()
# Insere dois status
self.status = list()
tw_status_elm = [self.tw_status[0]]
tw_status_json = self.lbt.status_to_json(tw_status_elm)
status_dict = dict(
origin='twitter',
inclusion_date=datetime.datetime.now().strftime("%d/%m/%Y"),
text=tw_status_elm[0].text,
source=tw_status_json,
search_term='crime')
tokenized = srl.srl_tokenize(tw_status_elm[0].text)
if tokenized.get('arg_structures'):
status_dict['arg_structures'] = tokenized.get('arg_structures')
if tokenized.get('tokens'):
status_dict['tokens'] = tokenized.get('tokens')
status = conv.dict2document(self.lbbase, status_dict)
status_json = conv.document2json(self.lbbase, status)
result = self.status_base.documentrest.create(status_json)
self.status.append(status)
# Segundo status
tw_status_elm = [self.tw_status[1]]
tw_status_json = self.lbt.status_to_json(tw_status_elm)
status_dict = dict(
origin='twitter',
inclusion_date=datetime.datetime.now().strftime("%d/%m/%Y"),
text=tw_status_elm[0].text,
source=tw_status_json,
search_term='crime')
tokenized = srl.srl_tokenize(tw_status_elm[0].text)
if tokenized.get('arg_structures'):
status_dict['arg_structures'] = tokenized.get('arg_structures')
if tokenized.get('tokens'):
status_dict['tokens'] = tokenized.get('tokens')
status = conv.dict2document(self.lbbase, status_dict)
status_json = conv.document2json(self.lbbase, status)
result = self.status_base.documentrest.create(status_json)
self.status.append(status)
# Base de dicionário
self.dictionary_base = model_dict.DictionaryBase()