diff options
Diffstat (limited to 'service/test/integration/retrieve_attachment_test.py')
| -rw-r--r-- | service/test/integration/retrieve_attachment_test.py | 11 |
1 files changed, 5 insertions, 6 deletions
diff --git a/service/test/integration/retrieve_attachment_test.py b/service/test/integration/retrieve_attachment_test.py index 37bc3ca2..5f754a27 100644 --- a/service/test/integration/retrieve_attachment_test.py +++ b/service/test/integration/retrieve_attachment_test.py @@ -13,18 +13,17 @@ # # You should have received a copy of the GNU Affero General Public License # along with Pixelated. If not, see <http://www.gnu.org/licenses/>. -import unittest -from test.support.integration_helper import SoledadTestBase +from test.support.integration.soledad_test_base import SoledadTestBase -class RetrieveAttachmentTest(unittest.TestCase, SoledadTestBase): +class RetrieveAttachmentTest(SoledadTestBase): def setUp(self): - self.setup_soledad() + SoledadTestBase.setUp(self) def tearDown(self): - self.teardown_soledad() + SoledadTestBase.tearDown(self) def test_attachment_content_is_retrieved(self): ident = 'F4E99C1CEC4D300A4223A96CCABBE0304BDBC31C550A5A03E207A5E4C3C71A22' @@ -35,7 +34,7 @@ class RetrieveAttachmentTest(unittest.TestCase, SoledadTestBase): 'phash': ident, 'content-type': 'text/plain; charset=US-ASCII; name="attachment_pequeno.txt"'} - self.add_document_to_soledad(attachment_dict) + self.client.add_document_to_soledad(attachment_dict) attachment = self.get_attachment(ident, 'base64') |