class FormattedTextGeneratorTests(TestCase):
    def setUp(self):
        self.formattedTextGenerator = FormattedTextGenerator()

    def tearDown(self):
        self.formattedTextGenerator = None

    def testShouldFormatWarehouseReportTitle(self):
        self.assertEqual("<h3>Warehouse- Agra</h3>",
                self.formattedTextGenerator.generateWarehouseReportTitle(TestNames.WAREHOUSE_AGRA))
        self.assertEqual("<h3>Warehouse- Bangalore</h3>",
                self.formattedTextGenerator.generateWarehouseReportTitle(TestNames.WAREHOUSE_BANGALORE))

    def testShouldFormatSkuReportTitle(self):
        self.assertEqual("<h3>SKU- Apple iPad</h3>",
                self.formattedTextGenerator.generateSkuReportTitle(TestNames.SKU_APPLE))
        self.assertEqual("<h3>SKU- Bose Portable Dock II</h3>",
                self.formattedTextGenerator.generateSkuReportTitle(TestNames.SKU_BOSE))

    def testShouldFormatSkuWarehouseReportTitle(self):
        self.assertEqual("<h3>SKU- Apple iPad, Warehouse- Agra</h3>",
                self.formattedTextGenerator.generateSkuWarehouseReportTitle(
                        TestNames.SKU_APPLE, TestNames.WAREHOUSE_AGRA))
        self.assertEqual("<h3>SKU- Bose Portable Dock II, Warehouse- Bangalore</h3>",
                self.formattedTextGenerator.generateSkuWarehouseReportTitle(
                        TestNames.SKU_BOSE, TestNames.WAREHOUSE_BANGALORE))
    def testShouldFormatErrorenousReportTitle(self):
        self.assertEqual("""<h3 class="text-error">Error generating your report!</h3>""",
                self.formattedTextGenerator.generateErroneousReportTitle())

    def testShouldFormatReportLine(self):
        self.assertEqual("<strong>Quantity-</strong> 100",
                self.formattedTextGenerator.generateReportLine("Quantity", "100"))
        self.assertEquals("<strong>Value-</strong> INR 1220",
                self.formattedTextGenerator.generateReportLine("Value", "INR 1220"))
 def setUp(self):
     self.formattedTextGenerator = FormattedTextGenerator()