Added test and move function.

Signed-off-by: Dennis Labordus <dennis.labordus@alliander.com>

Author: Dennis Labordus

src/compas/CompasLabelsField.ts

@@ -77,6 +77,16 @@ export class CompasLabelsFieldElement extends LitElement {
     this.requestUpdate('labels');
   }
 
+  public updateLabelsInPrivateElement(privateElement: Element): void {
+    // We will just add or replace the complete Labels Element, so if it exists
+    // first remove it and always add the new one.
+    if (this.originalLabelsElement) {
+      privateElement?.removeChild(this.originalLabelsElement);
+    }
+    privateElement?.append(this.newLabelsElement);
+    this.originalLabelsElement = this.newLabelsElement;
+  }
+
   render(): TemplateResult {
     const labels = this.labels;
     return html`

src/compas/CompasSave.ts

@@ -76,13 +76,7 @@ export default class CompasSaveElement extends CompasExistsIn(LitElement) {
   private updateLabels() {
     const sclElement = this.doc.documentElement;
     const privateElement = getPrivate(sclElement, COMPAS_SCL_PRIVATE_TYPE);
-    // We will just add or replace the complete Labels Element, so if it exists
-    // first remove it and always add the new one.
-    if (this.labelsField.originalLabelsElement) {
-      privateElement?.removeChild(this.labelsField.originalLabelsElement);
-    }
-    privateElement?.append(this.labelsField.newLabelsElement);
-    this.labelsField.originalLabelsElement = this.labelsField.newLabelsElement;
+    this.labelsField.updateLabelsInPrivateElement(privateElement!);
   }
 
   private async addSclToCompas(doc: XMLDocument): Promise<boolean> {

test/unit/compas/CompasLabelsField.test.ts

@@ -72,6 +72,15 @@ describe('compas-labels-field', () => {
       expect(labelElements.length).to.be.equal(2);
     });
 
+    it('when calling updateLabelsInPrivateElement then Private Element is updated', async () => {
+      expect(privateElement.querySelectorAll('Label').length).to.be.equal(1);
+
+      await addLabel(element, 'Label2');
+      element.updateLabelsInPrivateElement(privateElement);
+
+      expect(privateElement.querySelectorAll('Label').length).to.be.equal(2);
+    });
+
     it('when removing a label then label element removed', async () => {
       await removeLabel(element, 'Label1');