From d9b9ec39de5d786ad708ef7c327297b3d36be5fa Mon Sep 17 00:00:00 2001
From: YiscahLevySilas1 <yiscahls@armosec.io>
Date: Mon, 25 Mar 2024 12:13:34 +0200
Subject: [PATCH] add constructor for regolibrary V2

Signed-off-by: YiscahLevySilas1 <yiscahls@armosec.io>
---
 gitregostore/datastructures.go       |  7 +++++++
 gitregostore/gitstoremethods_test.go | 12 ++++++++++++
 2 files changed, 19 insertions(+)

diff --git a/gitregostore/datastructures.go b/gitregostore/datastructures.go
index 9f24d57a9..5bc357be2 100644
--- a/gitregostore/datastructures.go
+++ b/gitregostore/datastructures.go
@@ -98,6 +98,13 @@ func (gs *GitRegoStore) SetRegoObjects() error {
 	return err
 }
 
+// NewDefaultGitRegoStore - generates git store object for production regolibrary release files.
+// Release files source: "https://github.com/kubescape/regolibrary/releases/latest/download"
+func NewGitRegoStoreV2(frequency int) *GitRegoStore {
+	gs := NewGitRegoStore("https://github.com", "kubescape", "regolibrary", "releases", "download/v2", "", frequency)
+	return gs
+}
+
 // NewDefaultGitRegoStore - generates git store object for production regolibrary release files.
 // Release files source: "https://github.com/kubescape/regolibrary/releases/latest/download"
 func NewDefaultGitRegoStore(frequency int) *GitRegoStore {
diff --git a/gitregostore/gitstoremethods_test.go b/gitregostore/gitstoremethods_test.go
index 400f96bdf..b25797925 100644
--- a/gitregostore/gitstoremethods_test.go
+++ b/gitregostore/gitstoremethods_test.go
@@ -222,6 +222,18 @@ func gs_tests(t *testing.T, gs *GitRegoStore) {
 	})
 }
 
+func TestGetPoliciesMethodsNewV2(t *testing.T) {
+	t.Parallel()
+
+	gs := NewGitRegoStoreV2(-1)
+	t.Run("shoud set objects in rego store", func(t *testing.T) {
+		require.NoError(t, gs.SetRegoObjects())
+	})
+
+	gs_tests(t, gs)
+
+}
+
 func TestGetPoliciesMethodsNew(t *testing.T) {
 	t.Parallel()