-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathtest_funload.py
101 lines (73 loc) · 4.27 KB
/
test_funload.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
#!/usr/bin/env python3
import xml.etree.ElementTree as ElemT
import unittest
import os
import datetime
import funload
def get_test_file(testfile):
return open(os.path.expanduser("test_funload_files/" + testfile), 'r')
def get_test_xml(testfile):
return ElemT.parse(get_test_file(testfile))
def get_item(testfile):
return get_test_xml(testfile).getroot().find('channel').find('item')
class MockConfig(funload.Config):
files = {}
def write(self, name, content):
self.files[name] = content
def get_last_build(self, name):
return self.files[name]
class TestHornoxe(unittest.TestCase):
def test_multiple_youtube_links(self):
item = get_item("test_multiple_youtube_links")
date = datetime.datetime(2007, 12, 5)
urls = funload.item_node_parse(item, date)
self.assertEqual(urls, ["https://www.youtube.com/embed/Vx-1LQu6mAE?rel=0&showinfo=0",
"https://www.youtube.com/embed/_df8tHFOxgg?rel=0&showinfo=0",
"https://www.youtube.com/embed/BRvfZmwwHF0?rel=0&showinfo=0"])
def test_buzz_driver_link(self):
item = get_item("test_buzz_driver_link")
date = datetime.datetime(2007, 12, 6)
urls = funload.item_node_parse(item, date)
self.assertEquals(urls, ["3TYlQjkH1OE"])
class TestEmok(unittest.TestCase):
def test_get_all_new_emok_video_pages__updates_config(self):
config = MockConfig()
xml = get_test_xml("test_emok_videofeed")
config.write('emok', datetime.datetime(2007, 12, 5))
funload.get_all_new_emok_video_pages(config, 'emok', xml)
self.assertEquals(config.get_last_build('emok'), "Sun, 22 Mar 2015 11:50:52 +0000")
def test_get_all_new_emok_video_pages(self):
config = MockConfig()
xml = get_test_xml("test_emok_videofeed")
config.write('emok', datetime.datetime(2007, 12, 5))
urls = funload.get_all_new_emok_video_pages(config, 'emok', xml)
self.assertEqual(urls, ["http://www.emok.tv/own-content/monsterbumserin-iii.html",
"http://www.emok.tv/videos/the-walking-dead-russland-edition.html",
"http://www.emok.tv/videos/profi-fuellt-sein-zippo-nach.html",
"http://www.emok.tv/videos/frau-beim-staubsaugen.html",
"http://www.emok.tv/videos/betrunkene-fail-compilation-3.html",
"http://www.emok.tv/videos/piranhas-fuettern.html",
"http://www.emok.tv/videos/abheben-beim-rennen.html",
"http://www.emok.tv/videos/wiesel-will-spielen.html",
"http://www.emok.tv/own-content/racing-grannies.html",
"http://www.emok.tv/videos/varoufakis-und-der-stinkefinger.html",
"http://www.emok.tv/videos/too-many-zooz.html",
"http://www.emok.tv/videos/waschbaer-nimmt-ganz-behutsam-das-leckerli.html",
"http://www.emok.tv/videos/evolution-der-videospiele.html",
"http://www.emok.tv/videos/besoffen-und-auf-dem-weg-nach-hause.html",
"http://www.emok.tv/videos/betrunkenes-eichhoernchen.html",
"http://www.emok.tv/videos/photoshop-experten-probieren-photoshop-1-0-aus.html",
"http://www.emok.tv/videos/uptown-funk-auf-dem-laufband.html",
"http://www.emok.tv/videos/dragon-ball-z-light-of-hope.html",
"http://www.emok.tv/videos/salchichitas-envueltas-kochen.html",
"http://www.emok.tv/emo/emo-der-woche-309.html"])
def test_extract_emok_video_urls_nourl(self):
file = get_test_file("test_emok_nourl")
urls = funload.extract_emok_video_urls(file)
self.assertEquals(urls, [])
def test_extract_emok_video_urls_oneurl(self):
file = get_test_file("test_emok_oneurl")
urls = funload.extract_emok_video_urls(file)
self.assertEquals(urls, ["http://www.emok.tv/wp-content/2015/03/walkinddead.mp4"])
if __name__ == '__main__':
unittest.main()