def get_queryset(self):
if settings.LIMIT_EDIT_TO_CURRENT_COMMUNITY:
community = site_mappers.get_current_mapper().map_area
if PlotIdentifier.get_notebook_plot_lists():
return super(SubjectConsentManager, self).get_queryset(
).filter(
community=community,
household_member__household_structure__household__plot__plot_identifier__in
=PlotIdentifier.get_notebook_plot_lists())
else:
return super(SubjectConsentManager,
self).get_queryset().filter(community=community)
return super(SubjectConsentManager, self).get_queryset()
def test_create_household_given_plot_identifier(self):
original_save = Plot.save
Plot.save = Model.save
plot_identifier = PlotIdentifier('10', 'default').get_identifier()
plot = Plot.objects.create(plot_identifier=plot_identifier)
Household.objects.create(plot=plot)
Plot.save = original_save
def test_plot_identifier_class(self):
plot_identifier = PlotIdentifier('10', 'default').get_identifier()
self.assertEqual(plot_identifier, '100001-00')
self.assertEqual(PlotIdentifierHistory.objects.all().count(), 1)
plot_identifier_history = PlotIdentifierHistory.objects.get(
identifier='100001-00')
self.assertEqual(plot_identifier_history.created.date(), date.today())
self.assertEqual(plot_identifier_history.modified.date(), date.today())