Example #1
0
    def test_that_get_node_or_fail_calls_coroutine_in_fail_case(self):
        class CallableCheck(object):
            called = False
            args = None
            kwargs = None

            def __call__(self, *args, **kwargs):
                self.called = True
                self.args = args
                self.kwargs = kwargs

        coroutine = CallableCheck()
        cargs = ("Error happened", )
        ckwargs = {"node_id": "15"}

        conn = self.conn_cls([])

        node = get_node_or_fail(conn, "15", coroutine, cargs, ckwargs)

        assert_true(coroutine.called)
        assert_equal(coroutine.args, cargs)
        assert_equal(coroutine.kwargs, ckwargs)
Example #2
0
    def test_that_get_node_or_fail_calls_coroutine_in_fail_case(self):
        class CallableCheck(object):
            called = False
            args = None
            kwargs = None

            def __call__(self, *args, **kwargs):
                self.called = True
                self.args = args
                self.kwargs = kwargs

        coroutine = CallableCheck()
        cargs = ("Error happened",)
        ckwargs = {"node_id": "15"}

        conn = self.conn_cls([])

        node = get_node_or_fail(conn, "15",
                coroutine, cargs, ckwargs)

        assert_true(coroutine.called)
        assert_equal(coroutine.args, cargs)
        assert_equal(coroutine.kwargs, ckwargs)
Example #3
0
    def test_that_get_node_or_fail_returns_none_in_fail_case(self):
        conn = self.conn_cls([])

        node = get_node_or_fail(conn, "15")

        assert_true(node is None)
Example #4
0
    def test_that_get_node_or_fail_returns_node_object_for_existing_node(self):
        conn = self.conn_cls([self.node_cls("15"), self.node_cls("21")])

        node = get_node_or_fail(conn, "15")

        assert_equal(node.id, "15")
Example #5
0
    def test_that_get_node_or_fail_returns_none_in_fail_case(self):
        conn = self.conn_cls([])

        node = get_node_or_fail(conn, "15")

        assert_true(node is None)
Example #6
0
    def test_that_get_node_or_fail_returns_node_object_for_existing_node(self):
        conn = self.conn_cls([self.node_cls("15"), self.node_cls("21")])

        node = get_node_or_fail(conn, "15")

        assert_equal(node.id, "15")