Beispiel #1
0
def test_load_from_file(tmpdir):
    tmpdir.join('my_service.thrift').write('''
        struct Foo {
            1: required string a
            2: optional string b
        }
    ''')

    my_service = Loader().load(str(tmpdir.join('my_service.thrift')))
    my_service.Foo(b='b', a='a')
def test_include_as_disabled(tmpdir):
    loader = Loader()
    tmpdir.join('a.thrift').write('include t "./b.thrift"')
    tmpdir.join('b.thrift').write('typedef string UUID')

    with pytest.raises(ThriftCompilerError) as exc_info:
        loader.load(str(tmpdir.join('a.thrift')))

    assert 'Cannot include "b" as "t"' in str(exc_info)
    assert '"include-as" syntax is currently disabled' in str(exc_info)
Beispiel #3
0
def test_load_from_file_non_strict_missing_requiredness(tmpdir):
    tmpdir.join('my_service.thrift').write('''
        struct Foo {
            1: required string a
            2: string b
        }
    ''')

    loader = Loader(strict=False)
    my_service = loader.load(str(tmpdir.join('my_service.thrift')))
    my_service.Foo(b='b', a='a')
def test_load_from_file_non_strict_missing_requiredness(tmpdir):
    tmpdir.join('my_service.thrift').write('''
        struct Foo {
            1: required string a
            2: string b
        }
    ''')

    loader = Loader(strict=False)
    my_service = loader.load(str(tmpdir.join('my_service.thrift')))
    my_service.Foo(b='b', a='a')
Beispiel #5
0
def test_caching(tmpdir, monkeypatch):
    tmpdir.join('my_service.thrift').write('''
        struct Foo {
            1: required string a
            2: optional string b
        }
    ''')

    path = str(tmpdir.join('my_service.thrift'))
    loader = Loader()

    mod1 = loader.load(path)
    mod2 = loader.load(path)
    assert mod1 is mod2
Beispiel #6
0
def test_load_from_file_non_strict_missing_requiredness(tmpdir):
    tmpdir.join('my_service.thrift').write('''
        struct Foo {
            1: required string a
            2: string b
        }

        exception Bar {
            1: string message
        }
    ''')

    loader = Loader(strict=False)
    m = loader.load(str(tmpdir.join('my_service.thrift')))
    m.Foo(b='b', a='a')

    m.Bar(message='foo')
    m.Bar()
Beispiel #7
0
def test_caching(tmpdir, monkeypatch):
    tmpdir.join('my_service.thrift').write('''
        struct Foo {
            1: required string a
            2: optional string b
        }
    ''')

    path = str(tmpdir.join('my_service.thrift'))
    loader = Loader()

    mod1 = loader.load(path)
    assert path in loader.compiled_modules

    mod2 = loader.load(path)
    assert mod1 is mod2

    mod3 = loader.load(path, force=True)
    assert mod3 is not mod2
Beispiel #8
0
def test_load_from_file_missing_requiredness(tmpdir):
    tmpdir.join('my_service.thrift').write('''
        struct Foo {
            1: required string a
            2: string b
        }
    ''')

    with pytest.raises(ThriftCompilerError) as exc_info:
        Loader().load(str(tmpdir.join('my_service.thrift')))

    assert ('"b" of "Foo" on line 4 does not explicitly specify requiredness.'
            in str(exc_info))
import pytest

from thriftrw.loader import Loader
from thriftrw.protocol import BinaryProtocol
from thriftrw.wire import mtype


service = Loader(BinaryProtocol()).loads(
    'runtime_service',
    '''
        exception GreatSadness {
            1: optional string message
        }

        service Service {
            oneway void write(1: binary data)

            i32 read()
                throws (1: GreatSadness sadness)

        }
    '''
)
Service = service.Service


@pytest.mark.parametrize('name, seqid, typ, obj, bs', [
    (b'write', 42, mtype.ONEWAY, Service.write.request(b'hello'), [
        0x00, 0x00, 0x00, 0x05,         # length:4 = 5
        0x77, 0x72, 0x69, 0x74, 0x65,   # 'write'
        0x04,                           # type:1 = ONEWAY
Beispiel #10
0
def loads(request):
    return partial(Loader(BinaryProtocol()).loads, request.node.name)
def loads(request):
    return partial(Loader().loads, request.node.name)