mirror of
https://github.com/django/django.git
synced 2025-10-24 06:06:09 +00:00
Refs #26022 -- Used context manager version of assertRaises in tests.
This commit is contained in:
@@ -156,11 +156,13 @@ class LayerMapTest(TestCase):
|
||||
|
||||
# Testing invalid params for the `unique` keyword.
|
||||
for e, arg in ((TypeError, 5.0), (ValueError, 'foobar'), (ValueError, ('name', 'mpolygon'))):
|
||||
self.assertRaises(e, LayerMapping, County, co_shp, co_mapping, transform=False, unique=arg)
|
||||
with self.assertRaises(e):
|
||||
LayerMapping(County, co_shp, co_mapping, transform=False, unique=arg)
|
||||
|
||||
# No source reference system defined in the shapefile, should raise an error.
|
||||
if connection.features.supports_transform:
|
||||
self.assertRaises(LayerMapError, LayerMapping, County, co_shp, co_mapping)
|
||||
with self.assertRaises(LayerMapError):
|
||||
LayerMapping(County, co_shp, co_mapping)
|
||||
|
||||
# Passing in invalid ForeignKey mapping parameters -- must be a dictionary
|
||||
# mapping for the model the ForeignKey points to.
|
||||
@@ -168,14 +170,17 @@ class LayerMapTest(TestCase):
|
||||
bad_fk_map1['state'] = 'name'
|
||||
bad_fk_map2 = copy(co_mapping)
|
||||
bad_fk_map2['state'] = {'nombre': 'State'}
|
||||
self.assertRaises(TypeError, LayerMapping, County, co_shp, bad_fk_map1, transform=False)
|
||||
self.assertRaises(LayerMapError, LayerMapping, County, co_shp, bad_fk_map2, transform=False)
|
||||
with self.assertRaises(TypeError):
|
||||
LayerMapping(County, co_shp, bad_fk_map1, transform=False)
|
||||
with self.assertRaises(LayerMapError):
|
||||
LayerMapping(County, co_shp, bad_fk_map2, transform=False)
|
||||
|
||||
# There exist no State models for the ForeignKey mapping to work -- should raise
|
||||
# a MissingForeignKey exception (this error would be ignored if the `strict`
|
||||
# keyword is not set).
|
||||
lm = LayerMapping(County, co_shp, co_mapping, transform=False, unique='name')
|
||||
self.assertRaises(MissingForeignKey, lm.save, silent=True, strict=True)
|
||||
with self.assertRaises(MissingForeignKey):
|
||||
lm.save(silent=True, strict=True)
|
||||
|
||||
# Now creating the state models so the ForeignKey mapping may work.
|
||||
State.objects.bulk_create([
|
||||
@@ -222,11 +227,13 @@ class LayerMapTest(TestCase):
|
||||
# Bad feature id ranges should raise a type error.
|
||||
bad_ranges = (5.0, 'foo', co_shp)
|
||||
for bad in bad_ranges:
|
||||
self.assertRaises(TypeError, lm.save, fid_range=bad)
|
||||
with self.assertRaises(TypeError):
|
||||
lm.save(fid_range=bad)
|
||||
|
||||
# Step keyword should not be allowed w/`fid_range`.
|
||||
fr = (3, 5) # layer[3:5]
|
||||
self.assertRaises(LayerMapError, lm.save, fid_range=fr, step=10)
|
||||
with self.assertRaises(LayerMapError):
|
||||
lm.save(fid_range=fr, step=10)
|
||||
lm.save(fid_range=fr)
|
||||
|
||||
# Features IDs 3 & 4 are for Galveston County, Texas -- only
|
||||
|
||||
Reference in New Issue
Block a user