jpt-archive/oyster
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

track_url with same parameters returns original id

Browse Source
This commit is contained in:
James Turk 2011-09-28 15:50:32 -04:00
parent da1a3385e1
commit 51480fafcc
2 changed files with 20 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
oyster/client.py
View File

@ -84,9 +84,19 @@ class Client(object):
url
URL to start tracking
"""
if self.db.tracked.find_one({'url': url}):
self.log('track', url=url, error='already tracked')
raise ValueError('%s is already tracked' % url)
tracked = self.db.tracked.find_one({'url': url})
# if data is already tracked and this is just a duplicate call
# return the original object
if tracked:
if (tracked['metadata'] == kwargs and
tracked['versioning'] == versioning and
tracked['update_mins'] == update_mins):
return tracked['_id']
else:
self.log('track', url=url, error='tracking conflict')
raise ValueError('%s is already tracked with different '
'metadata' % url)
self.log('track', url=url)
return self.db.tracked.insert(dict(url=url, versioning=versioning,

10
oyster/tests/test_client.py
View File

@ -42,7 +42,7 @@ class ClientTests(TestCase):
def test_track_url(self):
# basic insert
self.client.track_url('http://example.com', update_mins=30, pi=3)
id1 = self.client.track_url('http://example.com', update_mins=30, pi=3)
obj = self.client.db.tracked.find_one()
assert '_random' in obj
assert obj['update_mins'] == 30
@ -53,11 +53,15 @@ class ClientTests(TestCase):
assert log['action'] == 'track'
assert log['url'] == 'http://example.com'
# can't track same URL twice
# track same url again with same metadata returns id
id2 = self.client.track_url('http://example.com', update_mins=30, pi=3)
assert id1 == id2
# can't track same URL twice with different metadata
assert_raises(ValueError, self.client.track_url, 'http://example.com')
# logged error
assert self.client.db.logs.find_one({'error': 'already tracked'})
assert self.client.db.logs.find_one({'error': 'tracking conflict'})
def test_md5_versioning(self):