wip command_interface test

Author: maxd-nordic

src/nrfcredstore/command_interface.py

@@ -75,11 +75,6 @@ def get_mfw_version(self):
         """Get modem firmware version, if applicable"""
         return
 
-class ATKeygenException(Exception):
-    def __init__(self, message, exit_code):
-        super().__init__(message)
-        self.exit_code = exit_code
-
 class ATCommandInterface(CredentialCommandInterface):
     shell = False
 

tests/test_command_interface.py

@@ -0,0 +1,23 @@
+from unittest.mock import patch, Mock
+from collections import namedtuple
+import pytest
+
+from nrfcredstore.command_interface import CredentialCommandInterface, ATCommandInterface, TLSCredShellInterface
+
+@pytest.fixture
+def comms():
+    """Mock comms object"""
+    comms = Mock()
+    return comms
+
+@pytest.fixture
+def at_command_interface(comms):
+    """Mock command interface"""
+    command_interface = ATCommandInterface(comms)
+    return command_interface
+
+def test_write_raw(at_command_interface):
+    """Test writing raw AT commands"""
+    at_command_interface.comms.write_line = Mock()
+    at_command_interface.write_raw('AT+TEST=1')
+    at_command_interface.comms.write_line.assert_called_once_with('AT+TEST=1')