66 行
2.5 KiB
Python
66 行
2.5 KiB
Python
import pytest
|
|
|
|
from yandex_music import Block
|
|
|
|
|
|
@pytest.fixture(scope='class')
|
|
def block_with_entity_and_data(block_entity, data):
|
|
return Block(TestBlock.id, TestBlock.type, TestBlock.type_for_from, TestBlock.title, [block_entity],
|
|
TestBlock.description, data), block_entity, data
|
|
|
|
|
|
class TestBlock:
|
|
id = 'ObsM5DkU'
|
|
type = 'personal-playlists'
|
|
type_for_from = 'personal-playlists'
|
|
title = 'Собрано на основе ваших предпочтений'
|
|
description = None
|
|
|
|
def test_expected_values(self, block_with_entity_and_data):
|
|
block, block_entity, data = block_with_entity_and_data
|
|
|
|
assert block.id == self.id
|
|
assert block.type == self.type
|
|
assert block.type_for_from == self.type_for_from
|
|
assert block.title == self.title
|
|
assert block.entities == [block_entity]
|
|
assert block.description == self.description
|
|
assert block.data == data
|
|
|
|
def test_de_json_required(self, client, block_entity):
|
|
json_dict = {'id': self.id, 'type': self.type, 'type_for_from': self.type_for_from, 'title': self.title,
|
|
'entities': [block_entity.to_dict()]}
|
|
block = Block.de_json(json_dict, client)
|
|
|
|
assert block.id == self.id
|
|
assert block.type == self.type
|
|
assert block.type_for_from == self.type_for_from
|
|
assert block.title == self.title
|
|
assert block.entities == [block_entity]
|
|
|
|
def test_de_json_all(self, client, block_entity, data_with_type):
|
|
data, type = data_with_type
|
|
|
|
json_dict = {'id': self.id, 'type': type, 'type_for_from': self.type_for_from, 'title': self.title,
|
|
'entities': [block_entity.to_dict()], 'description': self.description, 'data': data.to_dict()}
|
|
block = Block.de_json(json_dict, client)
|
|
|
|
assert block.id == self.id
|
|
assert block.type == type
|
|
assert block.type_for_from == self.type_for_from
|
|
assert block.title == self.title
|
|
assert block.entities == [block_entity]
|
|
assert block.description == self.description
|
|
assert block.data == data
|
|
|
|
def test_equality(self, block_entity):
|
|
a = Block(self.id, self.type, self.type_for_from, self.title, [block_entity])
|
|
b = Block('', self.type, self.type_for_from, self.title, None)
|
|
c = Block(self.id, self.type, self.type_for_from, self.title, [block_entity])
|
|
|
|
assert a != b
|
|
assert hash(a) != hash(b)
|
|
assert a is not b
|
|
|
|
assert a == c
|