Beispiel #1
0
def assert_all(session,
               query,
               expected,
               cl=None,
               ignore_order=False,
               timeout=None):
    """
    Assert query returns all expected items optionally in the correct order
    @param session Session in use
    @param query Query to run
    @param expected Expected results from query
    @param cl Optional Consistency Level setting. Default ONE
    @param ignore_order Optional boolean flag determining whether response is ordered
    @param timeout Optional query timeout, in seconds

    Examples:
    assert_all(session, "LIST USERS", [['aleksey', False], ['cassandra', True]])
    assert_all(self.session1, "SELECT * FROM ttl_table;", [[1, 42, 1, 1]])
    """
    simple_query = SimpleStatement(query, consistency_level=cl)
    res = session.execute(
        simple_query) if timeout is None else session.execute(simple_query,
                                                              timeout=timeout)
    list_res = _rows_to_list(res)
    if ignore_order:
        expected = list_to_hashed_dict(expected)
        list_res = list_to_hashed_dict(list_res)
    assert list_res == expected, "Expected {} from {}, but got {}".format(
        expected, query, list_res)
Beispiel #2
0
    def assertEqualIgnoreOrder(self, actual, expected):
        hashed_expected = list_to_hashed_dict(expected)
        hashed_actual = list_to_hashed_dict(actual)
        for key, expected in hashed_expected.items():
            assert key in hashed_actual, "expected %s not in actual" % str(expected)
            actual = hashed_actual[key]
            assert actual == expected, "actual %s not same as expected %s" % (str(actual), str(expected))

        for key, actual in hashed_actual.items():
            assert key in hashed_expected, "actual %s not in expected" % str(actual)
            expected = hashed_expected[key]
            assert expected == actual, "expected %s not same as actual %s" % (str(expected), str(actual))

        assert hashed_expected == hashed_actual
Beispiel #3
0
def assert_all(session, query, expected, cl=None, ignore_order=False, timeout=None):
    """
    Assert query returns all expected items optionally in the correct order
    @param session Session in use
    @param query Query to run
    @param expected Expected results from query
    @param cl Optional Consistency Level setting. Default ONE
    @param ignore_order Optional boolean flag determining whether response is ordered
    @param timeout Optional query timeout, in seconds

    Examples:
    assert_all(session, "LIST USERS", [['aleksey', False], ['cassandra', True]])
    assert_all(self.session1, "SELECT * FROM ttl_table;", [[1, 42, 1, 1]])
    """
    simple_query = SimpleStatement(query, consistency_level=cl)
    res = session.execute(simple_query) if timeout is None else session.execute(simple_query, timeout=timeout)
    list_res = _rows_to_list(res)
    if ignore_order:
        expected = list_to_hashed_dict(expected)
        list_res = list_to_hashed_dict(list_res)
    assert list_res == expected, "Expected {} from {}, but got {}".format(expected, query, list_res)
 def assertEqualIgnoreOrder(self, actual, expected):
     assert list_to_hashed_dict(actual) == list_to_hashed_dict(expected)