Python DataSource.get_by_id Examples

Example #1

File: test_data_sources.py Project: ryotarai/redash

    def test_pause_sets_reason(self):
        admin = self.factory.create_admin()
        rv = self.make_request(
            "post",
            "/api/data_sources/{}/pause".format(self.factory.data_source.id),
            user=admin,
            data={"reason": "testing"},
        )
        self.assertEqual(rv.status_code, 200)
        self.assertEqual(DataSource.get_by_id(self.factory.data_source.id).paused, True)
        self.assertEqual(DataSource.get_by_id(self.factory.data_source.id).pause_reason, "testing")

        rv = self.make_request(
            "post", "/api/data_sources/{}/pause?reason=test".format(self.factory.data_source.id), user=admin
        )
        self.assertEqual(DataSource.get_by_id(self.factory.data_source.id).pause_reason, "test")

Example #2

File: test_data_sources.py Project: ryotarai/redash

 def test_resumes_data_source(self):
     admin = self.factory.create_admin()
     self.factory.data_source.pause()
     self.factory.data_source.save()
     rv = self.make_request("delete", "/api/data_sources/{}/pause".format(self.factory.data_source.id), user=admin)
     self.assertEqual(rv.status_code, 200)
     self.assertEqual(DataSource.get_by_id(self.factory.data_source.id).paused, False)

Example #3

File: test_data_sources.py Project: solutionrooms/testredash

 def test_pauses_data_source(self):
     admin = self.factory.create_admin()
     rv = self.make_request('post',
                            '/api/data_sources/{}/pause'.format(
                                self.factory.data_source.id),
                            user=admin)
     self.assertEqual(rv.status_code, 200)
     self.assertEqual(
         DataSource.get_by_id(self.factory.data_source.id).paused, True)

Example #4

File: test_groups.py Project: 5t111111/redash

    def test_can_delete_group_with_data_sources(self):
        group = self.factory.create_group()
        data_source = self.factory.create_data_source(group=group)

        response = self.make_request('delete', '/api/groups/{}'.format(group.id), user=self.factory.create_admin())

        self.assertEqual(response.status_code, 200)

        self.assertEqual(data_source, DataSource.get_by_id(data_source.id))

Example #5

File: test_data_sources.py Project: solutionrooms/testredash

 def test_resumes_data_source(self):
     admin = self.factory.create_admin()
     self.factory.data_source.pause()
     self.factory.data_source.save()
     rv = self.make_request('delete',
                            '/api/data_sources/{}/pause'.format(
                                self.factory.data_source.id),
                            user=admin)
     self.assertEqual(rv.status_code, 200)
     self.assertEqual(
         DataSource.get_by_id(self.factory.data_source.id).paused, False)

Example #6

    def test_can_delete_group_with_data_sources(self):
        group = self.factory.create_group()
        data_source = self.factory.create_data_source(group=group)

        response = self.make_request('delete',
                                     '/api/groups/{}'.format(group.id),
                                     user=self.factory.create_admin())

        self.assertEqual(response.status_code, 200)

        self.assertEqual(data_source, DataSource.get_by_id(data_source.id))

Example #7

File: test_data_sources.py Project: solutionrooms/testredash

    def test_pause_sets_reason(self):
        admin = self.factory.create_admin()
        rv = self.make_request('post',
                               '/api/data_sources/{}/pause'.format(
                                   self.factory.data_source.id),
                               user=admin,
                               data={'reason': 'testing'})
        self.assertEqual(rv.status_code, 200)
        self.assertEqual(
            DataSource.get_by_id(self.factory.data_source.id).paused, True)
        self.assertEqual(
            DataSource.get_by_id(self.factory.data_source.id).pause_reason,
            'testing')

        rv = self.make_request('post',
                               '/api/data_sources/{}/pause?reason=test'.format(
                                   self.factory.data_source.id),
                               user=admin)
        self.assertEqual(
            DataSource.get_by_id(self.factory.data_source.id).pause_reason,
            'test')

Example #8

File: test_data_sources.py Project: ryotarai/redash

    def test_updates_data_source(self):
        admin = self.factory.create_admin()
        new_name = "New Name"
        new_options = {"dbname": "newdb"}
        rv = self.make_request(
            "post", self.path, data={"name": new_name, "type": "pg", "options": new_options}, user=admin
        )

        self.assertEqual(rv.status_code, 200)
        data_source = DataSource.get_by_id(self.factory.data_source.id)

        self.assertEqual(data_source.name, new_name)
        self.assertEqual(data_source.options.to_dict(), new_options)

Example #9

    def test_updates_data_source(self):
        admin = self.factory.create_admin()
        new_name = 'New Name'
        new_options = {"dbname": "newdb"}
        rv = self.make_request('post', self.path,
                               data={'name': new_name, 'type': 'pg', 'options': new_options},
                               user=admin)

        self.assertEqual(rv.status_code, 200)
        data_source = DataSource.get_by_id(self.factory.data_source.id)

        self.assertEqual(data_source.name, new_name)
        self.assertEqual(data_source.options.to_dict(), new_options)

Example #10

File: test_data_sources.py Project: ryotarai/redash

 def test_pauses_data_source(self):
     admin = self.factory.create_admin()
     rv = self.make_request("post", "/api/data_sources/{}/pause".format(self.factory.data_source.id), user=admin)
     self.assertEqual(rv.status_code, 200)
     self.assertEqual(DataSource.get_by_id(self.factory.data_source.id).paused, True)