コード例 #1
0
class TestCompany(unittest.TestCase):
    def setUp(self):
        self.info = {
            "uid": "",
            "name": "Test Company",
            "address": "new york, NY",
            "website": "www.testing.com",
            "phone": "555-123-4567"}
        self.company = Company(
            self.info["uid"],
            self.info["name"],
            self.info["address"],
            self.info["website"],
            self.info["phone"],
            testing=True
        )
        self.new_uid = "6"
        self.new_name = "Testing!!!"
        self.total_num_of_companies = 5

        #Reset the database
        db = SQL(TEST_DB)
        db.create_db()
        db.add_test_data_to_db()

    def test_properties(self):
        self.assertEqual(self.company.uid, self.info["uid"])
        self.assertEqual(self.company.name, self.info["name"])
        self.assertEqual(self.company.address, self.info["address"])
        self.assertEqual(self.company.website, self.info["website"])
        self.assertEqual(self.company.phone, self.info["phone"])

    def test_get_all_companies(self):
        self.assertEqual(5, len(self.company.get_all_companies()))

    def test_get_a_company(self):
        data = self.company.get_a_company(1)[0]
        self.assertEqual("Google", data.name)

    def test_add_company_to_db(self):
        self.company.add_company_to_db()
        self.assertTrue(self.company.uid)

    def test_update_company_in_db(self):
        test_class = self.company.get_a_company(1)[0]
        test_class.name = self.new_name
        test_class._testing = True
        test_class.update_company_in_db()
        test_class = self.company.get_a_company(1)[0]
        self.assertEqual(test_class.name, self.new_name)

    def test_delete_company_in_db(self):
        test_class = self.company.get_a_company(1)[0]
        test_class._testing = True
        test_class.delete_company_in_db()
        list_of_companies = self.company.get_all_companies()
        self.assertEqual(len(list_of_companies), self.total_num_of_companies -1)
コード例 #2
0
    def setUp(self):
        self.info = {
            "uid": "",
            "name": "Test Company",
            "address": "new york, NY",
            "website": "www.testing.com",
            "phone": "555-123-4567"}
        self.company = Company(
            self.info["uid"],
            self.info["name"],
            self.info["address"],
            self.info["website"],
            self.info["phone"],
            testing=True
        )
        self.new_uid = "6"
        self.new_name = "Testing!!!"
        self.total_num_of_companies = 5

        #Reset the database
        db = SQL(TEST_DB)
        db.create_db()
        db.add_test_data_to_db()