def test_external_water(self): # If external water is sea, and the scenariobreach has # waterlevels, it should add a link to a externalwater graph # image. The string is HTML, so it should be a SafeString. The # link depends on the scenariobreach id. scenario = ScenarioF.create() externalwater = ExternalWaterF.create( type=models.ExternalWater.TYPE_SEA) breach = BreachF.create(externalwater=externalwater) waterlevelset = WaterlevelSetF.create() WaterlevelF.create(waterlevelset=waterlevelset) scenariobreach = ScenarioBreachF.create( scenario=scenario, breach=breach, waterlevelset=waterlevelset) fields = extra_infowindow_information_fields( _('External Water'), scenario) self.assertEquals(len(fields), 1) value = fields[0].value_str self.assertTrue(isinstance(value, SafeString)) self.assertTrue('get_externalwater_graph_infowindow' in value) self.assertTrue(str(scenariobreach.id) in value)
def test_scenario(self): # Should add scenario ID scenario = ScenarioF.create() fields = extra_infowindow_information_fields( 'Scenario', scenario) self.assertEquals(len(fields), 1) self.assertEquals(fields[0].value_str, str(scenario.id))
def test_unknown_header(self): self.assertEquals( len(extra_infowindow_information_fields('unknown_header', None)), 0)