!728 Add ut testcase for debugger

Merge pull request !728 from maning202007/ut_debugger

tests/ut/debugger/stream_handler/test_metadata_handler.py

@@ -0,0 +1,96 @@
+# Copyright 2020 Huawei Technologies Co., Ltd
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ============================================================================
+"""Test metadata_handler.py."""
+from mindinsight.debugger.common.utils import ServerStatus
+from mindinsight.debugger.stream_handler.metadata_handler import MetadataHandler
+from mindinsight.debugger.proto.debug_grpc_pb2 import Metadata
+
+class TestMetadataHandler:
+    """test class for MetadataHandler"""
+    def setup_method(self):
+        """set up function, init a MetaDataHandler for use"""
+        self.metadata_handler = MetadataHandler()
+
+    def test_device_name(self):
+        """test device_name property"""
+        res = self.metadata_handler.device_name
+        assert res == ""
+
+    def test_step(self):
+        """test step property."""
+        res = self.metadata_handler.step
+        assert res == 0
+
+    def test_node_name(self):
+        """test property of current node name"""
+        res = self.metadata_handler.node_name
+        assert res == ""
+
+    def test_node_name_setter(self):
+        """test set property of node name"""
+        self.metadata_handler.node_name = "node_name"
+        res = self.metadata_handler.node_name
+        assert res == "node_name"
+
+    def test_full_name(self):
+        """test full_name property"""
+        res = self.metadata_handler.full_name
+        assert res == ""
+
+    def test_backend(self):
+        """test backend property"""
+        res = self.metadata_handler.backend
+        assert res == ""
+
+    def test_state(self):
+        """test state property"""
+        res = self.metadata_handler.state
+        assert res == ServerStatus.PENDING.value
+
+    def test_state_setter(self):
+        """test set the property of state"""
+        self.metadata_handler.state = ServerStatus.RUNNING
+        res = self.metadata_handler.state
+        assert res == ServerStatus.RUNNING.value
+
+    def test_client_ip(self):
+        """test client_ip property"""
+        res = self.metadata_handler.client_ip
+        assert res == ""
+
+    def test_client_ip_setter(self):
+        """test Set the property of client ip"""
+        self.metadata_handler.client_ip = "0.0.0.0"
+        res = self.metadata_handler.client_ip
+        assert res == "0.0.0.0"
+
+    def test_put(self):
+        """test put value into metadata cache"""
+        value = Metadata()
+        self.metadata_handler.put(value)
+        res_device_name = self.metadata_handler.device_name
+        assert res_device_name == value.device_name.split(':')[0]
+        res_step = self.metadata_handler.step
+        assert res_step == value.cur_step
+        res_cur_node = self.metadata_handler.full_name
+        assert res_cur_node == value.cur_node
+        res_backend = self.metadata_handler.backend
+        expect_backend = value.backend if value.backend else "Ascend"
+        assert res_backend == expect_backend
+
+    def test_get(self):
+        """test get updated value"""
+        res = self.metadata_handler.get('state')
+        assert res == {'metadata': {'state': ServerStatus.PENDING.value}}

tests/ut/debugger/stream_handler/test_tensor_handler.py

@@ -0,0 +1,86 @@
+# Copyright 2020 Huawei Technologies Co., Ltd
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ============================================================================
+"""Test tensor_handler.py"""
+from unittest import mock
+from unittest.mock import MagicMock
+import pytest
+
+from mindinsight.debugger.common.exceptions.exceptions import DebuggerParamValueError
+from mindinsight.debugger.common.log import logger as log
+from mindinsight.debugger.stream_handler.tensor_handler import TensorHandler
+
+
+class TestTensorHandler:
+    """Test TensorHandler."""
+
+    def setup_method(self):
+        self.tensor_handler = TensorHandler()
+
+    @mock.patch.object(TensorHandler, '_get_tensor')
+    @mock.patch.object(log, "error")
+    @pytest.mark.parametrize("filter_condition", {})
+    def test_get(self, mock_get_tensor, mock_error, filter_condition):
+        """
+        Test get full tensor value.
+        """
+        mock_get_tensor.return_value = None
+        mock_error.return_value = None
+        with pytest.raises(DebuggerParamValueError) as ex:
+            self.tensor_handler.get(filter_condition)
+        assert "No tensor named {}".format(None) in str(ex.value)
+
+    @mock.patch.object(TensorHandler, '_get_prev_tensor_value_status')
+    @pytest.mark.parametrize(
+        "node_type, tensor_name, tensor_info", [('Parameter', 'name', {'full_name': 'name', 'step': 1})])
+    def test_update_has_prev_step_field(self, mock_get_pre, node_type, tensor_name, tensor_info):
+        """Test update has_prev_step field in tensor info."""
+        mock_get_pre.return_value = True
+        res = self.tensor_handler._update_has_prev_step_field(tensor_info, tensor_name, node_type)
+        assert res
+
+    def test_get_prev_tensor_value_status_none(self):
+        """
+        test _get_prev_tensor_value_status.
+        """
+        res = self.tensor_handler._get_prev_tensor_value_status('tensor_name')
+        assert res is None
+
+    @mock.patch.object(TensorHandler, '_get_tensor')
+    def test_get_prev_tensor_value_status_false(self, mock_get_tensor):
+        """
+        test _get_prev_tensor_value_status.
+        """
+        self.tensor_handler._cur_step = 1
+        mock_tensor = MagicMock()
+        mock_tensor.value = None
+        mock_get_tensor.return_value = mock_tensor
+        res = self.tensor_handler._get_prev_tensor_value_status('tensor_name')
+        assert not res
+
+    def test_get_tensor_value_by_name_none(self):
+        """Test get_tensor_value_by_name."""
+        res = self.tensor_handler.get_tensor_value_by_name('tensor_name', True)
+        assert res is None
+
+    @mock.patch.object(log, "error")
+    @pytest.mark.parametrize("tensor_name", "name")
+    def test_get_tensors_diff_error(self, mock_error, tensor_name):
+        """
+            Test get_tensors_diff.
+        """
+        mock_error.return_value = None
+        with pytest.raises(DebuggerParamValueError) as ex:
+            self.tensor_handler.get_tensors_diff(tensor_name, {1, 1})
+        assert f"Get current step and previous step for this tensor name {tensor_name} failed." in str(ex.value)