Exemplo n.º 1
0
    def test_close_when_not_open(self, serial):
        driver = Driver().open('PORTNAME')

        driver.close()
        driver.close()

        assert_that(serial().close, called_once())
        assert_that(driver._serial, is_(None))
Exemplo n.º 2
0
def test_called_once_ok():
    matcher = called_once()
    mock = Mock()
    mock('foo')

    ok = matcher.matches(mock)

    assert_that(ok, equal_to(True))
Exemplo n.º 3
0
def test_called_once_ok():
    matcher = called_once()
    mock = Mock()
    mock('foo')

    ok = matcher.matches(mock)

    assert_that(ok, equal_to(True))
Exemplo n.º 4
0
def test_closes_when_started_with_error(protocol: Protocol,
                                        transport: Transport,
                                        loop: Loop) -> None:
    header = b'59 BAD_REQUEST\r\n'
    res = Response(status=Status.BAD_REQUEST)
    res._start(transport, protocol, loop)
    assert_that(transport.write, called_with(header))
    assert_that(transport.close, called_once())
    res.write(b'more data')
Exemplo n.º 5
0
def test_writes_payload_when_started(protocol: Protocol, transport: Transport,
                                     loop: Loop) -> None:
    data = b'somedata'
    header = b'20 text/gemini; charset=utf-8\r\n'
    res = Response(data=data)
    res._start(transport, protocol, loop)
    assert_that(transport.write, called_with(header))
    assert_that(transport.write, called_with(data))
    assert_that(transport.close, called_once())
Exemplo n.º 6
0
def test_called_once_fail_not_called(desc):
    expected = 'was called 0 times'
    matcher = called_once()
    mock = Mock()

    ok = matcher.matches(mock, desc)

    assert_that(ok, equal_to(False))
    assert_that(str(desc), equal_to(expected))
Exemplo n.º 7
0
def test_called_once_fail_not_called(desc):
    expected = 'was called 0 times'
    matcher = called_once()
    mock = Mock()

    ok = matcher.matches(mock, desc)

    assert_that(ok, equal_to(False))
    assert_that(str(desc), equal_to(expected))
Exemplo n.º 8
0
def test_called_once_fail_called_twice(desc):
    expected = "was called 2 times with ('foo', ) and ('bar', )"
    matcher = called_once()
    mock = Mock()
    mock('foo')
    mock('bar')

    ok = matcher.matches(mock, desc)

    assert_that(ok, equal_to(False))
    assert_that(str(desc), equal_to(expected))
Exemplo n.º 9
0
def test_called_once_fail_called_twice(desc):
    expected = "was called 2 times with ('foo', ) and ('bar', )"
    matcher = called_once()
    mock = Mock()
    mock('foo')
    mock('bar')

    ok = matcher.matches(mock, desc)

    assert_that(ok, equal_to(False))
    assert_that(str(desc), equal_to(expected))
Exemplo n.º 10
0
def test_called_once_fail_not_called():
    expected = '''
Expected: Mock called with ANYTHING 1 times
     but: was called 0 times
'''

    mock = Mock()

    with assert_raises(AssertionError) as e:
        assert_that(mock, called_once())

    assert_that(str(e.exception), equal_to(expected))
Exemplo n.º 11
0
async def test_drain(protocol: Protocol, transport: Transport,
                     loop: Loop) -> None:
    res = Response()
    res._start(transport, protocol, loop)
    await res.drain()
    assert_that(protocol, has_property('_drain_helper', called_once()))
Exemplo n.º 12
0
    def test_close(self):
        self.usb_iss.open('PORTNAME')

        self.usb_iss.close()

        assert_that(self.driver.close, called_once())
Exemplo n.º 13
0
def test_called_once_ok():
    mock = Mock()
    mock('foo')

    assert_that(mock, called_once())
Exemplo n.º 14
0
def test_called_once_description(desc):
    expected = 'Mock called <1> times with ANYTHING'
    matcher = called_once()
    matcher.describe_to(desc)
    assert_that(str(desc), equal_to(expected))
Exemplo n.º 15
0
def test_called_once_description(desc):
    expected = 'Mock called <1> times with ANYTHING'
    matcher = called_once()
    matcher.describe_to(desc)
    assert_that(str(desc), equal_to(expected))