def test_port_translate_filter_func_not_in_double():
"""Test to see if the port_translate_filter_func behaves as expected when the NOT_IN
operator is used with two ports"""
input_column = Column("user_ip", Integer)
input_operation = FilterOperator.NOT_IN
input_values = [[443, 80]]
port_translate_filter_response: sqlalchemy.sql.expression.BinaryExpression = ~(
input_column.in_(input_values[0]))
assert port.translate_filter(
input_column, input_operation,
input_values).compare(port_translate_filter_response)
def test_port_translate_filter_func_less_than():
"""Test to see if the port_translate_filter_func behaves as expected when the LESS_THAN
operator is used"""
input_column = Column("user_ip", Integer)
input_operation = FilterOperator.LESS_THAN
input_values = [[443]]
port_translate_filter_response: sqlalchemy.sql.expression.BinaryExpression = (
input_column < input_values[0][0])
assert port.translate_filter(
input_column, input_operation,
input_values).compare(port_translate_filter_response)
def test_port_translate_filter_func_in_single(app_context: None):
"""Test to see if the port_translate_filter_func behaves as expected when the IN operator
is used with a single port"""
input_column = Column("user_ip", Integer)
input_operation = FilterOperator.IN
input_values = [[443]]
port_translate_filter_response: sqlalchemy.sql.expression.BinaryExpression = (
input_column.in_(input_values[0]))
assert port.translate_filter(
input_column, input_operation,
input_values).compare(port_translate_filter_response)
def test_port_translate_filter_func_greater_than_or_equals():
"""Test to see if the port_translate_filter_func behaves as expected when the
GREATER_THAN_OR_EQUALS operator is used"""
input_column = Column("user_ip", Integer)
input_operation = FilterOperator.GREATER_THAN_OR_EQUALS
input_values = [[443]]
port_translate_filter_response: sqlalchemy.sql.expression.BinaryExpression = (
input_column >= input_values[0][0])
assert port.translate_filter(
input_column, input_operation,
input_values).compare(port_translate_filter_response)
def test_port_translate_filter_func_not_equals(app_context: None):
"""Test to see if the port_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 = [[443]]
port_translate_filter_response: sqlalchemy.sql.expression.BinaryExpression = ~(
input_column.in_(input_values[0]))
assert port.translate_filter(
input_column, input_operation,
input_values).compare(port_translate_filter_response)