def setUp(self):
self.old_cwd = os.getcwd()
os.chdir(TEST_DIR)
Module().create("Quux")
os.chdir("Quux")
self.block = Block()
class BlockTest(unittest.TestCase):
def setUp(self):
self.old_cwd = os.getcwd()
os.chdir(TEST_DIR)
Module().create("Quux")
os.chdir("Quux")
self.block = Block()
def tearDown(self):
os.chdir("..")
remove_module("Foo", "Quux")
os.chdir(self.old_cwd)
del self.block
def test_create(self):
self.block.create("Product")
with open(os.path.join("Block", "Product.php")) as block_file:
self.assertEqual(self._get_reference_class("Product"),
block_file.read())
with open(os.path.join("etc", "config.xml")) as config:
self.assertEqual(reference_config, config.read())
os.remove(os.path.join("Block", "Product.php"))
def test_create_works_in_depth(self):
os.chdir("Block")
self.block.create("Product")
os.chdir("..")
with open(os.path.join("Block", "Product.php")) as block_file:
self.assertEqual(self._get_reference_class("Product"),
block_file.read())
with open(os.path.join("etc", "config.xml")) as config:
self.assertEqual(reference_config, config.read())
os.remove(os.path.join("Block", "Product.php"))
def test_create_works_with_underscores(self):
self.block.create("Product_List")
with open(os.path.join("Block", "Product", "List.php")) as block_file:
self.assertEqual(self._get_reference_class("Product_List"),
block_file.read())
with open(os.path.join("etc", "config.xml")) as config:
self.assertEqual(reference_config, config.read())
os.remove(os.path.join("Block", "Product", "List.php"))
os.rmdir(os.path.join("Block", "Product"))
def test_overriding_works(self):
superclass = "Mage_Tag_Block_Product_List"
Block(superclass, True).create("List")
with open(os.path.join("Block", "List.php")) as block_file:
self.assertEqual(self._get_reference_class("List", superclass),
block_file.read())
with open(os.path.join("etc", "config.xml")) as config:
self.assertEqual(reference_override_config, config.read())
os.remove(os.path.join("Block", "List.php"))
def test_register(self):
self.block.register()
with open(os.path.join("etc", "config.xml")) as config:
self.assertEqual(reference_config, config.read())
def test_register_twice(self):
self.block.register()
self.block.register()
with open(os.path.join("etc", "config.xml")) as config:
self.assertEqual(reference_config, config.read())
def _get_reference_class(self, name, superclass="Mage_Core_Block_Template"):
return """<?php
class Foo_Quux_Block_%s extends %s
{
}
""" % (name, superclass)
