def test_cidr_translate_filter_func_not_equals():
"""Test to see if the cidr_translate_filter_func behaves as expected when the NOT_EQUALS
operator is used"""
input_column = Column("user_ip", Integer)
input_operation = FilterOperator.NOT_EQUALS
input_values = [16843009]
cidr_translate_filter_response = input_column != input_values[0]
assert internet_address.translate_filter(
input_column, input_operation,
input_values).compare(cidr_translate_filter_response)
def test_cidr_translate_filter_func_not_in_single():
"""Test to see if the cidr_translate_filter_func behaves as expected when the NOT_IN
operator is used with a single IP"""
input_column = Column("user_ip", Integer)
input_operation = FilterOperator.NOT_IN
input_values = [16843009]
cidr_translate_filter_response: sqlalchemy.sql.expression.BinaryExpression = ~(
input_column.in_(input_values))
assert internet_address.translate_filter(
input_column, input_operation,
input_values).compare(cidr_translate_filter_response)
def test_cidr_translate_filter_func_less_than_or_equals():
"""Test to see if the cidr_translate_filter_func behaves as expected when the
LESS_THAN_OR_EQUALS operator is used"""
input_column = Column("user_ip", Integer)
input_operation = FilterOperator.LESS_THAN_OR_EQUALS
input_values = [16843009]
cidr_translate_filter_response: sqlalchemy.sql.expression.BinaryExpression = (
input_column <= input_values[0])
assert internet_address.translate_filter(
input_column, input_operation,
input_values).compare(cidr_translate_filter_response)
def test_cidr_translate_filter_func_greater_than():
"""Test to see if the cidr_translate_filter_func behaves as expected when the
GREATER_THAN operator is used"""
input_column = Column("user_ip", Integer)
input_operation = FilterOperator.GREATER_THAN
input_values = [16843009]
cidr_translate_filter_response: sqlalchemy.sql.expression.BinaryExpression = (
input_column > input_values[0])
assert internet_address.translate_filter(
input_column, input_operation,
input_values).compare(cidr_translate_filter_response)
def test_cidr_translate_filter_func_not_in_double():
"""Test to see if the cidr_translate_filter_func behaves as expected when the NOT_IN
operator is used with two IP's"""
input_column = Column("user_ip", Integer)
input_operation = FilterOperator.NOT_IN
input_values = [{"start": 16843009, "end": 33686018}]
input_condition = ~(input_column.in_([]))
cidr_translate_filter_response: sqlalchemy.sql.expression.BinaryExpression = (
input_condition & (input_column > 33686018) &
(input_column < 16843009))
assert internet_address.translate_filter(
input_column, input_operation,
input_values).compare(cidr_translate_filter_response)