test_base_model.py

#!/usr/bin/python3
""" Module of Unittests """
import os
import unittest
import datetime
from models.base_model import BaseModel
import os
from models import storage
from models.engine.file_storage import FileStorage


class BaseModelTests(unittest.TestCase):

    my_model = BaseModel()

    def testBaseModel1(self):
        """ This Tests attribute values of a BaseModel instance """

        self.my_model.name = "myFirstModel"
        self.my_model.my_number = 89
        self.my_model.save()
        my_model_json = self.my_model.to_dict()

        self.assertEqual(self.my_model.name, my_model_json['name'])
        self.assertEqual(self.my_model.my_number, my_model_json['my_number'])
        self.assertEqual('BaseModel', my_model_json['__class__'])
        self.assertEqual(self.my_model.id, my_model_json['id'])

    def testSave(self):
        """ This tests Checks if save method updates the public instance
        attribute updated_at """

        self.my_model.first_name = "First"
        self.my_model.save()

        self.assertIsInstance(self.my_model.id, str)
        self.assertIsInstance(self.my_model.created_at, datetime.datetime)
        self.assertIsInstance(self.my_model.updated_at, datetime.datetime)

        first_dict = self.my_model.to_dict()

        self.my_model.first_name = "Second"
        self.my_model.save()
        sec_dict = self.my_model.to_dict()

        self.assertEqual(first_dict['created_at'], sec_dict['created_at'])
        self.assertNotEqual(first_dict['updated_at'], sec_dict['updated_at'])


if __name__ == '__main__':
    unittest.main()