def init(self):
if self.cookiefile is not None:
self.cookiejar = cookielib.LWPCookieJar(self.cookiefile)
if os.path.exists(self.cookiefile):
try:
self.cookiejar.load(ignore_discard=True)
except cookielib.LoadError:
log.warn('error loading cookies')
#print 'start cookies %r' % self.cookiejar
self.mss = HTTPMetawebSession(self.service_host,
cookiejar=self.cookiejar)
def test_login_logout(self):
mss = HTTPMetawebSession(API_HOST, username=USERNAME,
password=PASSWORD)
mss.login()
user_info = mss.user_info()
self.assertNotEqual(None, user_info)
self.assertEqual(user_info.code, "/api/status/ok")
self.assertEqual(mss.loggedin(), True)
mss.logout()
self.assertRaises(MetawebError, mss.user_info)
self.assertEqual(mss.loggedin(), False)
def test_read(self):
query = {'type':'/music/artist','guid':[{}],'name':'Sting', 'album':[{}]}
mss = HTTPMetawebSession(API_HOST)
result = mss.mqlread(query)
self.assertNotEqual(None, result)
self.assert_(result.has_key('guid'))
self.assert_(result.has_key('type'))
self.assert_(result.has_key('name'))
self.assert_(result.has_key('album'))
self.assertEqual(type([]), type(result['album']))
self.assert_(len(result['album']) > 0)
self.assertEqual( 'Sting', result['name'])
self.assertEqual('#9202a8c04000641f8000000000092a01', result['guid'][0]['value'])
def test_write(self):
read_query = {'type':'/music/artist','name':'Yanni\'s Cousin Tom', 'id':{}}
mss = HTTPMetawebSession(API_HOST, username=USERNAME, password=PASSWORD)
result = mss.mqlread(read_query)
self.assertEqual(None, result)
write_query = {'create':'unless_exists', 'type':'/music/artist','name':'Yanni'}
mss.login()
write_result = mss.mqlwrite(write_query)
self.assertNotEqual(None, write_result)
self.assert_(write_result.has_key('create'))
self.assert_(write_result.has_key('type'))
self.assert_(write_result.has_key('name'))
self.assertEqual('existed', write_result['create'])
self.assertEqual('Yanni', write_result['name'])
self.assertEqual('/music/artist', write_result['type'])
import unittest
import sys, logging
import freebase
import random
import getlogindetails
from freebase.api import HTTPMetawebSession, MetawebError
USERNAME = '******'
PASSWORD = '******'
API_HOST = 'http://sandbox-freebase.com'
TEST_QUERY = {'id': 'null', 'name': 'Sting'}
s = HTTPMetawebSession(API_HOST)
if USERNAME == "username" and PASSWORD == "password":
USERNAME, PASSWORD = getlogindetails.main()
s.login(USERNAME, PASSWORD)
class TestFreebase(unittest.TestCase):
def test_freebase_dot_login_logout(self):
freebase.login(username=USERNAME, password=PASSWORD)
self.assertNotEqual(freebase.user_info(), None)
self.assertEqual(freebase.loggedin(), True)
freebase.logout()
self.assertRaises(MetawebError, freebase.user_info)
self.assertEqual(freebase.loggedin(), False)
