53 lines
1.2 KiB
Python
53 lines
1.2 KiB
Python
|
import unittest
|
||
|
|
|
||
|
|
from lbrynet.daemon.claims_comparator import arrange_results
|
||
|
|
|
||
|
|
|
||
|
|
class ClaimsComparatorTest(unittest.TestCase):
|
||
|
|
def test_arrange_results(self):
|
||
|
|
results = [
|
||
|
|
{
|
||
|
|
'height': 1,
|
||
|
|
'name': 'res',
|
||
|
|
'claim_id': 'ccc'
|
||
|
|
},
|
||
|
|
{
|
||
|
|
'height': 1,
|
||
|
|
'name': 'res',
|
||
|
|
'claim_id': 'aaa'
|
||
|
|
},
|
||
|
|
{
|
||
|
|
'height': 1,
|
||
|
|
'name': 'res',
|
||
|
|
'claim_id': 'bbb'
|
||
|
|
}
|
||
|
|
]
|
||
|
|
data = {'result': results}
|
||
|
|
|
||
|
|
expected = [
|
||
|
|
{
|
||
|
|
'height': 1,
|
||
|
|
'name': 'res',
|
||
|
|
'claim_id': 'aaa'
|
||
|
|
},
|
||
|
|
{
|
||
|
|
'height': 1,
|
||
|
|
'name': 'res',
|
||
|
|
'claim_id': 'bbb'
|
||
|
|
},
|
||
|
|
{
|
||
|
|
'height': 1,
|
||
|
|
'name': 'res',
|
||
|
|
'claim_id': 'ccc'
|
||
|
|
}
|
||
|
|
]
|
||
|
|
claims = arrange_results([data])
|
||
|
|
claim = claims[0]
|
||
|
|
actual = claim['result']
|
||
|
|
|
||
|
|
self.assertEqual(expected, actual)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
unittest.main()
|