From dc5d55cedbb5191aa672a20a78763dbeb7718dda Mon Sep 17 00:00:00 2001
From: Stefano Borini <sborini@enthought.com>
Date: Thu, 17 May 2018 17:02:55 +0100
Subject: [PATCH] Ported data source probe to new API
---
force_bdss/tests/probe_classes/data_source.py | 15 +++++++++++----
1 file changed, 11 insertions(+), 4 deletions(-)
diff --git a/force_bdss/tests/probe_classes/data_source.py b/force_bdss/tests/probe_classes/data_source.py
index f50f81a..9a96f60 100644
--- a/force_bdss/tests/probe_classes/data_source.py
+++ b/force_bdss/tests/probe_classes/data_source.py
@@ -49,10 +49,17 @@ class ProbeDataSourceModel(BaseDataSourceModel):
class ProbeDataSourceFactory(BaseDataSourceFactory,
ProbeEvaluatorFactory):
- id = Str(factory_id("enthought", "test_ds"))
- name = Str('test_data_source')
+ def get_identifier(self):
+ return "test_ds"
- model_class = Type(ProbeDataSourceModel)
+ def get_name(self):
+ return "test_data_source"
+
+ def get_model_class(self):
+ return ProbeDataSourceModel
+
+ def get_data_source_class(self):
+ return ProbeDataSource
def create_model(self, model_data=None):
if model_data is None:
@@ -67,7 +74,7 @@ class ProbeDataSourceFactory(BaseDataSourceFactory,
)
def create_data_source(self):
- return ProbeDataSource(
+ return self.data_source_class(
factory=self,
run_function=self.run_function,
)
--
GitLab