def test_create(self): department = Department() department.Name = self.name department.save(qb=self.qb_client) query_department = Department.get(department.Id, qb=self.qb_client) self.assertEqual(department.Id, query_department.Id) self.assertEqual(query_department.Name, self.name)
def test_update(self): department = Department.all(max_results=1, qb=self.qb_client)[0] unique_name = "Test Department {0}".format(datetime.now().strftime('%d%H%M%S')) department.Name = unique_name department.save(qb=self.qb_client) query_department = Department.get(department.Id, qb=self.qb_client) self.assertEqual(query_department.Name, unique_name)
def test_get_with_qb(self): with patch.object(self.qb_client, 'get_single_object') as get_single_object: Department.get(1, qb=self.qb_client) self.assertTrue(get_single_object.called)
def test_get(self, get_single_object): Department.get(1) get_single_object.assert_called_once_with("Department", pk=1)