Exemple #1
0
    def test_create_database_cps(self):
        client = AsyncfluxClient()
        db_name = 'foo'

        # Using a string
        with self.patch_fetch_mock(client) as m:
            self.setup_fetch_mock(m, 201)
            client.create_database(db_name, callback=self.stop_op)
            response = self.wait()
            self.assertIsInstance(response, Database)
            self.assertEqual(response.name, db_name)

            self.assert_mock_args(m,
                                  '/db',
                                  method='POST',
                                  body=json.dumps({'name': db_name}))

        # Existing database
        response_body = 'database %s exists' % db_name
        with self.patch_fetch_mock(client) as m:
            self.setup_fetch_mock(m, 409, body=response_body)
            with self.assertRaisesRegexp(AsyncfluxError, response_body):
                client.create_database(db_name, callback=self.stop_op)
                response = self.wait()
                self.assertEqual(response, response_body)

            self.assert_mock_args(m,
                                  '/db',
                                  method='POST',
                                  body=json.dumps({'name': db_name}))
    def test_create_database_cps(self):
        client = AsyncfluxClient()
        db_name = 'foo'

        # Using a string
        with self.patch_fetch_mock(client) as m:
            self.setup_fetch_mock(m, 201)
            client.create_database(db_name, callback=self.stop_op)
            response = self.wait()
            self.assertIsInstance(response, Database)
            self.assertEqual(response.name, db_name)

            self.assert_mock_args(m, '/db', method='POST',
                                  body=json.dumps({'name': db_name}))

        # Existing database
        response_body = 'database %s exists' % db_name
        with self.patch_fetch_mock(client) as m:
            self.setup_fetch_mock(m, 409, body=response_body)
            with self.assertRaisesRegexp(AsyncfluxError, response_body):
                client.create_database(db_name, callback=self.stop_op)
                response = self.wait()
                self.assertEqual(response, response_body)

            self.assert_mock_args(m, '/db', method='POST',
                                  body=json.dumps({'name': db_name}))
Exemple #3
0
    def test_create_database(self):
        client = AsyncfluxClient()
        db_name = 'foo'

        # Using a string
        with self.patch_fetch_mock(client) as m:
            self.setup_fetch_mock(m, 201)
            response = yield client.create_database(db_name)
            self.assertIsInstance(response, Database)
            self.assertEqual(response.name, db_name)

            self.assert_mock_args(m,
                                  '/db',
                                  method='POST',
                                  body=json.dumps({'name': db_name}))

        # Using an instance of Database
        db = Database(client, db_name)
        with self.patch_fetch_mock(client) as m:
            self.setup_fetch_mock(m, 201)
            response = yield client.create_database(db)
            self.assertIsInstance(response, Database)
            self.assertEqual(response.name, db_name)

            self.assert_mock_args(m,
                                  '/db',
                                  method='POST',
                                  body=json.dumps({'name': db_name}))

        # Using an unsupported type
        with self.patch_fetch_mock(client) as m:
            re_exc_msg = r'^name_or_database must be an instance'
            with self.assertRaisesRegexp(TypeError, re_exc_msg):
                yield client.create_database(None)
            self.assertFalse(m.called)

        # Existing database
        response_body = 'database %s exists' % db_name
        with self.patch_fetch_mock(client) as m:
            self.setup_fetch_mock(m, 409, body=response_body)
            with self.assertRaisesRegexp(AsyncfluxError, response_body):
                response = yield client.create_database(db_name)
                self.assertEqual(response, response_body)

            self.assert_mock_args(m,
                                  '/db',
                                  method='POST',
                                  body=json.dumps({'name': db_name}))
    def test_create_database(self):
        client = AsyncfluxClient()
        db_name = 'foo'

        # Using a string
        with self.patch_fetch_mock(client) as m:
            self.setup_fetch_mock(m, 201)
            response = yield client.create_database(db_name)
            self.assertIsInstance(response, Database)
            self.assertEqual(response.name, db_name)

            self.assert_mock_args(m, '/db', method='POST',
                                  body=json.dumps({'name': db_name}))

        # Using an instance of Database
        db = Database(client, db_name)
        with self.patch_fetch_mock(client) as m:
            self.setup_fetch_mock(m, 201)
            response = yield client.create_database(db)
            self.assertIsInstance(response, Database)
            self.assertEqual(response.name, db_name)

            self.assert_mock_args(m, '/db', method='POST',
                                  body=json.dumps({'name': db_name}))

        # Using an unsupported type
        with self.patch_fetch_mock(client) as m:
            re_exc_msg = r'^name_or_database must be an instance'
            with self.assertRaisesRegexp(TypeError, re_exc_msg):
                yield client.create_database(None)
            self.assertFalse(m.called)

        # Existing database
        response_body = 'database %s exists' % db_name
        with self.patch_fetch_mock(client) as m:
            self.setup_fetch_mock(m, 409, body=response_body)
            with self.assertRaisesRegexp(AsyncfluxError, response_body):
                response = yield client.create_database(db_name)
                self.assertEqual(response, response_body)

            self.assert_mock_args(m, '/db', method='POST',
                                  body=json.dumps({'name': db_name}))