Ejemplos de get_tubes en Python

Ejemplo n.º 1

Archivo: test_util.py Proyecto: EasyPost/beancmd

def test_get_tubes_with_all_tubes():
    mock_client = mock.Mock(spec=pystalk.BeanstalkClient)
    mock_client.list_tubes.return_value = ['foo', 'default']

    assert util.get_tubes(mock_client, []) == set(['foo', 'default'])

    assert util.get_tubes(mock_client, None) == set(['foo', 'default'])

Ejemplo n.º 2

Archivo: test_util.py Proyecto: EasyPost/beancmd

def test_get_tubes_with_exact_list():
    mock_client = mock.Mock(spec=pystalk.BeanstalkClient)
    mock_client.list_tubes.return_value = ['foo', 'bar', 'default']

    assert util.get_tubes(mock_client, ['bar', 'foo']) == set(['foo', 'bar'])