Example #1
0
class RiakTest(BaseTest):
    def setUp(self):
        self.expected_defaults = {
            'host': '127.0.0.1',
            'port': 8098,
            'prefix': 'riak',
            'mapred_prefix': 'mapred',
            'client_id': None,
        }

        self.modified_props = {
            'host': '127.0.0.254',
            'port': 8908,
            'prefix': 'kair',
            'mapred_prefix': 'derpam',
            'client_id': 'MjgxMDg2MzQx',
        }

        self.conn = Riak(num=0)
        self.modified_conn = Riak(num=1, **self.modified_props)

    def test_init_defaults(self):
        self.assertDictContainsSubset(self.expected_defaults,
                                      self.conn.__dict__)

    def test_init_properties(self):
        self.assertDictContainsSubset(self.modified_props,
                                      self.modified_conn.__dict__)

    def test_identifier(self):
        expected_identifier = 'http://%(host)s:%(port)s/%(prefix)s' % self.conn.__dict__
        self.assertEquals(expected_identifier, self.conn.identifier)

    def test_identifier_properties(self):
        expected_identifier = 'http://%(host)s:%(port)s/%(prefix)s' % self.modified_props
        self.assertEquals(expected_identifier, self.modified_conn.identifier)

    @mock.patch('nydus.db.backends.riak.RiakClient')
    def test_connect_riakclient_options(self, _RiakClient):
        self.conn.connect()

        _RiakClient.assert_called_with(host=self.conn.host, port=self.conn.port, prefix=self.conn.prefix, \
                                        mapred_prefix=self.conn.mapred_prefix, client_id=self.conn.client_id)

    @mock.patch('nydus.db.backends.riak.RiakClient')
    def test_connect_riakclient_modified_options(self, _RiakClient):
        self.modified_conn.connect()

        _RiakClient.assert_called_with(host=self.modified_conn.host, port=self.modified_conn.port, prefix=self.modified_conn.prefix, \
                                        mapred_prefix=self.modified_conn.mapred_prefix, client_id=self.modified_conn.client_id)

    def test_connect_returns_riakclient(self):
        client = self.conn.connect()

        self.assertIsInstance(client, RiakClient)

    def test_provides_retryable_exceptions(self):
        self.assertItemsEqual([RiakError, HTTPException, SocketError],
                              self.conn.retryable_exceptions)
Example #2
0
class RiakTest(BaseTest):
    def setUp(self):
        self.expected_defaults = {
            'host': '127.0.0.1',
            'port': 8098,
            'prefix': 'riak',
            'mapred_prefix': 'mapred',
            'client_id': None,
        }
        
        self.modified_props = {
            'host': '127.0.0.254',
            'port': 8908,
            'prefix': 'kair',
            'mapred_prefix': 'derpam',
            'client_id': 'MjgxMDg2MzQx',
        }

        self.conn = Riak(num=0)
        self.modified_conn = Riak(num=1, **self.modified_props)

    def test_init_defaults(self):
        self.assertDictContainsSubset(self.expected_defaults, self.conn.__dict__)

    def test_init_properties(self):
        self.assertDictContainsSubset(self.modified_props, self.modified_conn.__dict__)

    def test_identifier(self):
        expected_identifier = 'http://%(host)s:%(port)s/%(prefix)s' % self.conn.__dict__
        self.assertEquals(expected_identifier, self.conn.identifier)

    def test_identifier_properties(self):
        expected_identifier = 'http://%(host)s:%(port)s/%(prefix)s' % self.modified_props
        self.assertEquals(expected_identifier, self.modified_conn.identifier)

    @mock.patch('nydus.db.backends.riak.RiakClient')
    def test_connect_riakclient_options(self, _RiakClient):
        self.conn.connect()

        _RiakClient.assert_called_with(host=self.conn.host, port=self.conn.port, prefix=self.conn.prefix, \
                                        mapred_prefix=self.conn.mapred_prefix, client_id=self.conn.client_id)

    @mock.patch('nydus.db.backends.riak.RiakClient')
    def test_connect_riakclient_modified_options(self, _RiakClient):
        self.modified_conn.connect()

        _RiakClient.assert_called_with(host=self.modified_conn.host, port=self.modified_conn.port, prefix=self.modified_conn.prefix, \
                                        mapred_prefix=self.modified_conn.mapred_prefix, client_id=self.modified_conn.client_id)

    def test_connect_returns_riakclient(self):
        client = self.conn.connect()

        self.assertIsInstance(client, RiakClient)
        
    def test_provides_retryable_exceptions(self):
        self.assertItemsEqual([RiakError, HTTPException, SocketError], self.conn.retryable_exceptions)
Example #3
0
    def setUp(self):
        self.expected_defaults = {
            'host': '127.0.0.1',
            'port': 8098,
            'prefix': 'riak',
            'mapred_prefix': 'mapred',
            'client_id': None,
        }

        self.modified_props = {
            'host': '127.0.0.254',
            'port': 8908,
            'prefix': 'kair',
            'mapred_prefix': 'derpam',
            'client_id': 'MjgxMDg2MzQx',
        }

        self.conn = Riak(num=0)
        self.modified_conn = Riak(num=1, **self.modified_props)
