Esempio n. 1
0
    def test_map(self):
        c = DummyConnection(num=0, resp="foo")
        c2 = DummyConnection(num=1, resp="bar")

        # test dummy router
        r = DummyRouter()
        p = Cluster(hosts={0: c, 1: c2}, router=r)
        with p.map() as conn:
            foo = conn.foo()
            bar = conn.foo("foo")
            self.assertEquals(foo, None)
            self.assertEquals(bar, None)

        self.assertEquals(bar, "bar")
        self.assertEquals(foo, "foo")

        # test default routing behavior
        p = Cluster(hosts={0: c, 1: c2})
        with p.map() as conn:
            foo = conn.foo()
            bar = conn.foo("foo")
            self.assertEquals(foo, None)
            self.assertEquals(bar, None)

        self.assertEquals(foo, ["foo", "bar"])
        self.assertEquals(bar, ["foo", "bar"])
Esempio n. 2
0
 def test_disconnect(self):
     c = Dingus()
     p = Cluster(
         hosts={0: c},
     )
     p.disconnect()
     calls = dingus_calls_to_dict(c.calls)
     self.assertTrue('disconnect' in calls)
Esempio n. 3
0
    def test_get_conn(self):
        c = DummyConnection(alias="foo", num=0, resp="foo")
        c2 = DummyConnection(alias="foo", num=1, resp="bar")

        # test dummy router
        r = DummyRouter()
        p = Cluster(hosts={0: c, 1: c2}, router=r)
        self.assertEquals(p.get_conn(), c)
        self.assertEquals(p.get_conn("foo"), c2)

        # test default routing behavior
        p = Cluster(hosts={0: c, 1: c2})
        self.assertEquals(p.get_conn(), [c, c2])
        self.assertEquals(p.get_conn("foo"), [c, c2])
Esempio n. 4
0
    def test_with_router(self):
        c = DummyConnection(num=0, resp="foo")
        c2 = DummyConnection(num=1, resp="bar")

        # test dummy router
        r = DummyRouter()
        p = Cluster(hosts={0: c, 1: c2}, router=r)
        self.assertEquals(p.foo(), "foo")
        self.assertEquals(p.foo("foo"), "bar")

        # test default routing behavior
        p = Cluster(hosts={0: c, 1: c2})
        self.assertEquals(p.foo(), ["foo", "bar"])
        self.assertEquals(p.foo("foo"), ["foo", "bar"])
Esempio n. 5
0
    def test_get_conn(self):
        c = DummyConnection(alias='foo', num=0, resp='foo')
        c2 = DummyConnection(alias='foo', num=1, resp='bar')

        # test dummy router
        r = DummyRouter()
        p = Cluster(
            hosts={0: c, 1: c2},
            router=r,
        )
        self.assertEquals(p.get_conn(), c)
        self.assertEquals(p.get_conn('foo'), c2)

        # test default routing behavior
        p = Cluster(
            hosts={0: c, 1: c2},
        )
        self.assertEquals(p.get_conn(), [c, c2])
        self.assertEquals(p.get_conn('foo'), [c, c2])
Esempio n. 6
0
    def test_with_router(self):
        c = DummyConnection(num=0, resp='foo')
        c2 = DummyConnection(num=1, resp='bar')

        # test dummy router
        r = DummyRouter()
        p = Cluster(
            hosts={0: c, 1: c2},
            router=r,
        )
        self.assertEquals(p.foo(), 'foo')
        self.assertEquals(p.foo('foo'), 'bar')

        # test default routing behavior
        p = Cluster(
            hosts={0: c, 1: c2},
        )
        self.assertEquals(p.foo(), ['foo', 'bar'])
        self.assertEquals(p.foo('foo'), ['foo', 'bar'])
Esempio n. 7
0
 def test_with_cluster(self):
     p = Cluster(
         hosts={0: self.db},
     )
     cursor = p.execute('SELECT 1')
     self.assertEquals(cursor.fetchone(), (1,))
Esempio n. 8
0
 def test_with_cluster(self):
     p = Cluster(
         hosts={0: self.redis},
     )
     self.assertEquals(p.incr('RedisTest_with_cluster'), 1)
Esempio n. 9
0
 def test_proxy(self):
     c = DummyConnection(num=1, resp="bar")
     p = Cluster(hosts={0: c})
     self.assertEquals(p.foo(), "bar")