def test_backfill_daily_with_site_domain(self): """Test that proper site id gets passed to task func when passing integer.""" SiteFactory.reset_sequence(0) site2 = SiteFactory() # site-0.example.com with mock.patch(self.PLAIN_PATH) as mock_populate: call_command('backfill_figures_daily_metrics', site="site-0.example.com", no_delay=True) assert mock_populate.called_with(site_id=2)
def test_get_side_ids(self, site, expected_result): """Test that get_site_ids will return the correct Site id if passed that id or the corresponding Site's domain, will return all Site ids if none passed, and will return an empty list if a non-existing Site id or domain passed """ # Site table will have an existing example.com Site as well as whatever we # create w/ SiteFactory, due to data migration example_site = Site.objects.get(domain='example.com') SiteFactory.reset_sequence(0) site1 = SiteFactory() # site-0.example.com site2 = SiteFactory() # site-1.example.com with mock.patch('figures.management.base.get_sites') as mock_get_sites: mock_get_sites.return_value = [example_site, site1, site2] site_ids = BaseBackfillCommand().get_site_ids(site) assert site_ids == expected_result