chore(*): Add and fix tests.

Author: MKirova

projects/igniteui-angular/src/lib/grids/grid-base.directive.ts

@@ -4071,7 +4071,7 @@ export abstract class IgxGridBaseDirective extends DisplayDensityBase implements
         this.cancelAddMode = false;
         const isInPinnedArea = this.isRecordPinnedByViewIndex(index);
         const pinIndex = this.pinnedRecords.findIndex(x => x[this.primaryKey] === rowID);
-        const unpinIndex = this.unpinnedRecords.findIndex(x => x[this.primaryKey] === rowID);
+        const unpinIndex = this.getUnpinnedIndexById(rowID); 
 
         if (this.expansionStates.get(rowID)) {
             this.collapseRow(rowID);
@@ -6606,6 +6606,10 @@ export abstract class IgxGridBaseDirective extends DisplayDensityBase implements
         return this.dataView.findIndex(data => data === rec);
     }
 
+    protected getUnpinnedIndexById(id) {
+        return this.unpinnedRecords.findIndex(x => x[this.primaryKey] === id);
+    }
+
     /**
      * @hidden
      */

projects/igniteui-angular/src/lib/grids/grid/grid-add-row.spec.ts

@@ -12,6 +12,7 @@ import { By } from '@angular/platform-browser';
 import { IgxActionStripComponent } from '../../action-strip/action-strip.component';
 import { IgxActionStripModule } from '../../action-strip/action-strip.module';
 import { UIInteractions } from '../../test-utils/ui-interactions.spec';
+import { IgxGridRowComponent } from './grid-row.component';
 
 describe('IgxGrid - Row Adding #grid', () => {
         let fixture;
@@ -111,5 +112,50 @@ describe('IgxGrid - Row Adding #grid', () => {
             expect(grid.getRowByIndex(1).addRow).toBeFalse();
         });
 
+        it('Should allow adding row from pinned row.', () => {
+            let row = grid.getRowByIndex(0);
+            row.pin();
+            fixture.detectChanges();
+            expect(grid.pinnedRecords.length).toBe(1);
+
+            row = grid.getRowByIndex(0);
+            row.beginAddRow();
+            fixture.detectChanges();
+
+            //add row should be pinned
+            const addRow = grid.getRowByIndex(1) as IgxGridRowComponent;
+            expect(addRow.addRow).toBe(true);
+            expect(grid.pinnedRows[1]).toBe(addRow);
+
+            grid.endEdit(true);
+            fixture.detectChanges();
+
+            // added record should be pinned.
+            expect(grid.pinnedRecords.length).toBe(2);
+            expect(grid.pinnedRecords[1]).toBe(grid.data[grid.data.length - 1]);
+
+        });
+        it('Should allow adding row from ghost row.', () => {
+            let row = grid.getRowByIndex(0);
+            row.pin();
+            fixture.detectChanges();
+            expect(grid.pinnedRecords.length).toBe(1);
+
+            const ghostRow = grid.getRowByIndex(1);
+            ghostRow.beginAddRow();
+            fixture.detectChanges();
+
+            //add row should be unpinned
+            const addRow = grid.getRowByIndex(2);
+            expect(addRow.addRow).toBe(true);
+            expect(grid.pinnedRows.length).toBe(1);
+
+            grid.endEdit(true);
+            fixture.detectChanges();
+
+            // added record should be unpinned.
+            expect(grid.pinnedRecords.length).toBe(1);
+            expect(grid.unpinnedRecords[grid.unpinnedRecords.length - 1]).toBe(grid.data[grid.data.length - 1]);
+        });
     });
 });

projects/igniteui-angular/src/lib/grids/tree-grid/tree-grid.component.ts

@@ -439,6 +439,10 @@ export class IgxTreeGridComponent extends IgxGridBaseDirective implements GridTy
         return this.dataView.findIndex(x => x.data === rec);
     }
 
+    protected getUnpinnedIndexById(id) {
+        return this.unpinnedRecords.findIndex(x => x.data[this.primaryKey] === id);
+    }
+
     private cloneMap(mapIn: Map<any, boolean>): Map<any, boolean> {
         const mapCloned: Map<any, boolean> = new Map<any, boolean>();