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()