def test_get_album_exists(self):
a = Album.get_or_create_default_for_user(username='******')
self.assertEqual(a.name, settings.DEFAULT_ALBUM_NAME)
self.assertEqual(a.user, self.user)
# Try to create again
other_a = Album.get_or_create_default_for_user(username='******')
self.assertEqual(other_a, a)
import os
parentdir = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
os.sys.path.insert(0,parentdir)
os.environ.setdefault("DJANGO_SETTINGS_MODULE", "frebapps.settings")
from geosnapr.models import User, Album, Image
import django
if __name__ == '__main__':
django.setup()
# Get all users
users = User.objects.all()
for user in users:
# Create the default album
album = Album.get_or_create_default_for_user(username=user.username)
# Associate all images with the default album
images = Image.objects.filter(user=user)
for image in images:
if image not in album.images.all():
album.images.add(image)
album.save()
# Create the public album
public = Album.get_or_create_public_for_user(username=user.username)
def test_get_no_album(self):
a = Album.get_or_create_default_for_user(username='******')
self.assertEqual(a.name, settings.DEFAULT_ALBUM_NAME)
self.assertEqual(a.user, self.user)
def test_get_bad_user(self):
a = Album.get_or_create_default_for_user(username='******')
self.assertEqual(a, None)
