Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 4 additions & 0 deletions extlinks/links/management/commands/linkevents_collect.py
Original file line number Diff line number Diff line change
Expand Up @@ -80,6 +80,9 @@ def _process_events(self, url):
# A tuple value sets each respective value independently.
# https://requests.readthedocs.io/en/latest/user/advanced/#timeouts
timeout=(3.05, 7),
headers={
"User-Agent": 'Wikilink'
}
):
if event.event == "message":
try:
Expand Down Expand Up @@ -208,4 +211,5 @@ def _add_linkevent_to_db(self, link, change, event_data):
)
for url_pattern in url_patterns:
url_pattern.link_events.add(link_event)
url_pattern.collections.add(this_link_collection)
url_pattern.save()
3 changes: 3 additions & 0 deletions extlinks/links/tests.py
Original file line number Diff line number Diff line change
Expand Up @@ -291,18 +291,21 @@ def test_management_command_non_proxy(self):
with self.assertRaises(SystemExit):
call_command("linkevents_collect", test=self.event_data1)
self.assertEqual(LinkEvent.objects.count(), 1)
self.assertEqual("JSTOR", URLPattern.objects.first().collections.first().name)

def test_management_command_proxy_urls(self):
self.assertEqual(LinkEvent.objects.count(), 0)
with self.assertRaises(SystemExit):
call_command("linkevents_collect", test=self.event_data2)
self.assertEqual(LinkEvent.objects.count(), 2)
self.assertEqual("JSTOR", URLPattern.objects.first().collections.first().name)

def test_management_command_dates_with_micro_seconds(self):
self.assertEqual(LinkEvent.objects.count(), 0)
with self.assertRaises(SystemExit):
call_command("linkevents_collect", test=self.event_data3)
self.assertEqual(LinkEvent.objects.count(), 2)
self.assertEqual("JSTOR", URLPattern.objects.first().collections.first().name)

class LinkEventsArchiveCommandTest(TransactionTestCase):
def setUp(self):
Expand Down