Add an obvious test for the 'statusBar' command

and remove some junk introduced in 9a194f0.
This commit is contained in:
Dmitri Bogomolov 2021-12-11 20:01:10 +02:00
parent a44ce07949
commit 22d9387943
Signed by untrusted user: g1itch
GPG Key ID: 720A756F18DEED13
3 changed files with 25 additions and 30 deletions

View File

@ -67,7 +67,7 @@ import time
from binascii import hexlify, unhexlify
from struct import pack
from six.moves import configparser, http_client, queue, xmlrpc_server
from six.moves import configparser, http_client, xmlrpc_server
import defaults
import helper_inbox
@ -1455,25 +1455,11 @@ class BMRPCDispatcher(object):
"""Test two numeric params"""
return a + b
@testmode('clearUISignalQueue')
def HandleclearUISignalQueue(self):
"""clear UISignalQueue"""
queues.UISignalQueue.queue.clear()
return "success"
@command('statusBar')
def HandleStatusBar(self, message):
"""Update GUI statusbar message"""
queues.UISignalQueue.put(('updateStatusBar', message))
@testmode('getStatusBar')
def HandleGetStatusBar(self):
"""Get GUI statusbar message"""
try:
_, data = queues.UISignalQueue.get(block=False)
except queue.Empty:
return None
return data
return "success"
@testmode('undeleteMessage')
def HandleUndeleteMessage(self, msgid):

View File

@ -12,7 +12,7 @@ from six.moves import xmlrpc_client # nosec
import psutil
from .samples import (
sample_seed, sample_deterministic_addr3, sample_deterministic_addr4, sample_statusbar_msg,
sample_seed, sample_deterministic_addr3, sample_deterministic_addr4,
sample_inbox_msg_ids, sample_test_subscription_address, sample_subscription_name)
from .test_process import TestProcessProto
@ -86,18 +86,6 @@ class TestAPI(TestAPIProto):
'API Error 0020: Invalid method: test'
)
def test_statusbar_method(self):
"""Test statusbar method"""
self.api.clearUISignalQueue()
self.assertEqual(
self.api.statusBar(sample_statusbar_msg),
'null'
)
self.assertEqual(
self.api.getStatusBar(),
sample_statusbar_msg
)
def test_message_inbox(self):
"""Test message inbox methods"""
self.assertEqual(

View File

@ -1,10 +1,12 @@
import time
import unittest
from six.moves import xmlrpc_client
from six.moves import queue, xmlrpc_client
from pybitmessage import pathmagic
from .samples import sample_statusbar_msg # any
class TestAPIThread(unittest.TestCase):
"""Test case running the API thread"""
@ -15,16 +17,22 @@ class TestAPIThread(unittest.TestCase):
import helper_sql
import helper_startup
import queues
import state
from bmconfigparser import BMConfigParser
# pylint: disable=too-few-public-methods
class SqlReadyMock(object):
"""Mock helper_sql.sql_ready event with dummy class"""
@staticmethod
def wait():
"""Don't wait, return immediately"""
return
helper_sql.sql_ready = SqlReadyMock
cls.state = state
cls.queues = queues
helper_startup.loadConfig()
# helper_startup.fixSocket()
config = BMConfigParser()
@ -48,6 +56,19 @@ class TestAPIThread(unittest.TestCase):
self.assertEqual(
self.api.helloWorld('hello', 'world'), 'hello-world')
def test_statusbar(self):
"""Check UISignalQueue after issuing the 'statusBar' command"""
self.queues.UISignalQueue.queue.clear()
self.assertEqual(
self.api.statusBar(sample_statusbar_msg), 'success')
try:
cmd, data = self.queues.UISignalQueue.get(block=False)
except queue.Empty:
self.fail('UISignalQueue is empty!')
self.assertEqual(cmd, 'updateStatusBar')
self.assertEqual(data, sample_statusbar_msg)
@classmethod
def tearDownClass(cls):
cls.state.shutdown = 1