New tests.
[mono.git] / mcs / class / System.Web / System.Web.UI.WebControls / GridView.cs
index cefa1ab8d8ee142597ce3d147aac199e9852bb88..26ace72c5ce04e6ab91bba0de75cd5e2295ab890 100644 (file)
@@ -46,7 +46,7 @@ namespace System.Web.UI.WebControls
        [DefaultEventAttribute ("SelectedIndexChanged")]
        [AspNetHostingPermissionAttribute (SecurityAction.LinkDemand, Level = AspNetHostingPermissionLevel.Minimal)]
        [AspNetHostingPermissionAttribute (SecurityAction.InheritanceDemand, Level = AspNetHostingPermissionLevel.Minimal)]
-       public class GridView: CompositeDataBoundControl, ICallbackEventHandler, ICallbackContainer, IPostBackEventHandler, IPostBackContainer
+       public class GridView: CompositeDataBoundControl, ICallbackEventHandler, ICallbackContainer, IPostBackEventHandler, IPostBackContainer, IPersistedSelector
        {
                Table table;
                GridViewRowCollection rows;
@@ -80,22 +80,23 @@ namespace System.Web.UI.WebControls
                AutoGeneratedFieldProperties[] autoFieldProperties;
                string [] dataKeyNames = null;
                readonly string[] emptyKeys = new string[0];
-               
-               private static readonly object PageIndexChangedEvent = new object();
-               private static readonly object PageIndexChangingEvent = new object();
-               private static readonly object RowCancelingEditEvent = new object();
-               private static readonly object RowCommandEvent = new object();
-               private static readonly object RowCreatedEvent = new object();
-               private static readonly object RowDataBoundEvent = new object();
-               private static readonly object RowDeletedEvent = new object();
-               private static readonly object RowDeletingEvent = new object();
-               private static readonly object RowEditingEvent = new object();
-               private static readonly object RowUpdatedEvent = new object();
-               private static readonly object RowUpdatingEvent = new object();
-               private static readonly object SelectedIndexChangedEvent = new object();
-               private static readonly object SelectedIndexChangingEvent = new object();
-               private static readonly object SortedEvent = new object();
-               private static readonly object SortingEvent = new object();
+               IEnumerator _dataEnumerator;
+               
+               static readonly object PageIndexChangedEvent = new object();
+               static readonly object PageIndexChangingEvent = new object();
+               static readonly object RowCancelingEditEvent = new object();
+               static readonly object RowCommandEvent = new object();
+               static readonly object RowCreatedEvent = new object();
+               static readonly object RowDataBoundEvent = new object();
+               static readonly object RowDeletedEvent = new object();
+               static readonly object RowDeletingEvent = new object();
+               static readonly object RowEditingEvent = new object();
+               static readonly object RowUpdatedEvent = new object();
+               static readonly object RowUpdatingEvent = new object();
+               static readonly object SelectedIndexChangedEvent = new object();
+               static readonly object SelectedIndexChangingEvent = new object();
+               static readonly object SortedEvent = new object();
+               static readonly object SortingEvent = new object();
                
                // Control state
                int pageIndex;
@@ -343,6 +344,8 @@ namespace System.Web.UI.WebControls
                                return false;
                        }
                        set {
+                               if (value == AllowPaging)
+                                       return;
                                ViewState ["AllowPaging"] = value;
                                RequireBinding ();
                        }
@@ -357,6 +360,8 @@ namespace System.Web.UI.WebControls
                                return false;
                        }
                        set {
+                               if (value == AllowSorting)
+                                       return;
                                ViewState ["AllowSorting"] = value;
                                RequireBinding ();
                        }
@@ -386,6 +391,8 @@ namespace System.Web.UI.WebControls
                                return false;
                        }
                        set {
+                               if (value == AutoGenerateEditButton)
+                                       return;
                                ViewState ["AutoGenerateEditButton"] = value;
                                RequireBinding ();
                        }
@@ -400,6 +407,8 @@ namespace System.Web.UI.WebControls
                                return false;
                        }
                        set {
+                               if (value == AutoGenerateDeleteButton)
+                                       return;
                                ViewState ["AutoGenerateDeleteButton"] = value;
                                RequireBinding ();
                        }
@@ -414,6 +423,8 @@ namespace System.Web.UI.WebControls
                                return false;
                        }
                        set {
+                               if (value == AutoGenerateSelectButton)
+                                       return;
                                ViewState ["AutoGenerateSelectButton"] = value;
                                RequireBinding ();
                        }
