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()
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 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)
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)
def test_merge_vmargs(self): self.test_server = TestServer(vm_args={"-P": 65000}) self.assertEqual(self.test_server.vm_args["-P"], 65000)
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")
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")
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)
def setUp(self): self.test_server = TestServer()
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
def setUp(self): super(RiakTestCase, self).setUp() self.testserver = TestServer(bin_dir=os.environ["RIAK_BIN"])
def test_merge_luwak_options(self): self.test_server = TestServer(luwak={"enabled": False}) self.assertEquals(self.test_server.app_config["luwak"]["enabled"], False)