Madiator2011/lbry-sdk
1
0
Fork 0
forked from LBRYCommunity/lbry-sdk
Code Pull requests Activity

test add/remove/list subscriptions

Browse source
This commit is contained in:
Victor Shyba 2021-10-15 03:48:03 -03:00
parent e449cd01ba
commit 4b88b191e7
3 changed files with 18 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
lbry/extras/daemon/daemon.py
View file

@ -3030,6 +3030,19 @@ class Daemon(metaclass=JSONRPCServerType):
}
return base58.b58encode(json.dumps(export, separators=(',', ':')))
@requires(WALLET_COMPONENT)
def jsonrpc_channel_subscription_list(self):
"""
List subscribed channels and modes.
Usage:
channel_subscription_list
Returns:
(list) [(channel_id, download_latest, download_all)]
"""
return self.storage.get_subscriptions()
@requires(WALLET_COMPONENT)
def jsonrpc_channel_subscribe(self, channel_id, download_latest=None, download_all=False):
"""

2
lbry/extras/daemon/storage.py
View file

@ -553,7 +553,7 @@ class SQLiteStorage(SQLiteMixin):
(channel_id, download_latest or 0, 1 if download_all else 0))
def remove_subscription(self, channel_id):
return self.db.execute_fetchall("delete from subscriptions where channel_id=?", (channel_id,))
return self.db.execute_fetchall("delete from subscription where channel_id=?", (channel_id,))
def get_subscriptions(self):
return self.db.execute_fetchall("select channel_id, download_latest, download_all from subscription")

4
tests/integration/datanetwork/test_file_commands.py
View file

@ -618,3 +618,7 @@ class TestProactiveDownloaderComponent(CommandTestCase):
await proactive_downloader.start()
await proactive_downloader.finished_iteration.wait()
await self.assertFileList(content1, content2)
self.assertEqual([(channel_id, 0, 1)], await self.daemon.jsonrpc_channel_subscription_list())
await self.daemon.jsonrpc_channel_unsubscribe(channel_id)
self.assertEqual([], await self.daemon.jsonrpc_channel_subscription_list())