forked from LBRYCommunity/lbry-sdk
refactored sorting of claims and unit tests
This commit is contained in:
Sergey Rozhnov
jleute
parent
156b132070
commit
c290fb5908
|
|
@ -36,7 +36,6 @@ from lbrynet.daemon.Downloader import GetStream
|
|||
from lbrynet.daemon.Publisher import Publisher
|
||||
from lbrynet.daemon.ExchangeRateManager import ExchangeRateManager
|
||||
from lbrynet.daemon.auth.server import AuthJSONRPCServer
|
||||
from lbrynet.daemon.claims_comparator import arrange_results
|
||||
from lbrynet.core.PaymentRateManager import OnlyFreePaymentsManager
|
||||
from lbrynet.core import utils, system_info
|
||||
from lbrynet.core.StreamDescriptor import StreamDescriptorIdentifier, download_sd_blob
|
||||
|
|
|
|||
|
|
@ -1,10 +0,0 @@
|
|||
def arrange_results(claims):
|
||||
for claim in claims:
|
||||
results = claim['result']
|
||||
sorted_results = sorted(results, key=lambda d: (d['height'], d['name'], d['claim_id'], _outpoint(d)))
|
||||
claim['result'] = sorted_results
|
||||
return claims
|
||||
|
||||
|
||||
def _outpoint(claim):
|
||||
return '{}:{}'.format(claim['txid'], claim['nout'])
|
||||
|
|
@ -1,200 +0,0 @@
|
|||
{
|
||||
"cases": [
|
||||
{
|
||||
"description": "sort by claim_id",
|
||||
"results": [
|
||||
{
|
||||
"height": 1,
|
||||
"name": "res",
|
||||
"claim_id": "ccc",
|
||||
"nout": 0,
|
||||
"txid": "fdsafa"
|
||||
},
|
||||
{
|
||||
"height": 1,
|
||||
"name": "res",
|
||||
"claim_id": "aaa",
|
||||
"nout": 0,
|
||||
"txid": "w5tv8uorgt"
|
||||
},
|
||||
{
|
||||
"height": 1,
|
||||
"name": "res",
|
||||
"claim_id": "bbb",
|
||||
"nout": 0,
|
||||
"txid": "aecfaewcfa"
|
||||
}
|
||||
],
|
||||
"expected": [
|
||||
{
|
||||
"height": 1,
|
||||
"name": "res",
|
||||
"claim_id": "aaa",
|
||||
"nout": 0,
|
||||
"txid": "w5tv8uorgt"
|
||||
},
|
||||
{
|
||||
"height": 1,
|
||||
"name": "res",
|
||||
"claim_id": "bbb",
|
||||
"nout": 0,
|
||||
"txid": "aecfaewcfa"
|
||||
},
|
||||
{
|
||||
"height": 1,
|
||||
"name": "res",
|
||||
"claim_id": "ccc",
|
||||
"nout": 0,
|
||||
"txid": "fdsafa"
|
||||
}
|
||||
]
|
||||
},
|
||||
{
|
||||
"description": "sort by height",
|
||||
"results": [
|
||||
{
|
||||
"height": 1,
|
||||
"name": "res",
|
||||
"claim_id": "ccc",
|
||||
"nout": 0,
|
||||
"txid": "aecfaewcfa"
|
||||
},
|
||||
{
|
||||
"height": 3,
|
||||
"name": "res",
|
||||
"claim_id": "ccc",
|
||||
"nout": 0,
|
||||
"txid": "aecfaewcfa"
|
||||
},
|
||||
{
|
||||
"height": 2,
|
||||
"name": "res",
|
||||
"claim_id": "ccc",
|
||||
"nout": 0,
|
||||
"txid": "aecfaewcfa"
|
||||
}
|
||||
],
|
||||
"expected": [
|
||||
{
|
||||
"claim_id": "ccc",
|
||||
"height": 1,
|
||||
"name": "res",
|
||||
"nout": 0,
|
||||
"txid": "aecfaewcfa"
|
||||
},
|
||||
{
|
||||
"claim_id": "ccc",
|
||||
"height": 2,
|
||||
"name": "res",
|
||||
"nout": 0,
|
||||
"txid": "aecfaewcfa"
|
||||
},
|
||||
{
|
||||
"claim_id": "ccc",
|
||||
"height": 3,
|
||||
"name": "res",
|
||||
"nout": 0,
|
||||
"txid": "aecfaewcfa"
|
||||
}
|
||||
]
|
||||
},
|
||||
{
|
||||
"description": "sort by name",
|
||||
"results": [
|
||||
{
|
||||
"height": 1,
|
||||
"name": "res1",
|
||||
"claim_id": "ccc",
|
||||
"nout": 0,
|
||||
"txid": "aecfaewcfa"
|
||||
},
|
||||
{
|
||||
"height": 1,
|
||||
"name": "res3",
|
||||
"claim_id": "ccc",
|
||||
"nout": 0,
|
||||
"txid": "aecfaewcfa"
|
||||
},
|
||||
{
|
||||
"height": 1,
|
||||
"name": "res2",
|
||||
"claim_id": "ccc",
|
||||
"nout": 0,
|
||||
"txid": "aecfaewcfa"
|
||||
}
|
||||
],
|
||||
"expected": [
|
||||
{
|
||||
"height": 1,
|
||||
"name": "res1",
|
||||
"claim_id": "ccc",
|
||||
"nout": 0,
|
||||
"txid": "aecfaewcfa"
|
||||
},
|
||||
{
|
||||
"height": 1,
|
||||
"name": "res2",
|
||||
"claim_id": "ccc",
|
||||
"nout": 0,
|
||||
"txid": "aecfaewcfa"
|
||||
},
|
||||
{
|
||||
"height": 1,
|
||||
"name": "res3",
|
||||
"claim_id": "ccc",
|
||||
"nout": 0,
|
||||
"txid": "aecfaewcfa"
|
||||
}
|
||||
]
|
||||
},
|
||||
{
|
||||
"description": "sort by outpoint",
|
||||
"results": [
|
||||
{
|
||||
"height": 1,
|
||||
"name": "res1",
|
||||
"claim_id": "ccc",
|
||||
"nout": 2,
|
||||
"txid": "aecfaewcfa"
|
||||
},
|
||||
{
|
||||
"height": 1,
|
||||
"name": "res1",
|
||||
"claim_id": "ccc",
|
||||
"nout": 1,
|
||||
"txid": "aecfaewcfa"
|
||||
},
|
||||
{
|
||||
"height": 1,
|
||||
"name": "res1",
|
||||
"claim_id": "ccc",
|
||||
"nout": 3,
|
||||
"txid": "aecfaewcfa"
|
||||
}
|
||||
],
|
||||
"expected": [
|
||||
{
|
||||
"height": 1,
|
||||
"name": "res1",
|
||||
"claim_id": "ccc",
|
||||
"nout": 1,
|
||||
"txid": "aecfaewcfa"
|
||||
},
|
||||
{
|
||||
"height": 1,
|
||||
"name": "res1",
|
||||
"claim_id": "ccc",
|
||||
"nout": 2,
|
||||
"txid": "aecfaewcfa"
|
||||
},
|
||||
{
|
||||
"height": 1,
|
||||
"name": "res1",
|
||||
"claim_id": "ccc",
|
||||
"nout": 3,
|
||||
"txid": "aecfaewcfa"
|
||||
}
|
||||
]
|
||||
}
|
||||
]
|
||||
}
|
||||
|
|
@ -1,26 +1,213 @@
|
|||
import json
|
||||
import unittest
|
||||
|
||||
from lbrynet.daemon.claims_comparator import arrange_results
|
||||
from lbrynet.daemon.Daemon import arrange_results
|
||||
|
||||
|
||||
class ClaimsComparatorTest(unittest.TestCase):
|
||||
def setUp(self):
|
||||
with open('claims_comparator_cases.json') as f:
|
||||
document = json.load(f)
|
||||
self.cases = document['cases']
|
||||
def test_arrange_results_when_sorted_by_claim_id(self):
|
||||
self.run_test(
|
||||
[
|
||||
{
|
||||
"height": 1,
|
||||
"name": "res",
|
||||
"claim_id": "ccc",
|
||||
"nout": 0,
|
||||
"txid": "fdsafa"
|
||||
},
|
||||
{
|
||||
"height": 1,
|
||||
"name": "res",
|
||||
"claim_id": "aaa",
|
||||
"nout": 0,
|
||||
"txid": "w5tv8uorgt"
|
||||
},
|
||||
{
|
||||
"height": 1,
|
||||
"name": "res",
|
||||
"claim_id": "bbb",
|
||||
"nout": 0,
|
||||
"txid": "aecfaewcfa"
|
||||
}
|
||||
],
|
||||
[
|
||||
{
|
||||
"height": 1,
|
||||
"name": "res",
|
||||
"claim_id": "aaa",
|
||||
"nout": 0,
|
||||
"txid": "w5tv8uorgt"
|
||||
},
|
||||
{
|
||||
"height": 1,
|
||||
"name": "res",
|
||||
"claim_id": "bbb",
|
||||
"nout": 0,
|
||||
"txid": "aecfaewcfa"
|
||||
},
|
||||
{
|
||||
"height": 1,
|
||||
"name": "res",
|
||||
"claim_id": "ccc",
|
||||
"nout": 0,
|
||||
"txid": "fdsafa"
|
||||
}
|
||||
])
|
||||
|
||||
def test_arrange_results(self):
|
||||
for case in self.cases:
|
||||
results = case['results']
|
||||
data = {'result': results}
|
||||
expected = case['expected']
|
||||
def test_arrange_results_when_sorted_by_height(self):
|
||||
self.run_test(
|
||||
[
|
||||
{
|
||||
"height": 1,
|
||||
"name": "res",
|
||||
"claim_id": "ccc",
|
||||
"nout": 0,
|
||||
"txid": "aecfaewcfa"
|
||||
},
|
||||
{
|
||||
"height": 3,
|
||||
"name": "res",
|
||||
"claim_id": "ccc",
|
||||
"nout": 0,
|
||||
"txid": "aecfaewcfa"
|
||||
},
|
||||
{
|
||||
"height": 2,
|
||||
"name": "res",
|
||||
"claim_id": "ccc",
|
||||
"nout": 0,
|
||||
"txid": "aecfaewcfa"
|
||||
}
|
||||
],
|
||||
[
|
||||
{
|
||||
"claim_id": "ccc",
|
||||
"height": 1,
|
||||
"name": "res",
|
||||
"nout": 0,
|
||||
"txid": "aecfaewcfa"
|
||||
},
|
||||
{
|
||||
"claim_id": "ccc",
|
||||
"height": 2,
|
||||
"name": "res",
|
||||
"nout": 0,
|
||||
"txid": "aecfaewcfa"
|
||||
},
|
||||
{
|
||||
"claim_id": "ccc",
|
||||
"height": 3,
|
||||
"name": "res",
|
||||
"nout": 0,
|
||||
"txid": "aecfaewcfa"
|
||||
}
|
||||
])
|
||||
|
||||
claims = arrange_results([data])
|
||||
claim = claims[0]
|
||||
actual = claim['result']
|
||||
def test_arrange_results_when_sorted_by_name(self):
|
||||
self.run_test(
|
||||
[
|
||||
{
|
||||
"height": 1,
|
||||
"name": "res1",
|
||||
"claim_id": "ccc",
|
||||
"nout": 0,
|
||||
"txid": "aecfaewcfa"
|
||||
},
|
||||
{
|
||||
"height": 1,
|
||||
"name": "res3",
|
||||
"claim_id": "ccc",
|
||||
"nout": 0,
|
||||
"txid": "aecfaewcfa"
|
||||
},
|
||||
{
|
||||
"height": 1,
|
||||
"name": "res2",
|
||||
"claim_id": "ccc",
|
||||
"nout": 0,
|
||||
"txid": "aecfaewcfa"
|
||||
}
|
||||
],
|
||||
[
|
||||
{
|
||||
"height": 1,
|
||||
"name": "res1",
|
||||
"claim_id": "ccc",
|
||||
"nout": 0,
|
||||
"txid": "aecfaewcfa"
|
||||
},
|
||||
{
|
||||
"height": 1,
|
||||
"name": "res2",
|
||||
"claim_id": "ccc",
|
||||
"nout": 0,
|
||||
"txid": "aecfaewcfa"
|
||||
},
|
||||
{
|
||||
"height": 1,
|
||||
"name": "res3",
|
||||
"claim_id": "ccc",
|
||||
"nout": 0,
|
||||
"txid": "aecfaewcfa"
|
||||
}
|
||||
])
|
||||
|
||||
self.assertEqual(expected, actual, case['description'])
|
||||
def test_arrange_results_when_sort_by_outpoint(self):
|
||||
self.run_test(
|
||||
[
|
||||
{
|
||||
"height": 1,
|
||||
"name": "res1",
|
||||
"claim_id": "ccc",
|
||||
"nout": 2,
|
||||
"txid": "aecfaewcfa"
|
||||
},
|
||||
{
|
||||
"height": 1,
|
||||
"name": "res1",
|
||||
"claim_id": "ccc",
|
||||
"nout": 1,
|
||||
"txid": "aecfaewcfa"
|
||||
},
|
||||
{
|
||||
"height": 1,
|
||||
"name": "res1",
|
||||
"claim_id": "ccc",
|
||||
"nout": 3,
|
||||
"txid": "aecfaewcfa"
|
||||
}
|
||||
],
|
||||
[
|
||||
{
|
||||
"height": 1,
|
||||
"name": "res1",
|
||||
"claim_id": "ccc",
|
||||
"nout": 1,
|
||||
"txid": "aecfaewcfa"
|
||||
},
|
||||
{
|
||||
"height": 1,
|
||||
"name": "res1",
|
||||
"claim_id": "ccc",
|
||||
"nout": 2,
|
||||
"txid": "aecfaewcfa"
|
||||
},
|
||||
{
|
||||
"height": 1,
|
||||
"name": "res1",
|
||||
"claim_id": "ccc",
|
||||
"nout": 3,
|
||||
"txid": "aecfaewcfa"
|
||||
}
|
||||
])
|
||||
|
||||
def run_test(self, results, expected):
|
||||
data = {'result': results}
|
||||
|
||||
claims = arrange_results([data])
|
||||
claim = claims[0]
|
||||
actual = claim['result']
|
||||
|
||||
self.assertEqual(expected, actual)
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
|
|
|
|||
Loading…
Reference in a new issue