covering line for datum

ZacBlanco · Jul 1, 2016 · 0a8edc8 · 0a8edc8
1 parent ae1c5a4
commit 0a8edc8
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/tests/test_generator.py b/tests/test_generator.py
@@ -175,8 +175,15 @@ def test_map_datum_bad(self, mock1):
 		except ValueError as e:
 			print str(e)
 			assert 'Expected mapFromField key to be a dict object' in str(e)
-
-
+		finally:
+			missing_fields['mapFromField'] = 'field3'
+
+		try:
+			md = MapDatum(missing_fields)
+			data = {}
+			md.generate(random, data)
+		except ValueError as e:
+			assert 'Could not get key: ' in str(e)