from tests.base import ApiDBTestCase


class TasksCsvExportTestCase(ApiDBTestCase):

    def setUp(self):
        super(TasksCsvExportTestCase, self).setUp()

        self.generate_fixture_project_status()
        self.generate_fixture_project()
        self.generate_fixture_asset_type()
        self.generate_fixture_department()
        self.generate_fixture_task_type()

    def test_get_output_files(self):
        csv_task_types = self.get_raw("export/csv/task-types.csv")
        expected_result = """Department;Name\r
Animation;Animation\r
Modeling;Shaders\r
"""
        self.assertEqual(csv_task_types, expected_result)

from tests.base import ApiDBTestCase


class TasksCsvExportTestCase(ApiDBTestCase):

    def setUp(self):
        super(TasksCsvExportTestCase, self).setUp()

        self.generate_fixture_project_status()
        self.generate_fixture_project()
        self.generate_fixture_asset_type()
        self.generate_fixture_department()
        self.generate_fixture_task_type()

    def test_get_output_files(self):
        csv_task_types = self.get_raw("export/csv/task-types.csv")
        expected_result = """Department;Name\r
Animation;Animation\r
Animation;Layout\r
Modeling;Shaders\r
"""
        self.assertEqual(csv_task_types, expected_result)

Fix task type export test
Fix task type export test

  from tests.base import ApiDBTestCase
  
  
  class TasksCsvExportTestCase(ApiDBTestCase):
  
      def setUp(self):
          super(TasksCsvExportTestCase, self).setUp()
  
          self.generate_fixture_project_status()
          self.generate_fixture_project()
          self.generate_fixture_asset_type()
          self.generate_fixture_department()
          self.generate_fixture_task_type()
  
      def test_get_output_files(self):
          csv_task_types = self.get_raw("export/csv/task-types.csv")
          expected_result = """Department;Name\r
  Animation;Animation\r
+ Animation;Layout\r
  Modeling;Shaders\r
  """
          self.assertEqual(csv_task_types, expected_result)
  
Fix task type export test
## Code Before:
from tests.base import ApiDBTestCase


class TasksCsvExportTestCase(ApiDBTestCase):

    def setUp(self):
        super(TasksCsvExportTestCase, self).setUp()

        self.generate_fixture_project_status()
        self.generate_fixture_project()
        self.generate_fixture_asset_type()
        self.generate_fixture_department()
        self.generate_fixture_task_type()

    def test_get_output_files(self):
        csv_task_types = self.get_raw("export/csv/task-types.csv")
        expected_result = """Department;Name\r
Animation;Animation\r
Modeling;Shaders\r
"""
        self.assertEqual(csv_task_types, expected_result)

## Instruction:
Fix task type export test
## Code After:
from tests.base import ApiDBTestCase


class TasksCsvExportTestCase(ApiDBTestCase):

    def setUp(self):
        super(TasksCsvExportTestCase, self).setUp()

        self.generate_fixture_project_status()
        self.generate_fixture_project()
        self.generate_fixture_asset_type()
        self.generate_fixture_department()
        self.generate_fixture_task_type()

    def test_get_output_files(self):
        csv_task_types = self.get_raw("export/csv/task-types.csv")
        expected_result = """Department;Name\r
Animation;Animation\r
Animation;Layout\r
Modeling;Shaders\r
"""
        self.assertEqual(csv_task_types, expected_result)
