[LIB-837] Correct after tests;

opalczynski · opalczynski · commit e108b5a7806f · 2016-08-12T15:24:42.000+02:00
diff --git a/syncano/models/incentives.py b/syncano/models/incentives.py
@@ -222,7 +222,7 @@ class ScriptEndpoint(Model):
     """
 
     name = fields.SlugField(max_length=50, primary_key=True)
-    script = fields.ModelField('Script', label='script id')
+    script = fields.ModelField('Script', just_pk=True)
     public = fields.BooleanField(required=False, default=False)
     public_link = fields.ChoiceField(required=False, read_only=True)
     links = fields.LinksField()
diff --git a/tests/integration_test_custom_socket.py b/tests/integration_test_custom_socket.py
@@ -80,7 +80,7 @@ def test_custom_socket_update(self):
 
     def assert_custom_socket(self, suffix, dependency_method):
         custom_socket = self._create_custom_socket(suffix, dependency_method=dependency_method)
-        self.assertTrue(custom_socket.id)
+        self.assertTrue(custom_socket.name)
 
     @classmethod
     def _create_custom_socket(cls, suffix, dependency_method):
@@ -92,10 +92,6 @@ def _create_custom_socket(cls, suffix, dependency_method):
         custom_socket.publish()
         return custom_socket
 
-    @classmethod
-    def _initialize_socket(cls, suffix):
-        return CustomSocket(name='my_custom_socket_{}'.format(suffix))
-
     @classmethod
     def _define_endpoints(cls, suffix, custom_socket):
         endpoint = Endpoint(name='my_endpoint_{}'.format(suffix))
