def test_user_turn_1(self):
mock_dbutil = mock.MagicMock()
mock_dbutil.fetch_optimal_word_from_dictionary.return_value = False
with mock.patch("main.DBUtil", return_value=mock_dbutil):
wordchain = WordChain("test_dictionary.txt")
ret = wordchain.user_turn()
self.assertEqual(ret, ["", 1])
def test_user_turn_2(self):
mock_dbutil = mock.MagicMock()
mock_dbutil.fetch_optimal_word_from_dictionary.return_value = "hash"
mock_dbutil.confirm_word_in_dictionary.side_effect = [False, "hello"]
mock_dbutil.compare_start_and_end.return_value = False
with mock.patch("main.DBUtil", return_value=mock_dbutil), \
mock.patch("main.WordChain.user_input", return_value="hello"):
wordchain = WordChain("test_dictionary.txt")
ret = wordchain.user_turn()
self.assertEqual(ret, ["hello", 2])