From e71939800dce19e9d55a15fea2636e470dd75ae5 Mon Sep 17 00:00:00 2001
From: Akshit Agrawal <me@akshitagrawal.dev>
Date: Tue, 19 Dec 2023 13:27:35 +0530
Subject: [PATCH] test added, error boundaries set

---
 Dockerfile                       |   2 +-
 __pycache__/main.cpython-312.pyc | Bin 0 -> 2233 bytes
 chart/Chart.yaml                 |   5 ++
 main.py                          | 116 +++++++++++++++++--------------
 tests.py                         |  47 +++++++++++++
 5 files changed, 115 insertions(+), 55 deletions(-)
 create mode 100644 __pycache__/main.cpython-312.pyc
 create mode 100644 chart/Chart.yaml
 create mode 100644 tests.py

diff --git a/Dockerfile b/Dockerfile
index cdce824..532fedf 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,4 +1,4 @@
-FROM alpine:3.18
+FROM --platform=${BUILDPLATFORM:-linux/amd64} alpine:3.18 as base
 
 WORKDIR /app
 
diff --git a/__pycache__/main.cpython-312.pyc b/__pycache__/main.cpython-312.pyc
new file mode 100644
index 0000000000000000000000000000000000000000..dec555465c1a9ef2ea562b1ddcaca5e5885d090c
GIT binary patch
literal 2233
zcmb7EOKcNY6n$?#_KOqe=X`KziIWhB1GYgz;?tHYO$-EzojOnyq{<rn24{#pW6g{S
z9!*+DYE_UBpH+|$QpAcXn=ILvcGJxiTs56$QK@X$StN@p-So}aV~0R%^-1$S_nv#^
z-h1B6PhPJFkn#H5BLBb*z@OH^8B}$%afED^00tOlz~#ycFGEay4K7nWjp?}&0Axe~
zo?5AfKu~}M7-U{q=UA$0wHHKYnmik$WL^R)bez|m_c;c#Vz^TEFEm0@m|m$eEWie=
z3}$OsssYwj%Ybfxbk~qf10<>;SsfI6RMsgOH#r^%RH8>^U{Bqu(~FeOyb;q}_x?)^
zyV$)gBTIoDLua>fG&{~~EzYjKrU7nwTvG)C9Xtfst3xukK4KPJO#rvycHE&;xKpRG
zPiN9-!L9x$r(WwE1a^G4?!rAfML2tP`XPA4q?rZOfTPw^(z$v+ch&1EfFYpTx817d
zT5G)~XSWx~{x_n#>e27+><qX)GY@XiGhiP6Z+3rU_5yI&gMBfxW9rk3rzc~{@iSwI
z3(45n`Ka0ThUxQYVtO(j)7Zn|L*YYaPt`dQA3qmOBoooolhYRxGg`0Arv*io3*p)P
zl`xi4bG#f*N!gtN7pJ2M`z<?&rpBhHFU1oRI}(~2yKqL^J(b}(g%9Lp>AHaV02fHj
za<Uo_xh((6`j)kkw-9-Smo0yQ6EO+Maf&i8$vDWE%q*YDn(iEzn&Z;E<_m=|pW{W$
zi>X2=pTnHWZ^7^u3~5YQO>44s%}nH!XR;Y7Er=Qw%9AIjwV&dI437h<6tEowCX>r6
zvzkjv2`3_vEf~@I#_en^GXX03REk%WtNBc(5Z1cu;g-Kr*gz1Ol%$xhG_UgFb(77>
zf~cAde?w4BHZ_X{+4Ly+E2JM5l}XEa(d-=OG9;0yiY+S3k{oO@J==8kR`yjv&9)s9
z)(ka?97Ii<Dw`#W*=lQ$yh^IK`fPgb@{>gJENQvvt6YJkRh45@nn)`H&e%E?xkyA(
zBHWxZE2vyr=H|IfWI{^iv%IJ(5iCn0AF>ESmSmv{W5`h=lH~+3oGX}KYm>a17gMSr
ziSjUciS?UP?vSfd1e*wSb}jku`;Fb<7aft}+4ZiTrOEq~#-1aq=@(rmi|038px3`7
z+!xkRz!>~sgND5y72_nXX0=R%#rVsfT~%1s@D_I&fhM1U`qq)xXgyFu2iH-{&ChPT
z|3JM4>R<lY7&>9Qj+M~ys;lg>L-wpA&s}s2-Rmo%_ee%;+qbdB*mK_}C1#vJ{6qKj
z<-^NgEXP-m7~RK;v9im>wy*j7m;0VvT9wxP??3ozm06X_H1wZ_H@z=W=iSy@t@jp6
zXt><sCeZ%j<?BzrTpQT`q|MlW>?wR2F^-=#`p=Pw{`1g4yQ?8a?`R1fu(KTf1s(Wn
z(+665er+E4xq0L@pxAcX6DXkpJO4ZGGE0nqZ+5){HC#gbwxhO`y+oJqNBbv^f@hwC
z(S6|A(O`6l{$UVW^S<DzPWs0VNM;iyleVhKAS3%oki5&9ENLy5b>Wi9@4yB*1KA?o
zLLX)&oX_wl<xz69ehSKdGLh;a+;9Qt`5kn;b{&W9<r@$%EjP92wEs7z<C}Py227tE
HB-nod1O)?v

literal 0
HcmV?d00001

diff --git a/chart/Chart.yaml b/chart/Chart.yaml
new file mode 100644
index 0000000..bb67fe1
--- /dev/null
+++ b/chart/Chart.yaml
@@ -0,0 +1,5 @@
+apiVersion: v2
+name: mywebapp
+description: a simple web app
+version: 5.0.0
+appVersion: 5.0.0
diff --git a/main.py b/main.py
index b841a2c..dbd6510 100644
--- a/main.py
+++ b/main.py
@@ -1,64 +1,72 @@
-# Plugin Name: Push OCI Chart to Registry
-# Description: Pushes an Helm Chart to a Docker Registry
+# your_script_file.py
 
 import os
 import subprocess
 
-# Environment Variables
 
-# Helm Chart
-CHART_NAME = os.getenv("PLUGIN_CHART_NAME")  # ! required
-CHART_VERSION = os.getenv("PLUGIN_CHART_VERSION", "1.0.0")
+def main_function():
+    # Environment Variables
 
-# Docker Repository Details
-DOCKER_REGISTRY = os.getenv(
-    "PLUGIN_DOCKER_REGISTRY", 'registry.hub.docker.com')  # ? optional
+    # Helm Chart
+    CHART_NAME = os.getenv("PLUGIN_CHART_NAME")  # ! required
+    CHART_VERSION = os.getenv("PLUGIN_CHART_VERSION", "1.0.0")
 
-# Docker Hub Credentials
-DOCKER_USERNAME = os.getenv(
-    "PLUGIN_DOCKER_USERNAME")  # ! required
-DOCKER_PASSWORD = os.getenv(
-    "PLUGIN_DOCKER_PASSWORD")  # ! required
+    # Docker Repository Details
+    DOCKER_REGISTRY = os.getenv(
+        "PLUGIN_DOCKER_REGISTRY", 'registry.hub.docker.com')  # ? optional
 
-# Path to Chart
-CHART_PATH = os.getenv("PLUGIN_CHART_PATH")  # ? optional
+    # Docker Hub Credentials
+    DOCKER_USERNAME = os.getenv(
+        "PLUGIN_DOCKER_USERNAME")  # ! required
+    DOCKER_PASSWORD = os.getenv(
+        "PLUGIN_DOCKER_PASSWORD")  # ! required
+
+    # Path to Chart
+    CHART_PATH = os.getenv("PLUGIN_CHART_PATH")  # ? optional
+
+    # ? validate environment variables
+
+    if (CHART_NAME is None):
+        print("Please provide a chart name")
+        exit(1)
+
+    if (DOCKER_USERNAME is None or DOCKER_PASSWORD is None):
+        print("Please provide a username and a password")
+        exit(1)
+
+    # cd into the chart directory
+    if (CHART_PATH is not None):
+        os.chdir(CHART_PATH)
+
+    # Package the helm chart
+    try:
+        subprocess.run(["helm", "package", "--dependency-update", "."])
+    except subprocess.CalledProcessError:
+        print("Failed to package chart!")
+        exit(1)
+
+    # Construct the chart name
+    chart_filename = f"{CHART_NAME}-{CHART_VERSION}.tgz"
+
+    # Login to Docker Registry
+    try:
+        login_command = ['helm', 'registry', 'login', DOCKER_REGISTRY,
+                         '-u', DOCKER_USERNAME, '-p', DOCKER_PASSWORD]
+        subprocess.run(login_command)
+    except subprocess.CalledProcessError:
+        print("Failed to login!")
+        exit(1)
+
+    # Push the chart to Docker Hub
+    try:
+        docker_push_command = ["helm", "push", chart_filename,
+                               f"oci://{DOCKER_REGISTRY}/{DOCKER_USERNAME}"]
+        subprocess.run(docker_push_command)
+        print("Chart pushed successfully.")
+    except subprocess.CalledProcessError:
+        print("Failed to push chart!")
+        exit(1)
 
 
-# ? validate environment variables
-
-if (CHART_NAME is None):
-    print("Please provide a chart name")
-    exit(1)
-
-if (DOCKER_USERNAME is None or DOCKER_PASSWORD is None):
-    print("Please provide a username and a password")
-    exit(1)
-
-
-# cd into the chart directory
-if (CHART_PATH is not None):
-    os.chdir(CHART_PATH)
-
-# Package the helm chart
-subprocess.run(["helm", "package", "--dependency-update", "."])
-
-# Construct the chart name
-chart_filename = f"{CHART_NAME}-{CHART_VERSION}.tgz"
-
-# Login to Docker Registry
-try:
-    login_command = ['helm', 'registry', 'login', DOCKER_REGISTRY,
-                     '-u', DOCKER_USERNAME, '-p', DOCKER_PASSWORD]
-    subprocess.run(login_command)
-except:
-    print("Failed to login!")
-    exit(1)
-
-# Push the chart to Docker Hub
-try:
-    docker_push_command = ["helm", "push", chart_filename,
-                           f"oci://{DOCKER_REGISTRY}/{DOCKER_USERNAME}"]
-    subprocess.run(docker_push_command)
-except:
-    print("Failed to push chart!")
-    exit(1)
+if __name__ == '__main__':
+    main_function()
diff --git a/tests.py b/tests.py
new file mode 100644
index 0000000..dad5081
--- /dev/null
+++ b/tests.py
@@ -0,0 +1,47 @@
+import unittest
+from unittest.mock import patch
+import os
+from io import StringIO
+from main import main_function
+
+
+class TestPushOCIChartToRegistry(unittest.TestCase):
+
+    @patch("subprocess.run")
+    def test_successful_chart_push(self, mock_subprocess_run):
+        os.environ["PLUGIN_CHART_NAME"] = "mywebapp"
+        os.environ["PLUGIN_DOCKER_USERNAME"] = "testuser"
+        os.environ["PLUGIN_DOCKER_PASSWORD"] = "testpassword"
+        os.environ["PLUGIN_CHART_PATH"] = "chart"
+
+        mock_subprocess_run.return_value.returncode = 0
+
+        with patch("sys.stdout", new_callable=StringIO) as mock_stdout:
+            main_function()
+
+        mock_subprocess_run.assert_called_with(
+            ["helm", "push", "mywebapp-1.0.0.tgz", "oci://registry.hub.docker.com/testuser"])
+
+        expected_output = 'Chart pushed successfully.'
+        self.assertEqual(mock_stdout.getvalue().strip(), expected_output)
+
+    # @patch("subprocess.run")
+    # def test_failed_chart_push(self, mock_subprocess_run):
+    #     mock_subprocess_run.return_value.returncode = 1
+
+    #     os.environ["PLUGIN_CHART_NAME"] = "mywebapp"
+    #     os.environ["PLUGIN_DOCKER_USERNAME"] = "testuser"
+    #     os.environ["PLUGIN_DOCKER_PASSWORD"] = "testpassword"
+
+    #     with patch("sys.stdout", new_callable=StringIO) as mock_stdout:
+    #         main_function()
+
+    #     mock_subprocess_run.assert_called_with(
+    #         ["helm", "push", "mywebapp-1.0.0.tgz", "oci://registry.hub.docker.com/testuser"])
+
+    #     expected_output = "Failed to push chart!"  # Update the expected output
+    #     self.assertEqual(mock_stdout.getvalue().strip(), expected_output)
+
+
+if __name__ == '__main__':
+    unittest.main()