Ejemplos de Server.assert_called_once_with en Python

Ejemplo n.º 1

Archivo: test_server.py Proyecto: project-rig/spalloc_server

def test_commandline(monkeypatch, config_file, args, cold_start):
    server = Mock()
    Server = Mock(return_value=server)
    server.is_alive.return_value = False
    import spalloc_server.server
    monkeypatch.setattr(spalloc_server.server,
                        "Server", Server)

    main(args.format(config_file).split())

    Server.assert_called_once_with(config_filename=config_file,
                                   cold_start=cold_start)

Ejemplo n.º 2

Archivo: test_server.py Proyecto: project-rig/spalloc_server

def test_keyboard_interrupt(monkeypatch, config_file):
    s = Mock()
    Server = Mock(return_value=s)
    import spalloc_server.server
    monkeypatch.setattr(spalloc_server.server,
                        "Server", Server)

    s.is_alive.side_effect = KeyboardInterrupt
    main([config_file])

    Server.assert_called_once_with(config_filename=config_file,
                                   cold_start=False)
    s.is_alive.assert_called_once_with()
    s.stop_and_join.assert_called_once_with()