def test_create_scenario(self): s1 = Scenario(user=self.user, name="My Scenario", input_target_boardfeet=2000, input_target_carbon=1, input_property=self.prop1, input_rxs={self.stand1.pk: self.rx1, self.stand2.pk: self.rx2}, ) s1.save() self.assertEquals(Scenario.objects.get(name="My Scenario").input_target_boardfeet, 2000.0)
def test_scenario_results(self): s1 = Scenario(user=self.user, name="My Scenario", input_target_boardfeet=2000, input_target_carbon=1, input_property=self.prop1, input_rxs={self.stand1.pk: self.rx1, self.stand2.pk: self.rx2}, ) s1.save() out = s1.output_property_metrics self.assertTrue(out.has_key("__all__"))
def test_json_results(self): s1 = Scenario(user=self.user, name="My Scenario", input_target_boardfeet=2000, input_target_carbon=1, input_property=self.prop1, input_rxs={self.stand1.pk: self.rx1, self.stand2.pk: self.rx2}, ) s1.save() geojson_link = ForestProperty.get_options().get_link('Property Scenarios') url = geojson_link.reverse(self.prop1) # not logged in yet response = self.client.get(url) self.assertEqual(response.status_code, 401, response.content) # now we log in self.client.login(username='******', password='******') response = self.client.get(url) self.assertEqual(response.status_code, 200, response.content)
def cleanup(): try: owner = User.objects.get(username='******') except DoesNotExist: owner = False try: owner2 = User.objects.get(username='******') except DoesNotExist: owner2 = False try: manager = User.objects.get(username='******') except DoesNotExist: manager = False ownerScenarios = Scenario.objects.filter(user=owner) for scenario in ownerScenarios: for prop in scenario.forestproperty_set.all(): prop.shared_scenario = None prop.save() Scenario.delete(scenario) owner2Scenarios = Scenario.objects.filter(user=owner2) for scenario in owner2Scenarios: for prop in scenario.forestproperty_set.all(): prop.shared_scenario = None prop.save() Scenario.delete(scenario) ownerProperties = ForestProperty.objects.filter(user=owner) for fproperty in ownerProperties: ForestProperty.delete(fproperty) owner2Properties = ForestProperty.objects.filter(user=owner2) for fproperty in owner2Properties: ForestProperty.delete(fproperty) for cgroup in CarbonGroup.objects.filter(user=manager): CarbonGroup.delete(cgroup) if owner: User.delete(owner) if owner2: User.delete(owner2) if manager: User.delete(manager)
def setUp(self): self.client = Client() import_rasters() self.user = User.objects.create_user("featuretest", "*****@*****.**", password="******") self.stand1 = Stand(user=self.user, name="My Stand 1", geometry_orig=g1) self.stand1.save() self.stand2 = Stand(user=self.user, name="My Stand 2", geometry_orig=g1) self.stand2.save() self.stand3 = Stand(user=self.user, name="My Stand 3 (not on property)", geometry_orig=g1) self.stand3.save() self.prop1 = ForestProperty(user=self.user, name="My Property", geometry_final=p1) self.prop1.save() self.prop1.add(self.stand1) self.prop1.add(self.stand2) self.options = Scenario.get_options() self.create_url = self.options.get_create_form() self.rx1 = Rx.objects.get(internal_name=self.prop1.variant.code + "1").id self.rx2 = Rx.objects.get(internal_name=self.prop1.variant.code + "2").id enable_sharing()
def setUp(self): self.client = Client() import_rasters() self.user = User.objects.create_user( 'featuretest', '*****@*****.**', password='******') self.stand1 = Stand(user=self.user, name="My Stand 1", geometry_orig=g1) self.stand1.save() self.stand2 = Stand(user=self.user, name="My Stand 2", geometry_orig=g1) self.stand2.save() self.stand3 = Stand(user=self.user, name="My Stand 3 (not on property)", geometry_orig=g1) self.stand3.save() self.prop1 = ForestProperty(user=self.user, name="My Property", geometry_final=p1) self.prop1.save() self.prop1.add(self.stand1) self.prop1.add(self.stand2) self.options = Scenario.get_options() self.create_url = self.options.get_create_form() self.rx1 = Rx.objects.get(internal_name=self.prop1.variant.code + "1").id self.rx2 = Rx.objects.get(internal_name=self.prop1.variant.code + "2").id enable_sharing()