def setup(self):
user = self.helper('create_user')
create_album_for(user.get_profile(), user, PhotoAlbum)
super(TestPhoto, self).setup()
global TEST_PHOTO_COPY
TEST_PHOTO_COPY = tempfile.NamedTemporaryFile(dir=TEST_FILES,
suffix='.png')
shutil.copy2(TEST_PHOTO, TEST_PHOTO_COPY.name)
def album_detail(request, album):
"""get by slug, or return for username
if user requests his own album, create it
"""
try:
album = PhotoAlbum.objects.get(slug=album)
except PhotoAlbum.DoesNotExist:
try:
album_user = User.objects.get(username=album)
album = PhotoAlbum.objects.get(
user=album_user, object_pk=album_user.get_profile().id)
except (PhotoAlbum.DoesNotExist, User.DoesNotExist):
if request.user == album_user:
# using root function instead of signal
album = create_album_for(album_user.get_profile(), album_user,
PhotoAlbum)
else:
raise Http404
return render_to_response('photos/album_detail.html', {'album': album},
context_instance=RequestContext(request))
def test_update(self):
user = self.helper('create_user')
create_album_for(user.get_profile(), user, PhotoAlbum)
album = self.assert_read(PhotoAlbum, slug=SLUG)
self.assert_update(album, name='Super puper album')
self.assert_read(PhotoAlbum, name='Super puper album', slug=SLUG)
def test_delete(self):
user = self.helper('create_user')
create_album_for(user.get_profile(), user, PhotoAlbum)
album = self.assert_read(PhotoAlbum, slug=SLUG)
self.assert_delete(album)
def create_photo_album(sender, instance, **kwargs):
"""sender should be userprofile"""
create_album_for(instance, instance.user, model = PhotoAlbum, **kwargs)