From c517d996e5a77a30fadb8ec9e586ce8fd2263c83 Mon Sep 17 00:00:00 2001
From: SvetoslavTsenov <tsenov.svetoslav@gmail.com>
Date: Thu, 27 Jun 2019 17:12:29 +0300
Subject: [PATCH] fix(image-helper): increase image name

---
 lib/image-helper.ts | 31 ++++++++++++++++++++++++++++---
 1 file changed, 28 insertions(+), 3 deletions(-)

diff --git a/lib/image-helper.ts b/lib/image-helper.ts
index bad7907..37f40f2 100644
--- a/lib/image-helper.ts
+++ b/lib/image-helper.ts
@@ -111,12 +111,37 @@ export class ImageHelper {
 
     public reset() {
         this._imagesResults.clear();
+        this.testName = undefined;
     }
 
     private increaseImageName(imageName: string) {
-        if (this._imagesResults.size > 1) {
-            const number = /\d+$/.test(imageName) ? +`${/\d+$/.exec(imageName)}` + 1 : `2`;
-            imageName = `${imageName}_${number}`;
+        if (!imageName) {
+            logError(`Missing image name!`);
+            logError(`Consider to set
+            drive.imageHelper.testName
+            dirver.imageHelper.options.imageName
+    `);
+            throw new Error(`Missing image name!`)
+        }
+        if (this._imagesResults.size > 0) {
+            const images = new Array();
+            this._imagesResults.forEach((v, k, map) => {
+                if (k.includes(imageName)) {
+                    images.push(k);
+                }
+            });
+
+            images.sort((a, b) => {
+                const aNumber = +/\d+$/.exec(a);
+                const bNumber = +/\d+$/.exec(b);
+
+                return bNumber - aNumber;
+            });
+            if (images.length > 0) {
+                const lastImage = images[0];
+                const number = /\d+$/.test(lastImage) ? +`${/\d+$/.exec(lastImage)}` + 1 : `2`;
+                imageName = `${imageName}_${number}`;
+            }
         }
 
         return imageName;