Extract key only if event body is a dict of a list (#494)

[ML-5442](https://jira.iguazeng.com/browse/ML-5442)

storey/sources.py

@@ -142,7 +142,7 @@ def _build_event(self, element, key):
         else:
             body = element
 
-        if hasattr(body, "__getitem__") and not key and self._key_field:
+        if isinstance(body, (dict, list)) and not key and self._key_field:
             if isinstance(self._key_field, str) or isinstance(self._key_field, int):
                 key = body[self._key_field]
             else:

tests/test_flow.py

@@ -3425,8 +3425,8 @@ def test_key_field_and_non_dict_event_body():
         ]
     ).run()
 
-    body1 = 1
-    body2 = 2
+    body1 = b"a"
+    body2 = b"b"
 
     controller.emit(body1)
     controller.emit(Event(body2, "my_key"))