def test_save_shortLink(self):
counter = random.randrange(0, 100000)
url = "xlarrakoetxea.org"
creation_date = dateutils.unix_now_utc()
clicks = 20
title = "This is a title"
host = url
disabled = True
# Save the links
sl = ShortLink(counter=counter, url=url, creation_date=creation_date,
clicks=clicks, title=title, host=host)
sl.disabled = disabled
sl.save()
r = redis.StrictRedis(host=settings.REDIS_HOST,
port=settings.REDIS_PORT,
db=settings.REDIS_DB)
# Construct the Keys
rtk = ShortLink.REDIS_TOKEN_KEY.format(utils.counter_to_token(counter))
ruk = ShortLink.REDIS_URL_KEY.format(url)
# Check
token = utils.counter_to_token(counter)
self.assertTrue(token in r.smembers(ruk))
keys = ('url', 'creation_date', 'clicks', 'title', 'host')
data = [url, creation_date, clicks, title, host]
aux = r.hmget(rtk, keys)
data_result = [aux[0], int(aux[1]), int(aux[2]), aux[3], aux[4]]
self.assertEquals(data, data_result)
def test_shortlink_basic_object(self):
url = "xlarrakoetxea.org"
counter = random.randrange(0, 100000)
token = utils.counter_to_token(counter)
creation_date = None
clicks = 0
title = "This is a title"
host = url
disabled = True
# Setters from counter
sl = ShortLink(counter=counter, url=url, title=title, host=host)
sl.disabled = disabled
# Getters
self.assertEquals(url, sl.url)
self.assertEquals(counter, sl.counter)
self.assertEquals(token, sl.token)
self.assertEquals(creation_date, sl.creation_date)
self.assertEquals(clicks, sl.clicks)
self.assertEquals(title, sl.title)
self.assertEquals(host, sl.host)
self.assertEquals(disabled, sl.disabled)
# Setters from token
sl2 = ShortLink(token=token, url=url)
creation_date = dateutils.unix_now_utc()
sl2.creation_date = creation_date
clicks = 5
sl2.clicks = clicks
sl2.title = title
sl2.host = host
sl2.disabled = disabled
# Getters
self.assertEquals(url, sl2.url)
self.assertEquals(counter, sl2.counter)
self.assertEquals(token, sl2.token)
self.assertEquals(creation_date, sl2.creation_date)
self.assertEquals(clicks, sl2.clicks)
self.assertEquals(title, sl2.title)
self.assertEquals(host, sl2.host)
self.assertEquals(disabled, sl2.disabled)
def test_shortlink_basic_object_str(self):
url = "xlarrakoetxea.org"
counter = random.randrange(0, 100000)
token = utils.counter_to_token(counter)
creation_date = None
clicks = 0
title = "This is a title"
host = url
disabled = True
format = ShortLink.OBJECT_STR_FORMAT.format(counter,
token,
url,
creation_date,
clicks,
title,
host,
disabled)
sl = ShortLink(counter=counter, url=url, title=title, host=host)
sl.disabled = disabled
self.assertEquals(format, str(sl))
