def test_resolve_url_hits_counter(self):
tables = ShortenerTables(self.account, self.conn)
yield tables.create_tables()
url = 'http://en.wikipedia.org/wiki/Cthulhu'
yield self.service.shorten_url(url)
yield self.service.get_row_by_short_url('qr0')
yield self.service.get_row_by_short_url('qr0')
result = yield self.service.get_row_by_short_url('qr0')
audit = yield tables.get_audit_row(result['id'])
self.assertEqual(audit['hits'], 3)
def init_account(self, request):
'''
Initializes the account and creates the database tables
'''
conn = yield self.engine.connect()
account = self.config['account']
tables = ShortenerTables(account, conn)
try:
already_exists = yield tables.exists()
if not already_exists:
yield tables.create_tables()
finally:
yield conn.close()
returnValue({'created': not already_exists})
def init_account(self, request):
'''
Initializes the account and creates the database tables
'''
conn = yield self.engine.connect()
account = self.config['account']
tables = ShortenerTables(account, conn)
try:
already_exists = yield tables.exists()
if not already_exists:
yield tables.create_tables()
finally:
yield conn.close()
returnValue({'created': not already_exists})
def test_get_or_create_row(self):
tables = ShortenerTables('test-account', self.conn)
yield tables.create_tables()
row = yield tables.get_or_create_row(
'wiki.org', 'test', 'http://wiki.org/test/')
self.assertEqual(row['domain'], 'wiki.org')
self.assertEqual(row['short_url'], None)
self.assertEqual(row['user_token'], 'test')
self.assertEqual(row['long_url'], 'http://wiki.org/test/')
self.assertEqual(row['id'], 1)
row = yield tables.get_or_create_row(
'wiki.org', 'test', 'http://wiki.org/test/')
self.assertEqual(row['id'], 1)
audit = yield tables.get_audit_row(1)
self.assertEqual(audit['hits'], 0)
def test_get_or_create_row(self):
tables = ShortenerTables('test-account', self.conn)
yield tables.create_tables()
row = yield tables.get_or_create_row('wiki.org', 'test',
'http://wiki.org/test/')
self.assertEqual(row['domain'], 'wiki.org')
self.assertEqual(row['short_url'], None)
self.assertEqual(row['user_token'], 'test')
self.assertEqual(row['long_url'], 'http://wiki.org/test/')
self.assertEqual(row['id'], 1)
row = yield tables.get_or_create_row('wiki.org', 'test',
'http://wiki.org/test/')
self.assertEqual(row['id'], 1)
audit = yield tables.get_audit_row(1)
self.assertEqual(audit['hits'], 0)
def test_resolve_url(self):
tables = ShortenerTables('test-account', self.conn)
yield tables.create_tables()
yield tables.get_or_create_row(
'wiki.org', 'test', 'http://wiki.org/test/')
yield tables.update_short_url(1, 'aaa')
row = yield tables.get_row_by_short_url('aaa')
self.assertEqual(row['domain'], 'wiki.org')
self.assertEqual(row['short_url'], 'aaa')
self.assertEqual(row['id'], 1)
audit = yield tables.get_audit_row(1)
self.assertEqual(audit['hits'], 1)
#multiple hits
for i in range(0, 10):
yield tables.get_row_by_short_url('aaa')
audit = yield tables.get_audit_row(1)
self.assertEqual(audit['hits'], 11)
def test_resolve_url(self):
tables = ShortenerTables('test-account', self.conn)
yield tables.create_tables()
yield tables.get_or_create_row('wiki.org', 'test',
'http://wiki.org/test/')
yield tables.update_short_url(1, 'aaa')
row = yield tables.get_row_by_short_url('aaa')
self.assertEqual(row['domain'], 'wiki.org')
self.assertEqual(row['short_url'], 'aaa')
self.assertEqual(row['id'], 1)
audit = yield tables.get_audit_row(1)
self.assertEqual(audit['hits'], 1)
#multiple hits
for i in range(0, 10):
yield tables.get_row_by_short_url('aaa')
audit = yield tables.get_audit_row(1)
self.assertEqual(audit['hits'], 11)
def test_tables_create(self):
tables = ShortenerTables('test-account', self.conn)
self.successResultOf(tables.create_tables())
def test_tables_create(self):
tables = ShortenerTables('test-account', self.conn)
self.successResultOf(tables.create_tables())
