<s> from neutron . plugins . common import constants as p_const <EOL> from neutron . plugins . ml2 . drivers import type_geneve <EOL> from neutron . tests . unit . plugins . ml2 . drivers import base_type_tunnel <EOL> from neutron . tests . unit . plugins . ml2 import test_rpc <EOL> from neutron . tests . unit import testlib_api <EOL> TUNNEL_IP_ONE = "<STR_LIT>" <EOL> TUNNEL_IP_TWO = "<STR_LIT>" <EOL> HOST_ONE = '<STR_LIT>' <EOL> HOST_TWO = '<STR_LIT>' <EOL> class GeneveTypeTest ( base_type_tunnel . TunnelTypeTestMixin , <EOL> testlib_api . SqlTestCase ) : <EOL> DRIVER_CLASS = type_geneve . GeneveTypeDriver <EOL> TYPE = p_const . TYPE_GENEVE <EOL> def test_get_endpoints ( self ) : <EOL> self . driver . add_endpoint ( TUNNEL_IP_ONE , HOST_ONE ) <EOL> self . driver . add_endpoint ( TUNNEL_IP_TWO , HOST_TWO ) <EOL> endpoints = self . driver . get_endpoints ( ) <EOL> for endpoint in endpoints : <EOL> if endpoint [ '<STR_LIT>' ] == TUNNEL_IP_ONE : <EOL> self . assertEqual ( HOST_ONE , endpoint [ '<STR_LIT:host>' ] ) <EOL> elif endpoint [ '<STR_LIT>' ] == TUNNEL_IP_TWO : <EOL> self . assertEqual ( HOST_TWO , endpoint [ '<STR_LIT:host>' ] ) <EOL> class GeneveTypeMultiRangeTest ( base_type_tunnel . TunnelTypeMultiRangeTestMixin , <EOL> testlib_api . SqlTestCase ) : <EOL> DRIVER_CLASS = type_geneve . GeneveTypeDriver <EOL> class GeneveTypeRpcCallbackTest ( base_type_tunnel . TunnelRpcCallbackTestMixin , <EOL> test_rpc . RpcCallbacksTestCase , <EOL> testlib_api . SqlTestCase ) : <EOL> DRIVER_CLASS = type_geneve . GeneveTypeDriver <EOL> TYPE = p_const . TYPE_GENEVE </s>
