[tests] fixes attachments tests for attribute rename

As #435

2 files changed, 3 insertions, 3 deletions

diff --git a/service/test/integration/test_retrieve_attachment.py b/service/test/integration/test_retrieve_attachment.py
index fd6b2511..1cf8006f 100644
--- a/service/test/integration/test_retrieve_attachment.py
+++ b/service/test/integration/test_retrieve_attachment.py
@@ -28,7 +28,7 @@ class RetrieveAttachmentTest(SoledadTestBase):
                            'type': 'cnt',
                            'raw': 'cGVxdWVubyBhbmV4byA6RAo=',
                            'phash': ident,
-                           'content-type': 'text/plain; charset=US-ASCII; name="attachment_pequeno.txt"'}
+                           'content_type': 'text/plain; charset=US-ASCII; name="attachment_pequeno.txt"'}
 
         yield self.add_document_to_soledad(attachment_dict)
 
diff --git a/service/test/unit/adapter/test_soledad_querier.py b/service/test/unit/adapter/test_soledad_querier.py
index 8321ed72..03d62d58 100644
--- a/service/test/unit/adapter/test_soledad_querier.py
+++ b/service/test/unit/adapter/test_soledad_querier.py
@@ -96,7 +96,7 @@ class SoledadQuerierTest(unittest.TestCase):
     def test_attachment_base64(self):
         soledad = mock()
         bdoc = mock()
-        bdoc.content = {'raw': base64.encodestring('esse papo seu ta qualquer coisa'), 'content-type': 'text/plain'}
+        bdoc.content = {'raw': base64.encodestring('esse papo seu ta qualquer coisa'), 'content_type': 'text/plain'}
         when(soledad).get_from_index('by-type-and-payloadhash', 'cnt', any(unicode)).thenReturn([bdoc])
         querier = SoledadQuerier(soledad)
 
@@ -108,7 +108,7 @@ class SoledadQuerierTest(unittest.TestCase):
     def test_attachment_quoted_printable(self):
         soledad = mock()
         bdoc = mock()
-        bdoc.content = {'raw': quopri.encodestring('esse papo seu ta qualquer coisa'), 'content-type': 'text/plain'}
+        bdoc.content = {'raw': quopri.encodestring('esse papo seu ta qualquer coisa'), 'content_type': 'text/plain'}
         when(soledad).get_from_index('by-type-and-payloadhash', 'cnt', any(unicode)).thenReturn([bdoc])
         querier = SoledadQuerier(soledad)