Python TestServer Examples

Example #1

File: utils.py Project: AlexMikhalev/django-riak-engine

class TestCase(TestCase):
    """
    This should enable DEBUG and start riak test server 
    server = TestServer(bin_dir="/usr/local/riak/0.14.2/bin")
    You can also overwrite the default settings used to generate the app.config file for the in-memory Riak instance. Just specify a keyword pointing to a dictionary for every section in the app.config like so:
    server = TestServer(riak_core={"web_port": 8080})  
    """
    def setUp(self):
        super(TestCase, self).setUp()
        if getattr(settings, 'TEST_DEBUG', False):
            settings.DEBUG = True
        self.testserver = TestServer(bin_dir=RIAK_PATH)
        self.testserver.prepare()
        self.testserver.start()
    def tearDown(self):
        self.server.recycle()

Example #2

File: utils.py Project: AlexMikhalev/django-riak-engine

 def setUp(self):
     super(TestCase, self).setUp()
     if getattr(settings, 'TEST_DEBUG', False):
         settings.DEBUG = True
     self.testserver = TestServer(bin_dir=RIAK_PATH)
     self.testserver.prepare()
     self.testserver.start()

Example #3

File: model.py Project: Linux2Go/riakalchemy

def connect(host='127.0.0.1', port=8098, test_server=False):
    global client
    if test_server:
        global _test_server
        if _test_server:
            return
        from riak.test_server import TestServer, Atom
        import tempfile
        tmpdir = tempfile.mkdtemp()
        _test_server = TestServer(bin_dir="/usr/sbin",
                                  tmp_dir=tmpdir,
                                  riak_core={"web_port": port, 'handoff_port': port + 1},
                                  riak_kv={'pb_port': port + 2,
                                           'delete_mode': Atom('immediate')})
        _test_server.cleanup()
        _test_server.prepare()
        _test_server.start()

    client = riak.RiakClient(host=host, http_port=port)

Example #4

 def test_set_non_default_tmp_dir(self):
     tmp_dir = '/not/the/default/dir'
     server = TestServer(tmp_dir=tmp_dir)
     self.assertEqual(server.temp_dir, tmp_dir)

Example #5

 def test_merge_vmargs(self):
     self.test_server = TestServer(vm_args={"-P": 65000})
     self.assertEqual(self.test_server.vm_args["-P"], 65000)

Example #6

 def test_merge_riak_kv_options(self):
     self.test_server = TestServer(riak_kv={"pb_ip": "192.168.2.1"})
     self.assertEqual(self.test_server.app_config["riak_kv"]["pb_ip"],
                      "192.168.2.1")

Example #7

 def test_merge_riak_search_options(self):
     self.test_server = TestServer(
         riak_search={"search_backend": "riak_search_backend"})
     self.assertEqual(
         self.test_server.app_config["riak_search"]["search_backend"],
         "riak_search_backend")

Example #8

 def test_merge_riak_core_options(self):
     self.test_server = TestServer(riak_core={"handoff_port": 10000})
     self.assertEqual(
         self.test_server.app_config["riak_core"]["handoff_port"], 10000)

Example #9

 def setUp(self):
     self.test_server = TestServer()

Example #10

File: __init__.py Project: tobixx/riak-python-client

import logging
import os
import socket
import sys

from riak.test_server import TestServer
from riak.security import SecurityCreds

USE_TEST_SERVER = int(os.environ.get('USE_TEST_SERVER', '0'))
if USE_TEST_SERVER:
    HTTP_PORT = 9000
    PB_PORT = 9002
    test_server = TestServer()
    test_server.cleanup()
    test_server.prepare()
    test_server.start()

try:
    __import__('riak.pb')
    HAVE_PROTO = True
except ImportError:
    HAVE_PROTO = False


def hostname_resolves(hostname):
    try:
        socket.gethostbyname(hostname)
        return 1
    except socket.error:
        return 0

Example #11

File: base.py Project: k-bx/django-riak-engine

 def setUp(self):
     super(RiakTestCase, self).setUp()
     self.testserver = TestServer(bin_dir=os.environ["RIAK_BIN"])

Example #12

File: test_server_test.py Project: tazle/riak-python-client

 def test_merge_luwak_options(self):
     self.test_server = TestServer(luwak={"enabled": False})
     self.assertEquals(self.test_server.app_config["luwak"]["enabled"],
                       False)

Example #13

File: __init__.py Project: linkdd/riak-python-client

import logging
import os
import socket
import sys

from riak.test_server import TestServer
from riak.security import SecurityCreds

USE_TEST_SERVER = int(os.environ.get('USE_TEST_SERVER', '0'))
if USE_TEST_SERVER:
    HTTP_PORT = 9000
    PB_PORT = 9002
    test_server = TestServer()
    test_server.cleanup()
    test_server.prepare()
    test_server.start()

try:
    __import__('riak.pb')
    HAVE_PROTO = True
except ImportError:
    HAVE_PROTO = False


def hostname_resolves(hostname):
    try:
        socket.gethostbyname(hostname)
        return 1
    except socket.error:
        return 0