def test_unrestricted_migration_of_free_apps_exclusions(self):
self.webapp.addonexcludedregion.create(region=mkt.regions.US.id)
eq_(self.webapp.geodata.reload().restricted, False)
migrate_geodata.Command().handle()
eq_(self.webapp.reload().addonexcludedregion.count(), 0)
eq_(self.webapp.geodata.reload().restricted, False)
def test_restricted_no_migration_of_paid_apps_exclusions(self):
self.make_premium(self.webapp)
self.webapp.addonexcludedregion.create(region=mkt.regions.US.id)
eq_(self.webapp.geodata.reload().restricted, False)
migrate_geodata.Command().handle()
eq_(self.webapp.reload().addonexcludedregion.count(), 1)
eq_(self.webapp.geodata.reload().restricted, True)
def test_no_migration_of_unrated_games(self):
amo.tests.make_game(self.webapp, rated=False)
regions = (mkt.regions.BR.id, mkt.regions.DE.id)
for region in regions:
self.webapp.addonexcludedregion.create(region=region)
migrate_geodata.Command().handle()
self.assertSetEqual(self.webapp.reload().addonexcludedregion
.values_list('region', flat=True),
regions)
def test_migration_of_regional_content(self):
# Exclude in every where except Brazil.
regions = list(mkt.regions.REGIONS_CHOICES_ID_DICT)
regions.remove(mkt.regions.BR.id)
for region in regions:
self.webapp.addonexcludedregion.create(region=region)
eq_(self.webapp.geodata.reload().popular_region, None)
migrate_geodata.Command().handle()
eq_(self.webapp.reload().addonexcludedregion.count(), 0)
eq_(self.webapp.geodata.reload().popular_region, mkt.regions.BR.slug)
def test_migration_of_regional_content(self):
# Exclude in everywhere except Brazil.
regions = list(mkt.regions.REGIONS_CHOICES_ID_DICT)
regions.remove(mkt.regions.BR.id)
for region in regions:
self.webapp.addonexcludedregion.create(region=region)
eq_(self.webapp.geodata.reload().popular_region, None)
migrate_geodata.Command().handle()
self.assertSetEqual(self.webapp.reload().addonexcludedregion
.values_list('region', flat=True),
[mkt.regions.CN.id])
eq_(self.webapp.geodata.reload().popular_region, mkt.regions.BR.slug)
def test_migration_of_rated_games(self):
# This adds a ContentRating for only Brazil, not Germany.
amo.tests.make_game(self.webapp, rated=True)
self.webapp.content_ratings.filter(
ratings_body=mkt.regions.DE.ratingsbody.id).delete()
regions = (mkt.regions.BR.id, mkt.regions.DE.id)
for region in regions:
self.webapp.addonexcludedregion.create(region=region)
migrate_geodata.Command().handle()
self.assertSetEqual(self.webapp.reload().addonexcludedregion
.values_list('region', flat=True),
[mkt.regions.DE.id])
