class PlaylistTest(APITest): def setUp(self): super(PlaylistTest, self).setUp() dbconfig['lastfm.api_key'] = 'SEKRET_LASTFM_KEY' self.dj = User(dj_name='DJ Night Moves', first_name='Steve', last_name='Dolfin', email='*****@*****.**', roles=[auth.roles.DJ]) self.dj.save() self.playlist = ChirpBroadcast() (self.stevie, self.talking_book, self.tracks) = create_stevie_wonder_album_data() def play_stevie_song(self, song_name): self.playlist_track = PlaylistTrack( playlist=self.playlist, selector=self.dj, artist=self.stevie, album=self.talking_book, track=self.tracks[song_name], notes='from 1972!', freeform_label='Motown') self.playlist_track.save() return self.playlist_track
class PlaylistTest(APITest): def setUp(self): super(PlaylistTest, self).setUp() dbconfig['lastfm.api_key'] = 'SEKRET_LASTFM_KEY' self.dj = User(dj_name='DJ Night Moves', first_name='Steve', last_name='Dolfin', email='*****@*****.**', roles=[auth.roles.DJ]) self.dj.save() self.playlist = ChirpBroadcast() (self.stevie, self.talking_book, self.tracks) = create_stevie_wonder_album_data() def play_stevie_song(self, song_name): self.playlist_track = PlaylistTrack(playlist=self.playlist, selector=self.dj, artist=self.stevie, album=self.talking_book, track=self.tracks[song_name], notes='from 1972!', freeform_label='Motown') self.playlist_track.save() return self.playlist_track
class APITest(unittest.TestCase): def setUp(self): dbconfig['lastfm.api_key'] = 'SEKRET_LASTFM_KEY' self.client = TestApp(application) self.dj = User(dj_name='DJ Night Moves', first_name='Steve', last_name='Dolfin', email='*****@*****.**', roles=[auth.roles.DJ]) self.dj.save() self.playlist = ChirpBroadcast() (self.stevie, self.talking_book, self.tracks) = create_stevie_wonder_album_data() def tearDown(self): assert memcache.flush_all() clear_data() def play_stevie_song(self, song_name): self.playlist_track = PlaylistTrack( playlist=self.playlist, selector=self.dj, artist=self.stevie, album=self.talking_book, track=self.tracks[song_name], notes='from 1972!', freeform_label='Motown') self.playlist_track.save() return self.playlist_track def request(self, url): r = self.client.get(url) eq_(r.status, '200 OK') eq_(r.headers['content-type'], 'application/json') data = simplejson.loads(r.body) return data