commit 78f23b4f151ce102a8cc0c0b2387d72a5cebd4d9
author Treehugger Robot <treehugger-gerrit@google.com> Fri Sep 20 01:04:17 2019 +0000
committer Gerrit Code Review <noreply-gerritcodereview@google.com> Fri Sep 20 01:04:17 2019 +0000
tree 5c0933bc24287ddaca65dd4bd2ccb2133ea7a39a
parent 63f954d965cd0e30cdcc6c83fb4b254867e3379f
parent cda4deef01244fd641d637f2f9523352c7a5d746

Merge "AIDEGen: fix module_info_util_unittest failed."

aidegen/lib/module_info_util_unittest.py

diff --git a/aidegen/lib/module_info_util_unittest.py b/aidegen/lib/module_info_util_unittest.py
index 24260e1..9ed09cb 100644
--- a/aidegen/lib/module_info_util_unittest.py
+++ b/aidegen/lib/module_info_util_unittest.py
@@ -27,7 +27,6 @@
 from aidegen.lib import module_info_util
 
 from atest import atest_utils
-from atest import module_info
 
 _TEST_CLASS_DICT = {'class': ['JAVA_LIBRARIES']}
 _TEST_SRCS_BAR_DICT = {'srcs': ['Bar']}
@@ -136,9 +135,9 @@
     @mock.patch('os.path.isfile')
     def test_build_bp_info_normal(self, mock_isfile, mock_build, mock_time):
         """Test _build_target with verbose true and false."""
-        mock_isfile.return_value = True
-        amodule_info = module_info.ModuleInfo()
+        amodule_info = mock.MagicMock()
         skip = True
+        mock_isfile.return_value = True
         mock_build.return_value = True
         module_info_util._build_bp_info(amodule_info, unittest_constants.
                                         TEST_MODULE, False, skip)