test update for white speech bubble

freedomofpress · Jul 31, 2019 · b2a1650 · b2a1650
1 parent f605b40
commit b2a1650
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/tests/gui/test_widgets.py b/tests/gui/test_widgets.py
@@ -1101,9 +1101,12 @@ def test_SpeechBubble_init(mocker):
     mock_connect = mocker.Mock()
     mock_signal.connect = mock_connect
 
-    SpeechBubble('mock id', 'hello', mock_signal)
+    sb = SpeechBubble('mock id', 'hello', mock_signal)
+    ss = sb.styleSheet()
+
     mock_label.assert_called_once_with('hello')
     assert mock_connect.called
+    assert 'background-color' in ss
 
 
 def test_SpeechBubble_update_text(mocker):
@@ -1187,9 +1190,7 @@ def test_MessageWidget_init(mocker):
     mock_signal.connect = mock_connected
 
     mw = MessageWidget('mock id', 'hello', mock_signal)
-    ss = mw.styleSheet()
 
-    assert 'background-color' in ss
     assert mock_connected.called