@@ -428,6 +439,8 @@ namespace System.Web.UI.WebControls
                                return true;
                        }
                        set {
+                               if (value == AutoGenerateColumns)
+                                       return;
                                ViewState ["AutoGenerateColumns"] = value;
                                RequireBinding ();
                        }
@@ -468,7 +481,6 @@ namespace System.Web.UI.WebControls
                        }
                        set {
                                ViewState ["Caption"] = value;
-                               RequireBinding ();
                        }
                }
                
@@ -483,7 +495,6 @@ namespace System.Web.UI.WebControls
                        }
                        set {
                                ViewState ["CaptionAlign"] = value;
-                               RequireBinding ();
                        }
                }
 
@@ -532,6 +543,12 @@ namespace System.Web.UI.WebControls
                        }
                }
 
+               [BrowsableAttribute(false)]
+               public IAutoFieldGenerator ColumnsGenerator {
+                       get;
+                       set;
+               }
+
                [DefaultValueAttribute (null)]
                [WebCategoryAttribute ("Data")]
                [TypeConverter (typeof(StringArrayConverter))]
@@ -630,7 +647,7 @@ namespace System.Web.UI.WebControls
                [Browsable (false)]
                public virtual ITemplate EmptyDataTemplate {
                        get { return emptyDataTemplate; }
-                       set { emptyDataTemplate = value; RequireBinding (); }
+                       set { emptyDataTemplate = value; }
                }
                
                [LocalizableAttribute (true)]
@@ -643,6 +660,8 @@ namespace System.Web.UI.WebControls
                                return string.Empty;
                        }
                        set {
+                               if (value == EmptyDataText)
+                                       return;
                                ViewState ["EmptyDataText"] = value;
                                RequireBinding ();
                        }
@@ -657,6 +676,8 @@ namespace System.Web.UI.WebControls
                                return false;
                        }
                        set {
+                               if (value == EnableSortingAndPagingCallbacks)
+                                       return;
                                ViewState ["EnableSortingAndPagingCallbacks"] = value;
                                RequireBinding ();
                        }
@@ -799,6 +820,8 @@ namespace System.Web.UI.WebControls
                                return 10;
                        }
                        set {
+                               if (value == PageSize)
+                                       return;
                                ViewState ["PageSize"] = value;
                                RequireBinding ();
                        }
@@ -842,7 +865,7 @@ namespace System.Web.UI.WebControls
                [Browsable (false)]
                public virtual ITemplate PagerTemplate {
                        get { return pagerTemplate; }
-                       set { pagerTemplate = value; RequireBinding (); }
+                       set { pagerTemplate = value; }
                }
                
                [DefaultValueAttribute ("")]
@@ -855,6 +878,8 @@ namespace System.Web.UI.WebControls
                                return string.Empty;
                        }
                        set {
+                               if (value == RowHeaderColumn)
+                                       return;
                                ViewState ["RowHeaderColumn"] = value;
                                RequireBinding ();
                        }
@@ -899,6 +924,18 @@ namespace System.Web.UI.WebControls
                                        return null;
                        }
                }
+
+               [MonoTODO]
+               [BrowsableAttribute(false)]
+               public virtual DataKey SelectedPersistedDataKey {
+                       get; set;
+               }
+
+               [MonoTODO]
+               DataKey IPersistedSelector.DataKey {
+                       get { return SelectedPersistedDataKey; }
+                       set { SelectedPersistedDataKey = value; }
+               }
                
                [BindableAttribute (true)]
                [DefaultValueAttribute (-1)]
@@ -964,6 +1001,8 @@ namespace System.Web.UI.WebControls
                                return false;
                        }
                        set {
+                               if (value == ShowFooter)
+                                       return;
                                ViewState ["ShowFooter"] = value;
                                RequireBinding ();
                        }
@@ -978,6 +1017,8 @@ namespace System.Web.UI.WebControls
                                return true;
                        }
                        set {
+                               if (value == ShowHeader)
+                                       return;
                                ViewState ["ShowHeader"] = value;
                                RequireBinding ();
                        }
@@ -1031,6 +1072,8 @@ namespace System.Web.UI.WebControls
                                return true;
                        }
                        set {
+                               if (value == UseAccessibleHeader)
+                                       return;
                                ViewState ["UseAccessibleHeader"] = value;
                                RequireBinding ();
                        }
