[HOT-FIX] files hot fix: create tests;

opalczynski · opalczynski · commit c304f4b6f4f5 · 2016-09-27T11:58:00.000+02:00
diff --git a/tests/integration_test_data_objects.py b/tests/integration_test_data_objects.py
@@ -1,6 +1,12 @@
 # -*- coding: utf-8 -*-
 from hashlib import md5
-from StringIO import StringIO
+
+try:
+    # python2
+    from StringIO import StringIO
+except ImportError:
+    # python3
+    from io import StringIO
 
 import requests
 from syncano.models import Object
@@ -101,6 +107,7 @@ def test_manager_create(self):
         create_string = 'manager create'
         with open(self.file_path, 'rb') as f:
             data_object = Object.please.create(
+                class_name=self.class_name,
                 test_field_a=create_string,
                 test_field_file=f
             )
@@ -120,7 +127,8 @@ def assert_file_md5(self, data_object):
     def _create_object_with_file(self):
         with open('tests/test_files/python-logo.png', 'rb') as f:
             object = Object.please.create(
+                class_name=self.class_name,
                 test_field_a=self.initial_field_a,
-                test_field_file=f
+                test_field_file=f,
             )
         return object
