def case_update_record(self, refresh_db_before): """ Create test record in ServerType table and then update it.""" original_type = "TypeOrig" update_type = "TypeUpdate" original_type_obj = ServerTypeOp.add(original_type) updated_type_local = ServerTypeOp.update(original_type_obj, update_type) self.assertEqual(updated_type_local.name, update_type) updated_type_get = ServerTypeOp.get(name=update_type) self.assertTrue(len(updated_type_get) is 1) self.assertEqual(updated_type_get[0].name, update_type) self.assertEqual(updated_type_get[0].id, original_type_obj.id)
def test_update(self, mock_servtype, mock_val_name, mock_db): """ Assumptions: - update method run """ new_name = "TestName" ip_obj = mock_servtype() updated_ip = ServerTypeOp.update(ip_obj, new_name) self.assertTrue(mock_val_name.called) db_exp_calls = [ mock.call.session.add(updated_ip), mock.call.session.commit() ] mock_db.assert_has_calls(db_exp_calls)