1 files changed, 45 insertions, 7 deletions
diff --git a/test/mitmproxy/contentviews/test_protobuf.py b/test/mitmproxy/contentviews/test_protobuf.py
index 1224b8db..31e382ec 100644
--- a/test/mitmproxy/contentviews/test_protobuf.py
+++ b/test/mitmproxy/contentviews/test_protobuf.py
@@ -1,12 +1,50 @@
+from unittest import mock
+import pytest
+
 from mitmproxy.contentviews import protobuf
 from mitmproxy.test import tutils
 from . import full_eval
 
-if protobuf.ViewProtobuf.is_available():
-    def test_view_protobuf_request():
-        v = full_eval(protobuf.ViewProtobuf())
 
-        p = tutils.test_data.path("mitmproxy/data/protobuf01")
-        content_type, output = v(open(p, "rb").read())
-        assert content_type == "Protobuf"
-        assert output.next()[0][1] == '1: "3bbc333c-e61c-433b-819a-0b9a8cc103b8"'
+def test_view_protobuf_request():
+    v = full_eval(protobuf.ViewProtobuf())
+    p = tutils.test_data.path("mitmproxy/data/protobuf01")
+
+    with mock.patch('mitmproxy.contentviews.protobuf.ViewProtobuf.is_available'):
+        with mock.patch('subprocess.Popen') as n:
+            m = mock.Mock()
+            attrs = {'communicate.return_value': (b'1: "3bbc333c-e61c-433b-819a-0b9a8cc103b8"', True)}
+            m.configure_mock(**attrs)
+            n.return_value = m
+
+            content_type, output = v(open(p, "rb").read())
+            assert content_type == "Protobuf"
+            assert output[0] == [('text', b'1: "3bbc333c-e61c-433b-819a-0b9a8cc103b8"')]
+
+            m.communicate = mock.MagicMock()
+            m.communicate.return_value = (None, None)
+            with pytest.raises(ValueError, matches="Failed to parse input."):
+                v(b'foobar')
+
+
+def test_view_protobuf_availability():
+    with mock.patch('subprocess.Popen') as n:
+        m = mock.Mock()
+        attrs = {'communicate.return_value': (b'libprotoc fake version', True)}
+        m.configure_mock(**attrs)
+        n.return_value = m
+        assert protobuf.ViewProtobuf().is_available()
+
+        m = mock.Mock()
+        attrs = {'communicate.return_value': (b'command not found', True)}
+        m.configure_mock(**attrs)
+        n.return_value = m
+        assert not protobuf.ViewProtobuf().is_available()
+
+
+def test_view_protobuf_fallback():
+    with mock.patch('subprocess.Popen.communicate') as m:
+        m.side_effect = OSError()
+        v = full_eval(protobuf.ViewProtobuf())
+        with pytest.raises(NotImplementedError, matches='protoc not found'):
+            v(b'foobar')