def test_to_dic(self):
'''Test the correct output of the 'to_dic' method'''
test3 = BaseModel()
dic = test3.to_dict()
self.assertEqual(len(dic), 4)
test3.name = 'John'
dic = test3.to_dict()
self.assertEqual(len(dic), 5)
test3.last_name = 'Wick'
dic = test3.to_dict()
self.assertEqual(len(dic), 6)
for i in dic.values():
self.assertEqual(type(i), str)
def testUpdateExtraData(self):
"""Tests the update command with extra arguments"""
b1 = BaseModel()
b1.first_name = "Ben"
b1.last_name = "Keener"
with unittest.mock.patch("sys.stdout", new=StringIO()) as f:
HBNBCommand().onecmd("update BaseModel {} first_name \"Drew\"\
last_name \"foo\"".format(b1.id))
self.assertEqual(f.getvalue().rstrip(), "")
self.assertEqual(b1.first_name, "Drew")
self.assertEqual(b1.last_name, "Keener")
