workflow setup retries and tests (#3)

* typo

* new key cases

* key list cases

* change to pytest

* add worklfow

* image run

* fix

* try this

* image changes

* typo

* another one

* registry login changes

* changes

* spacing

* registry issue

* err

* change

* credential edit

* err

* try new

* new docker action

* new change

* check

* maybe

* errr

* try

* hate typos fr

* maybe?

* try installing docker itself

* now add back tests

* few cahnges

* docker liveness changes

* try without login

* modifying workflow paths

* typo

Author: pfefar

.github/workflows/tests.yaml

@@ -0,0 +1,37 @@
+name: Tests
+on:
+  push:
+    paths:
+      - 'fortress_sdk/**'
+      - 'tests/**'
+      - '.github/workflows/tests.yaml'
+
+jobs:
+  build-package-and-run-tests:
+    runs-on: ubuntu-latest
+    steps:
+      - name: Login to GitHub Container Registry
+        uses: docker/login-action@v2
+        with:
+          registry: ghcr.io
+          username: ${{ github.actor }}
+          password: ${{ secrets.GITHUB_TOKEN }}
+      - uses: docker-practice/actions-setup-docker@master
+      - run: |
+          docker run -d -t -e ADMIN_API_KEY="admin_key" -e SELLER_API_KEY="seller_key" -e BUYER_API_KEY="buyer_key" -p 8080:8080 ghcr.io/atbash-labs/enclave-test
+      - name: Checkout repository
+        uses: actions/checkout@v3
+      - name: Set up Python 3.8
+        uses: actions/setup-python@v4
+        with:
+          python-version: "3.8"
+      - name: Install dependencies
+        run: |
+          python -m pip install --upgrade pip
+          pip install -r requirements.txt
+      - name: install fortress sdk
+        run: |
+          pip install -e .
+          sleep 2m
+      - name: Test with pytest
+        run: pytest 

README.md

@@ -10,7 +10,7 @@ pip install git+https://github.com/Atbash-Labs/python-sdk@<tag>
 
 The module can then be used as follows:
 ```python
-from fortress_sdk import *
+from fortress_sdk import Buyer,Sftp
 ```
 
 check out these [demo](./examples)  interactions with the library

fortress_sdk/fortress_sdk.py

@@ -133,9 +133,11 @@ def get_key_list(self):
             error = rsp["message"]
             print(error)
 
-        subkey_list = rsp["subkeys"]
-
-        return subkey_list
+        try:
+            subkey_list = rsp["subkeys"]
+            return subkey_list
+        except:
+            return None
 
     def query(self, query_key=None, query=""):
         """Initiate the query and return the result

requirements.txt

@@ -1,2 +1,3 @@
 pysftp==0.2.9
-requests==2.28.1
+requests==2.28.1
+pytest==7.2.0

tests/__init__.py

@@ -0,0 +1,65 @@
+import pytest
+from fortress_sdk import Buyer
+
+
+def new_valid_buyer():
+    api_key = "buyer_key"
+    ip_addr = "127.0.0.1"
+    buyer = Buyer(api_key, ip_addr)
+    return buyer
+
+
+def new_invalid_buyer():
+    api_key = "invalid_buyer_key"
+    ip_addr = "127.0.0.1"
+    buyer = Buyer(api_key, ip_addr)
+    return buyer
+
+
+def test_get_key_list_success():
+    buyer = new_valid_buyer()
+    key_list = buyer.get_key_list()
+
+    assert len(key_list) == 1
+
+
+def test_key_list_added_key():
+    buyer = new_valid_buyer()
+
+    key_list = buyer.get_key_list()
+    assert len(key_list) == 1
+
+    _ = buyer.get_key()
+
+    key_list = buyer.get_key_list()
+    assert len(key_list) == 2
+
+
+def test_invalid_buyer_for_key_list():
+    buyer = new_invalid_buyer()
+    key_list = buyer.get_key_list()
+
+    assert key_list == None
+
+
+def test_get_key_success():
+    buyer = new_valid_buyer()
+    new_key = buyer.get_key()
+
+    assert new_key != None
+
+
+def test_invalid_buyer_for_new_key():
+    buyer = new_invalid_buyer()
+    new_key = buyer.get_key()
+
+    assert new_key == None
+
+
+def test_query_success():
+    buyer = new_valid_buyer()
+    sql_query = "select count(*) as numpeople from public.condition_era_death"
+    result, accuracy = buyer.query(query=sql_query)
+
+    assert result != None
+    assert accuracy != ""