ydb/library/python/resource/ut/lib/test_simple.py

import six  # noqa

import library.python.resource as rs

text = b'na gorshke sidel korol\n'


def test_find():
    assert rs.find('/qw.txt') == text


def test_iter():
    assert set(rs.iterkeys()).issuperset({'/qw.txt', '/prefix/1.txt', '/prefix/2.txt'})
    assert set(rs.iterkeys(prefix='/prefix/')) == {'/prefix/1.txt', '/prefix/2.txt'}
    assert set(rs.iterkeys(prefix='/prefix/', strip_prefix=True)) == {'1.txt', '2.txt'}
    assert set(rs.iteritems(prefix='/prefix')) == {
        ('/prefix/1.txt', text),
        ('/prefix/2.txt', text),
    }
    assert set(rs.iteritems(prefix='/prefix', strip_prefix=True)) == {
        ('/1.txt', text),
        ('/2.txt', text),
    }


def test_resfs_files():
    assert 'contrib/python/py/.dist-info/METADATA' in set(rs.resfs_files())


def test_resfs_read():
    assert 'Metadata-Version' in rs.resfs_read('contrib/python/py/.dist-info/METADATA').decode('utf-8')