def setUp(self):
self.block_library = BlockLibrary()
class BlockLibraryTest(TestCase):
def setUp(self):
self.block_library = BlockLibrary()
def tearDown(self):
self.block_library.remove_all_blocks()
def test_add_block(self):
"""
Tests that BlockLibrary adds block
"""
self.block_library.add_block(foo_block)
self.assertEqual(len(self.block_library.get_blocks()), 1, 'Number of blocks in library shoud be 1')
self.assertIn('blocks.blocks.foo_block', self.block_library.get_blocks(), 'blocks.blocks.foo_block key should be in the library dictionary')
def test_get_block(self):
"""
Tests that BlockLibrary returns selected block
"""
self.block_library.add_block(foo_block)
self.assertEqual(len(self.block_library.get_blocks()), 1, 'Number of blocks in library shoud be 1')
self.assertIn('blocks.blocks.foo_block', self.block_library.get_blocks(), 'blocks.blocks.foo_block key should be in the library dictionary')
self.assertIn('__call__', dir(self.block_library.get_block('blocks.blocks.foo_block')), 'BlockLibrary.get_block should return a function')
self.assertIs(self.block_library.get_block('blocks.blocks.foo_block'), foo_block, 'BlockLibrary.get_block should return foo_block')
def test_get_blocks(self):
"""
Tests that BlockLibrary returns the library
"""
self.block_library.add_block(foo_block)
self.block_library.add_block(another_foo_block)
the_library = self.block_library.get_blocks()
self.assertEqual(len(the_library), 2, 'the_library length shoud be 2')
self.assertIn('blocks.blocks.foo_block', the_library, 'blocks.blocks.foo should be in the_library')
self.assertIn('blocks.blocks.another_foo_block', the_library, 'blocks.blocks.another_foo_block should be in the_library')
def test_get_block_not_in_library(self):
"""
Tests that if given block is not in the library raises a NotInLibraryException
"""
self.assertRaisesRegexp(NotInLibraryException, 'foo.blocks.a_block is not in the library', self.block_library.get_block, 'foo.blocks.a_block')
def test_remove_all_blocks(self):
"""
Tests that BlockLibrary removes all blocks
"""
self.block_library.add_block(foo_block)
self.block_library.add_block(another_foo_block)
self.assertEqual(len(self.block_library.get_blocks()), 2, 'Number of blocks in the library shoud be 2')
self.assertIn('blocks.blocks.foo_block', self.block_library.get_blocks(), 'blocks.blocks.foo_block key should be in the library dictionary')
self.assertIn('blocks.blocks.another_foo_block', self.block_library.get_blocks(), 'blocks.blocks.another_foo_block key should be in the library dictionary')
self.block_library.remove_all_blocks()
self.assertEqual(len(self.block_library.get_blocks()), 0, 'Number of blocks in the library should be 0')
def test_remove_block(self):
"""
Tests that BlockLibrary removes a block
"""
self.block_library.add_block(foo_block)
self.block_library.add_block(another_foo_block)
self.assertEqual(len(self.block_library.get_blocks()), 2, 'Number of blocks in the library shoud be 2')
self.assertIn('blocks.blocks.foo_block', self.block_library.get_blocks(), 'blocks.blocks.foo_block key should be in the library dictionary')
self.assertIn('blocks.blocks.another_foo_block', self.block_library.get_blocks(), 'blocks.blocks.another_foo_block key should be in the library dictionary')
self.block_library.remove_block('blocks.blocks.another_foo_block')
self.assertNotIn('blocks.blocks.another_foo_block', self.block_library.get_blocks(), 'blocks.blocks.another_foo_block should not be in the library dictionary')
def test_remove_block_not_in_library(self):
"""
Tests that if the block wanted to remove is not in the library raises a NotInLibraryException
"""
self.assertRaisesRegexp(NotInLibraryException, 'foo.blocks.a_block is not in the library', self.block_library.remove_block, 'foo.blocks.a_block')
def test_autodiscover_blocks(self):
"""
Tests that BlockLibrary discovers all functions declared at blocks.py
"""
self.block_library.autodiscover()
self.assertIn('blocks.blocks.foo_block', self.block_library.get_blocks(), 'blocks.blocks.foo_block key should be in the library dictionary')
self.assertIn('blocks.blocks.another_foo_block', self.block_library.get_blocks(), 'blocks.blocks.another_foo_block key should be in the library dictionary')
self.assertIn('blocks.blocks.render_content_foo_block', self.block_library.get_blocks(), 'blocks.blocks.render_content_foo_block key should be in the library dictionary')
def test_empty_library(self):
"""
Tests BlockLibrary's is_empty method
"""
self.assertTrue(self.block_library.is_empty(), 'is_empty should return True with no blocks added')
self.block_library.add_block(foo_block)
self.assertFalse(self.block_library.is_empty(), 'is_empty should return False with a block in the library')