Example #4
0
    def setUp(self):
        self.expected_defaults = {
            'host': '127.0.0.1',
            'port': 8098,
            'prefix': 'riak',
            'mapred_prefix': 'mapred',
            'client_id': None,
        }
        
        self.modified_props = {
            'host': '127.0.0.254',
            'port': 8908,
            'prefix': 'kair',
            'mapred_prefix': 'derpam',
            'client_id': 'MjgxMDg2MzQx',
        }

        self.conn = Riak(num=0)
        self.modified_conn = Riak(num=1, **self.modified_props)
Example #5
0
File: tests.py Project: numan/nydus
    def setUp(self):
        self.expected_defaults = {
            "host": "127.0.0.1",
            "port": 8098,
            "prefix": "riak",
            "mapred_prefix": "mapred",
            "client_id": None,
        }

        self.modified_props = {
            "host": "127.0.0.254",
            "port": 8908,
            "prefix": "kair",
            "mapred_prefix": "derpam",
            "client_id": "MjgxMDg2MzQx",
        }

        self.conn = Riak(num=0)
        self.modified_conn = Riak(num=1, **self.modified_props)
Example #6
0
File: tests.py Project: thoas/nydus
    def setUp(self):
        self.expected_defaults = {
            'host': '127.0.0.1',
            'port': 8098,
            'prefix': 'riak',
            'mapred_prefix': 'mapred',
            'client_id': None,
        }

        self.modified_props = {
            'host': '127.0.0.254',
            'port': 8908,
            'prefix': 'kair',
            'mapred_prefix': 'derpam',
            'client_id': 'MjgxMDg2MzQx',
            'transport_options': {},
            'transport_class': mock.Mock,
            'solr_transport_class': mock.Mock,
        }

        self.conn = Riak(0)
        self.modified_conn = Riak(1, **self.modified_props)
Example #7
0
    def setUp(self):
        self.expected_defaults = {
            'host': '127.0.0.1',
            'port': 8098,
            'prefix': 'riak',
            'mapred_prefix': 'mapred',
            'client_id': None,
        }

        self.modified_props = {
            'host': '127.0.0.254',
            'port': 8908,
            'prefix': 'kair',
            'mapred_prefix': 'derpam',
            'client_id': 'MjgxMDg2MzQx',
            'transport_options': {},
            'transport_class': mock.Mock,
            'solr_transport_class': mock.Mock,
        }

        self.conn = Riak(0)
        self.modified_conn = Riak(1, **self.modified_props)
Example #8
0
File: tests.py Project: numan/nydus
class RiakTest(BaseTest):
    def setUp(self):
        self.expected_defaults = {
            "host": "127.0.0.1",
            "port": 8098,
            "prefix": "riak",
            "mapred_prefix": "mapred",
            "client_id": None,
        }

        self.modified_props = {
            "host": "127.0.0.254",
            "port": 8908,
            "prefix": "kair",
            "mapred_prefix": "derpam",
            "client_id": "MjgxMDg2MzQx",
        }

        self.conn = Riak(num=0)
        self.modified_conn = Riak(num=1, **self.modified_props)

    def test_init_defaults(self):
        self.assertDictContainsSubset(self.expected_defaults, self.conn.__dict__)

    def test_init_properties(self):
        self.assertDictContainsSubset(self.modified_props, self.modified_conn.__dict__)

    def test_identifier(self):
        expected_identifier = "http://%(host)s:%(port)s/%(prefix)s" % self.conn.__dict__
        self.assertEquals(expected_identifier, self.conn.identifier)

    def test_identifier_properties(self):
        expected_identifier = "http://%(host)s:%(port)s/%(prefix)s" % self.modified_props
        self.assertEquals(expected_identifier, self.modified_conn.identifier)

    @mock.patch("nydus.db.backends.riak.RiakClient")
    def test_connect_riakclient_options(self, _RiakClient):
        self.conn.connect()

        _RiakClient.assert_called_with(
            host=self.conn.host,
            port=self.conn.port,
            prefix=self.conn.prefix,
            mapred_prefix=self.conn.mapred_prefix,
            client_id=self.conn.client_id,
        )

    @mock.patch("nydus.db.backends.riak.RiakClient")
    def test_connect_riakclient_modified_options(self, _RiakClient):
        self.modified_conn.connect()

        _RiakClient.assert_called_with(
            host=self.modified_conn.host,
            port=self.modified_conn.port,
            prefix=self.modified_conn.prefix,
            mapred_prefix=self.modified_conn.mapred_prefix,
            client_id=self.modified_conn.client_id,
        )

    def test_connect_returns_riakclient(self):
        client = self.conn.connect()

        self.assertIsInstance(client, RiakClient)

    def test_provides_retryable_exceptions(self):
        self.assertItemsEqual([RiakError, HTTPException, SocketError], self.conn.retryable_exceptions)