def test_save_new_resource(self):
"""Test saving a new Resource."""
with File(get_in_memory_image_file()) as tempfile:
user = self.create_user()
resource = Resource(name='something cool',
attachment=tempfile,
created_by=user)
resource.save()
self.assertEqual(resource.slug, 'something-cool')
def test_save_new_resource(self):
"""Test saving a new Resource."""
with File(get_in_memory_image_file()) as tempfile:
user = self.create_user()
resource = Resource(
name='something cool',
attachment=tempfile,
created_by=user
)
resource.save()
self.assertEqual(resource.slug, 'something-cool')
def test_delete(self):
"""Test deleting a resource."""
user = self.create_user()
with File(get_in_memory_image_file()) as tempfile:
resource = Resource(
name='wasabi peas',
attachment=tempfile,
created_by=user
)
resource.save()
self.assertEqual(resource.status, 'active')
resource.delete()
self.assertEqual(resource.status, 'deleted')
def create_resource(self, **kwargs):
"""Creates a resource for testing."""
name = kwargs.get('name', 'test resource')
if 'user' not in kwargs:
user = self.create_user()
else:
user = kwargs['user']
if 'groups' not in kwargs:
groups = [mommy.make('groups.Group')]
groups[0].owners.add(user)
else:
groups = kwargs['groups']
with File(get_in_memory_image_file()) as tempfile:
resource = Resource(name=name,
attachment=tempfile,
created_by=user)
resource.save()
resource.groups.add(*groups)
return resource
def create_resource(self, **kwargs):
"""Creates a resource for testing."""
name = kwargs.get('name', 'test resource')
if 'user' not in kwargs:
user = self.create_user()
else:
user = kwargs['user']
if 'groups' not in kwargs:
groups = [mommy.make('groups.Group')]
groups[0].owners.add(user)
else:
groups = kwargs['groups']
with File(get_in_memory_image_file()) as tempfile:
resource = Resource(
name=name,
attachment=tempfile,
created_by=user
)
resource.save()
resource.groups.add(*groups)
return resource
def test_save_resource_with_same_name_as_another_resource(self):
"""Should still generate a unique slug."""
user = self.create_user()
with File(get_in_memory_image_file()) as tempfile:
resource = Resource(name='wasabi peas',
attachment=tempfile,
created_by=user)
resource.save()
with File(get_in_memory_image_file()) as tempfile:
new_resource = Resource(name='wasabi peas',
attachment=tempfile,
created_by=user)
new_resource.save()
self.assertTrue(new_resource.slug.startswith('wasabi-peas-'))
self.assertNotEqual(new_resource.slug, resource.slug)
def test_delete(self):
"""Test deleting a resource."""
user = self.create_user()
with File(get_in_memory_image_file()) as tempfile:
resource = Resource(name='wasabi peas',
attachment=tempfile,
created_by=user)
resource.save()
self.assertEqual(resource.status, 'active')
resource.delete()
self.assertEqual(resource.status, 'deleted')
def test_save_resource_with_same_name_as_another_resource(self):
"""Should still generate a unique slug."""
user = self.create_user()
with File(get_in_memory_image_file()) as tempfile:
resource = Resource(
name='wasabi peas',
attachment=tempfile,
created_by=user
)
resource.save()
with File(get_in_memory_image_file()) as tempfile:
new_resource = Resource(
name='wasabi peas',
attachment=tempfile,
created_by=user
)
new_resource.save()
self.assertTrue(new_resource.slug.startswith('wasabi-peas-'))
self.assertNotEqual(new_resource.slug, resource.slug)