@@ -1072,6 +1115,14 @@ namespace System.Web.UI.WebControls
                
                protected virtual ICollection CreateColumns (PagedDataSource dataSource, bool useDataSource)
                {
+                       bool autoGenerate = AutoGenerateColumns;
+
+                       if (autoGenerate) {
+                               IAutoFieldGenerator fieldGenerator = ColumnsGenerator;
+                               if (fieldGenerator != null)
+                                       return fieldGenerator.GenerateFields (this);
+                       }
+                       
                        ArrayList fields = new ArrayList ();
                        
                        if (AutoGenerateEditButton || AutoGenerateDeleteButton || AutoGenerateSelectButton) {
@@ -1084,7 +1135,7 @@ namespace System.Web.UI.WebControls
 
                        fields.AddRange (Columns);
                        
-                       if (AutoGenerateColumns) {
+                       if (autoGenerate) {
                                if (useDataSource)
                                        autoFieldProperties = CreateAutoFieldProperties (dataSource);
        
@@ -1126,8 +1177,10 @@ namespace System.Web.UI.WebControls
                                
                                if (prop_type == null || prop_type == typeof(object)) {
                                        IEnumerator en = source.GetEnumerator();
-                                       if (en.MoveNext())
+                                       if (en != null && en.MoveNext ()) {
                                                fitem = en.Current;
+                                               _dataEnumerator = en;
+                                       }
                                        if (fitem != null)
                                                prop_type = fitem.GetType();
                                }
@@ -1198,7 +1251,7 @@ namespace System.Web.UI.WebControls
                        Controls.Clear ();
                        table = null;
                        rows = null;
-                       
+
                        if (data == null) {
                                return 0;
                        }
@@ -1213,12 +1266,14 @@ namespace System.Web.UI.WebControls
                                if (AllowPaging) {
                                        dataSource.AllowPaging = true;
                                        dataSource.PageSize = PageSize;
-                                       dataSource.CurrentPageIndex = PageIndex;
                                        if (view.CanPage) {
                                                dataSource.AllowServerPaging = true;
                                                if (SelectArguments.RetrieveTotalRowCount)
                                                        dataSource.VirtualCount = SelectArguments.TotalRowCount;
                                        }
+                                       if (PageIndex >= dataSource.PageCount)
+                                               pageIndex = dataSource.PageCount - 1;
+                                       dataSource.CurrentPageIndex = PageIndex;
                                }
                                
                                PageCount = dataSource.PageCount;
@@ -1234,40 +1289,60 @@ namespace System.Web.UI.WebControls
                                }
                        }
 
-                       bool showPager = AllowPaging && (PageCount > 1);
-                       
+                       bool createPager = AllowPaging && (PageCount >= 1) && PagerSettings.Visible;
+
                        ArrayList list = new ArrayList ();
                        
                        // Creates the set of fields to show
-                       
+
+                       _dataEnumerator = null;
                        ICollection fieldCollection = CreateColumns (dataSource, dataBinding);
-                       DataControlField[] fields = new DataControlField [fieldCollection.Count];
+                       int fieldCount = fieldCollection.Count;
+                       DataControlField dcf;
+                       DataControlField[] fields = new DataControlField [fieldCount];
                        fieldCollection.CopyTo (fields, 0);
-
-                       foreach (DataControlField field in fields) {
-                               field.Initialize (AllowSorting, this);
+                       
+                       for (int i = 0; i < fieldCount; i++) {
+                               dcf = fields [i];
+                               dcf.Initialize (AllowSorting, this);
                                if (EnableSortingAndPagingCallbacks)
-                                       field.ValidateSupportsCallback ();
+                                       dcf.ValidateSupportsCallback ();
+                       }
+
+                       bool skip_first = false;
+                       IEnumerator enumerator;
+                       if (_dataEnumerator != null) {
+                               // replaced when creating bound columns
+                               enumerator = _dataEnumerator;
+                               skip_first = true;
+                       }
+                       else {
+                               enumerator = dataSource.GetEnumerator ();
                        }
 
                        // Main table creation
-                       foreach (object obj in dataSource) {
+                       Table mainTable = ContainedTable;
+                       while (skip_first || enumerator.MoveNext ()) {
+                               skip_first = false;
+                               object obj = enumerator.Current;
                                
                                if (list.Count == 0) {
-                                       if (showPager && PagerSettings.Position == PagerPosition.Top || PagerSettings.Position == PagerPosition.TopAndBottom) {
-                                               topPagerRow = CreatePagerRow (fields.Length, dataSource);
+                                       if (createPager && (PagerSettings.Position == PagerPosition.Top || PagerSettings.Position == PagerPosition.TopAndBottom)) {
+                                               topPagerRow = CreatePagerRow (fieldCount, dataSource);
                                                OnRowCreated (new GridViewRowEventArgs (topPagerRow));
-                                               ContainedTable.Rows.Add (topPagerRow);
+                                               mainTable.Rows.Add (topPagerRow);
                                                if (dataBinding) {
                                                        topPagerRow.DataBind ();
                                                        OnRowDataBound (new GridViewRowEventArgs (topPagerRow));
                                                }
+                                               if (PageCount == 1)
+                                                       topPagerRow.Visible = false;
                                        }
 
                                        GridViewRow headerRow = CreateRow (-1, -1, DataControlRowType.Header, DataControlRowState.Normal);
                                        InitializeRow (headerRow, fields);
                                        OnRowCreated (new GridViewRowEventArgs (headerRow));
-                                       ContainedTable.Rows.Add (headerRow);
+                                       mainTable.Rows.Add (headerRow);
                                        if (dataBinding) {
                                                headerRow.DataBind ();
                                                OnRowDataBound (new GridViewRowEventArgs (headerRow));
@@ -1280,21 +1355,21 @@ namespace System.Web.UI.WebControls
                                list.Add (row);
                                InitializeRow (row, fields);
                                OnRowCreated (new GridViewRowEventArgs (row));
-                               ContainedTable.Rows.Add (row);
+                               mainTable.Rows.Add (row);
                                if (dataBinding) {
-                                       row.DataBind ();
-                                       OnRowDataBound (new GridViewRowEventArgs (row));
+                                       row.DataBind ();                                        
                                        if (EditIndex == row.RowIndex)
                                                oldEditValues = new DataKey (GetRowValues (row, true, true));
                                        DataKeyArrayList.Add (new DataKey (CreateRowDataKey (row), DataKeyNames));
+                                       OnRowDataBound (new GridViewRowEventArgs (row));
                                } 
                        }
 
                        if (list.Count == 0) {
-                               GridViewRow emptyRow = CreateEmptyrRow (fields.Length);
+                               GridViewRow emptyRow = CreateEmptyrRow (fieldCount);
                                if (emptyRow != null) {
                                        OnRowCreated (new GridViewRowEventArgs (emptyRow));
-                                       ContainedTable.Rows.Add (emptyRow);
+                                       mainTable.Rows.Add (emptyRow);
                                        if (dataBinding) {
                                                emptyRow.DataBind ();
                                                OnRowDataBound (new GridViewRowEventArgs (emptyRow));
@@ -1306,20 +1381,22 @@ namespace System.Web.UI.WebControls
                                GridViewRow footerRow = CreateRow (-1, -1, DataControlRowType.Footer, DataControlRowState.Normal);
                                InitializeRow (footerRow, fields);
                                OnRowCreated (new GridViewRowEventArgs (footerRow));
-                               ContainedTable.Rows.Add (footerRow);
+                               mainTable.Rows.Add (footerRow);
                                if (dataBinding) {
                                        footerRow.DataBind ();
                                        OnRowDataBound (new GridViewRowEventArgs (footerRow));
                                }
 
-                               if (showPager && PagerSettings.Position == PagerPosition.Bottom || PagerSettings.Position == PagerPosition.TopAndBottom) {
-                                       bottomPagerRow = CreatePagerRow (fields.Length, dataSource);
+                               if (createPager && (PagerSettings.Position == PagerPosition.Bottom || PagerSettings.Position == PagerPosition.TopAndBottom)) {
+                                       bottomPagerRow = CreatePagerRow (fieldCount, dataSource);
                                        OnRowCreated (new GridViewRowEventArgs (bottomPagerRow));
-                                       ContainedTable.Rows.Add (bottomPagerRow);
+                                       mainTable.Rows.Add (bottomPagerRow);
                                        if (dataBinding) {
                                                bottomPagerRow.DataBind ();
                                                OnRowDataBound (new GridViewRowEventArgs (bottomPagerRow));
                                        }
+                                       if (PageCount == 1)
+                                               bottomPagerRow.Visible = false;
                                }
                        }
 
@@ -1419,6 +1496,7 @@ namespace System.Web.UI.WebControls
                        
                        for (int n=0; n<fields.Length; n++) {
                                DataControlField field = fields [n];
+                               
                                DataControlFieldCell cell;
                                if (((field is BoundField) && ((BoundField)field).DataField == RowHeaderColumn) || accessibleHeader)
                                        cell = new DataControlFieldHeaderCell (field, accessibleHeader ? TableHeaderScope.Column : TableHeaderScope.Row);
@@ -1457,10 +1535,17 @@ namespace System.Web.UI.WebControls
                
                protected virtual void ExtractRowValues (IOrderedDictionary fieldValues, GridViewRow row, bool includeReadOnlyFields, bool includePrimaryKey)
                {
+                       DataControlField field;
                        foreach (TableCell cell in row.Cells) {
                                DataControlFieldCell c = cell as DataControlFieldCell;
-                               if (c != null)
-                                       c.ContainingField.ExtractValuesFromCell (fieldValues, c, row.RowState, includeReadOnlyFields);
+                               if (c == null)
+                                       continue;
+                               
+                               field = c.ContainingField;
+                               if (field != null && !field.Visible)
+                                       continue;
+                               
+                               c.ContainingField.ExtractValuesFromCell (fieldValues, c, row.RowState, includeReadOnlyFields);
                        }
                        if (!includePrimaryKey && DataKeyNames != null)
                                foreach (string key in DataKeyNames)
@@ -1483,6 +1568,14 @@ namespace System.Web.UI.WebControls
                        base.DataBind ();
 
                        keys = new DataKeyArray (DataKeyArrayList);
+                       
+                       GridViewRow row = HeaderRow;
+                       if (row != null)
+                               row.Visible = ShowHeader;
+
+                       row = FooterRow;
+                       if (row != null)
+                               row.Visible = ShowFooter;
                }
                
                protected internal override void PerformDataBinding (IEnumerable data)
@@ -1494,9 +1587,10 @@ namespace System.Web.UI.WebControls
                {
                        if (table == null)
                                return;
-                       
+
                        table.Caption = Caption;
                        table.CaptionAlign = CaptionAlign;
+                       table.CopyBaseAttributes (this);
                        
                        foreach (GridViewRow row in table.Rows) {
                                switch (row.RowType) {
@@ -1536,6 +1630,13 @@ namespace System.Web.UI.WebControls
                                        DataControlFieldCell fcell = cell as DataControlFieldCell;
                                        if (fcell != null) {
                                                DataControlField field = fcell.ContainingField;
+                                               if (field == null)
+                                                       continue;
+                                               if (!field.Visible) {
+                                                       cell.Visible = false;
+                                                       continue;
+                                               }
+                                               
                                                switch (row.RowType) {
                                                case DataControlRowType.Header:
                                                        if (field.HeaderStyleCreated && !field.HeaderStyle.IsEmpty)
@@ -1609,6 +1710,7 @@ namespace System.Web.UI.WebControls
                
                void IPostBackEventHandler.RaisePostBackEvent (string eventArgument)
                {
+                       ValidateEvent (UniqueID, eventArgument);
                        RaisePostBackEvent (eventArgument);
                }
 
@@ -1784,9 +1886,8 @@ namespace System.Web.UI.WebControls
                        if (causesValidation && Page != null && !Page.IsValid)
                                return;
 
-                       currentEditOldValues = OldEditValues.Values;
-
-                       currentEditRowKeys = DataKeys [rowIndex].Values;
+                       currentEditOldValues = CopyOrderedDictionary (OldEditValues.Values);
+                       currentEditRowKeys = CopyOrderedDictionary (DataKeys [rowIndex].Values);
                        currentEditNewValues = GetRowValues (row, false, false);
                        
                        GridViewUpdateEventArgs args = new GridViewUpdateEventArgs (rowIndex, currentEditRowKeys, currentEditOldValues, currentEditNewValues);
@@ -1801,6 +1902,14 @@ namespace System.Web.UI.WebControls
                        view.Update (currentEditRowKeys, currentEditNewValues, currentEditOldValues, new DataSourceViewOperationCallback (UpdateCallback));
                }
 
+               static IOrderedDictionary CopyOrderedDictionary (IOrderedDictionary sourceDic) {
+                       OrderedDictionary copyDic = new OrderedDictionary ();
+                       foreach (object key in sourceDic.Keys) {
+                               copyDic.Add (key, sourceDic [key]);
+                       }
+                       return copyDic;
+               }
+
                bool UpdateCallback (int recordsAffected, Exception exception)
                {
                        GridViewUpdatedEventArgs dargs = new GridViewUpdatedEventArgs (recordsAffected, exception, currentEditRowKeys, currentEditOldValues, currentEditNewValues);
@@ -1815,7 +1924,7 @@ namespace System.Web.UI.WebControls
                public virtual void DeleteRow (int rowIndex)
                {
                        GridViewRow row = Rows [rowIndex];
-                       currentEditRowKeys = DataKeys [rowIndex].Values;
+                       currentEditRowKeys = CopyOrderedDictionary (DataKeys [rowIndex].Values);
                        currentEditNewValues = GetRowValues (row, true, true);
                        
                        GridViewDeleteEventArgs args = new GridViewDeleteEventArgs (rowIndex, currentEditRowKeys, currentEditNewValues);
@@ -2038,9 +2147,12 @@ namespace System.Web.UI.WebControls
                
                protected virtual string GetCallbackScript (IButtonControl control, string argument)
                {
-                       if (EnableSortingAndPagingCallbacks)
+                       if (EnableSortingAndPagingCallbacks) {
+                               Page page = Page;
+                               if (page != null)
+                                       page.ClientScript.RegisterForEventValidation (UniqueID, argument);
                                return "javascript:GridView_ClientEvent (\"" + ClientID + "\",\"" + control.CommandName + "$" + control.CommandArgument + "\"); return false;";
-                       else
+                       else
                                return null;
                }
                
@@ -2058,25 +2170,34 @@ namespace System.Web.UI.WebControls
                                SortExpression = Page.Request.Form [ClientID + "_SortExpression"];
                        }
                }
-               
+
+
+               const string onPreRenderScript = @"var {0} = new Object ();
+{0}.pageIndex = {1};
+{0}.sortExp = {2};
+{0}.sortDir = {3};
+{0}.uid = {4};
+{0}.form = {5};
+";
                protected internal override void OnPreRender (EventArgs e)
                {
                        base.OnPreRender (e);
-                       
-                       if (EnableSortingAndPagingCallbacks)
-                       {
+
+                       if (EnableSortingAndPagingCallbacks) {
                                if (!Page.ClientScript.IsClientScriptIncludeRegistered (typeof(GridView), "GridView.js")) {
                                        string url = Page.ClientScript.GetWebResourceUrl (typeof(GridView), "GridView.js");
                                        Page.ClientScript.RegisterClientScriptInclude (typeof(GridView), "GridView.js", url);
                                }
                                
                                string cgrid = ClientID + "_data";
-                               string script = string.Format ("var {0} = new Object ();\n", cgrid);
-                               script += string.Format ("{0}.pageIndex = {1};\n", cgrid, ClientScriptManager.GetScriptLiteral (PageIndex));
-                               script += string.Format ("{0}.sortExp = {1};\n", cgrid, ClientScriptManager.GetScriptLiteral (SortExpression == null ? "" : SortExpression));
-                               script += string.Format ("{0}.sortDir = {1};\n", cgrid, ClientScriptManager.GetScriptLiteral ((int) SortDirection));
-                               script += string.Format ("{0}.uid = {1};\n", cgrid, ClientScriptManager.GetScriptLiteral (UniqueID));
-                               script += string.Format ("{0}.form = {1};\n", cgrid, Page.theForm);
+                               string script = String.Format (onPreRenderScript,
+                                                       cgrid,
+                                                       ClientScriptManager.GetScriptLiteral (PageIndex),
+                                                       ClientScriptManager.GetScriptLiteral (SortExpression == null ? "" : SortExpression),
+                                                       ClientScriptManager.GetScriptLiteral ((int) SortDirection),
+                                                       ClientScriptManager.GetScriptLiteral (UniqueID),
+                                                       Page.theForm);
+                               
                                Page.ClientScript.RegisterStartupScript (typeof(TreeView), this.UniqueID, script, true);
                                
                                // Make sure the basic script infrastructure is rendered
@@ -2084,11 +2205,10 @@ namespace System.Web.UI.WebControls
                                Page.ClientScript.GetPostBackClientHyperlink (this, "");
                        }
                }
-
+               
                protected internal override void Render (HtmlTextWriter writer)
                {
                        PrepareControlHierarchy ();
-
                        if (EnableSortingAndPagingCallbacks)
                                writer.AddAttribute (HtmlTextWriterAttribute.Id, ClientID + "_div");
                        writer.RenderBeginTag (HtmlTextWriterTag.Div);