hacks to work around new pylint false positives

1 files changed, 15 insertions, 9 deletions

diff --git a/python/tests/transform_elasticsearch.py b/python/tests/transform_elasticsearch.py
index b9011892..c11c5972 100644
--- a/python/tests/transform_elasticsearch.py
+++ b/python/tests/transform_elasticsearch.py
@@ -204,15 +204,21 @@ def test_elasticsearch_container_transform(journal_metadata_importer):
 
 
 def test_elasticsearch_file_transform():
-    f = entity_from_json(open('./tests/files/file_bcah4zp5tvdhjl5bqci2c2lgfa.json', 'r').read(), FileEntity)
-
-    f.state = 'active'
-    es = file_to_elasticsearch(f)
-    assert es['sha1'] == f.sha1
-    assert es['sha256'] == f.sha256
-    assert es['md5'] == f.md5
-    assert es['size_bytes'] == f.size
-    assert es['mimetype'] == f.mimetype
+
+    with open('./tests/files/file_bcah4zp5tvdhjl5bqci2c2lgfa.json', 'r') as f:
+        json_str = f.read()
+
+    fe = entity_from_json(json_str, FileEntity)
+
+    fe.state = 'active'
+    es = file_to_elasticsearch(fe)
+
+    # pylint infers type of 'fe' incorrectly for some reason (as str/bytes)
+    assert es['sha1'] == fe.sha1  # pylint: disable=no-member
+    assert es['sha256'] == fe.sha256  # pylint: disable=no-member
+    assert es['md5'] == fe.md5  # pylint: disable=no-member
+    assert es['size_bytes'] == fe.size  # pylint: disable=no-member
+    assert es['mimetype'] == fe.mimetype  # pylint: disable=no-member
     assert es['in_ia'] == True
 
     assert 'web' in es['rels']