def test_get_status_from_db(self, label_patch): new_job = Job(1, self.connection) self.mock_row.status = 2 new_job.get_status_from_db(self.connection) self.assertEquals(new_job.status, 2)