cleanup

Author: liu-rebecca

tableauserverclient/server/endpoint/__init__.py

@@ -10,6 +10,7 @@
 from .fileuploads_endpoint import Fileuploads
 from .flow_runs_endpoint import FlowRuns
 from .flows_endpoint import Flows
+from .flow_task_endpoint import FlowTasks
 from .groups_endpoint import Groups
 from .jobs_endpoint import Jobs
 from .metadata_endpoint import Metadata

tableauserverclient/server/endpoint/flow_task_endpoint.py

@@ -24,6 +24,6 @@ def create(self, flow_item: TaskItem) -> TaskItem:
             raise ValueError(error)
         logger.info("Creating an flow task %s", flow_item)
         url = self.baseurl
-        create_req = RequestFactory.Task.create_flow_task_req(flow_item)
+        create_req = RequestFactory.FlowTask.create_flow_task_req(flow_item)
         server_response = self.post_request(url, create_req)
         return server_response.content

tableauserverclient/server/request_factory.py

@@ -1290,6 +1290,7 @@ class RequestFactory(object):
     Favorite = FavoriteRequest()
     Fileupload = FileuploadRequest()
     Flow = FlowRequest()
+    FlowTask = FlowTaskRequest()
     Group = GroupRequest()
     Metric = MetricRequest()
     Permission = PermissionRequest()

test/test_flowtask.py

@@ -32,8 +32,6 @@ def setUp(self):
         self.server._site_id = "dad65087-b08b-4603-af4e-2887b8aafc67"
         self.server._auth_token = "j80k54ll2lfMZ0tv97mlPvvSCRyD0DOM"
 
-        # default task type is extractRefreshes TODO change this
-        # self.baseurl = "{}/{}".format(self.server.tasks.baseurl, "extractRefreshes")
         self.baseurl = self.server.flow_tasks.baseurl
 
     def test_create_flow_task(self):
@@ -48,7 +46,6 @@ def test_create_flow_task(self):
         target_item = TSC.Target("flow_id", "flow")
 
         task = TaskItem(schedule_item=monthly_schedule, target=target_item)
-        # task = TaskItem(None, "FullRefresh", None, schedule_item=monthly_schedule, target=target_item)
 
         with open(GET_XML_CREATE_FLOW_TASK_RESPONSE, "rb") as f:
             response_xml = f.read().decode("utf-8")
@@ -58,4 +55,3 @@ def test_create_flow_task(self):
 
         self.assertTrue("task_id" in create_response_content)
         self.assertTrue("flow_id" in create_response_content)
-        #self.assertTrue("FullRefresh" in create_response_content)