2022-10-17 17:18:17 -05:00
|
|
|
import unittest
|
|
|
|
import requests
|
2023-04-10 22:51:37 -05:00
|
|
|
import json
|
2022-10-28 01:44:29 -05:00
|
|
|
from tomllib import load
|
2023-06-25 23:46:10 -05:00
|
|
|
from cclient import c_get_filtered_containers, c_start_container, c_stop_container, c_get_container_id
|
2023-04-10 22:51:37 -05:00
|
|
|
from cprocess import process_cont_list
|
2023-06-25 23:46:10 -05:00
|
|
|
unittest.TestLoader.sortTestMethodsUsing = None
|
|
|
|
requests.packages.urllib3.disable_warnings()
|
2022-10-17 17:18:17 -05:00
|
|
|
|
|
|
|
class TestCrane(unittest.TestCase):
|
|
|
|
def setUp(self):
|
2022-10-28 01:44:29 -05:00
|
|
|
with open('./config.toml', 'rb') as c:
|
2022-10-17 17:18:17 -05:00
|
|
|
self.config = load(c)
|
2022-10-28 01:44:29 -05:00
|
|
|
self.host = self.config["portainer"]["host"]
|
|
|
|
self.port = self.config["portainer"]["port"]
|
2023-04-10 22:51:37 -05:00
|
|
|
self.access_token = "ptr_ufS1nADXmrU3QSN3bvITLMQ7oOH9yo3ECb/QNwtIYJ4="
|
2022-10-28 01:44:29 -05:00
|
|
|
self.endpoint = self.config["portainer"]["endpoint"]
|
2022-10-17 17:18:17 -05:00
|
|
|
self.req_obj = requests
|
2023-04-10 22:51:37 -05:00
|
|
|
self.j_obj = json
|
2023-06-25 23:46:10 -05:00
|
|
|
self.hypercare_containers = ['hottub']
|
2023-04-10 22:51:37 -05:00
|
|
|
self.status_filters = ["paused","dead","created","exited","removing","restarting","created"]
|
2022-10-21 12:48:19 -05:00
|
|
|
|
2023-04-10 22:51:37 -05:00
|
|
|
def test_c_get_filtered_containers(self):
|
2023-06-25 23:46:10 -05:00
|
|
|
self.cont_obj = c_get_filtered_containers(self.req_obj,self.j_obj, self.host, self.port, self.access_token, self.endpoint,self.hypercare_containers,self.status_filters)
|
2023-04-10 22:51:37 -05:00
|
|
|
self.assertTrue(self.cont_obj, "No cont object returned by c_get_filtered_containers.")
|
2022-10-21 12:48:19 -05:00
|
|
|
|
2022-10-17 17:18:17 -05:00
|
|
|
def test_c_start_container(self):
|
2023-06-25 23:46:10 -05:00
|
|
|
self.cid = c_get_container_id(self.req_obj,self.j_obj, self.host, self.port, self.access_token, self.endpoint, self.hypercare_containers,self.status_filters)
|
|
|
|
c_stop_container(self.req_obj, self.host, self.port, self.access_token, self.endpoint, self.cid)
|
2023-04-10 22:51:37 -05:00
|
|
|
self.c_start_container_response = c_start_container(self.req_obj, self.host, self.port, self.access_token, self.endpoint, self.cid)
|
2022-10-17 17:18:17 -05:00
|
|
|
self.assertTrue(self.c_start_container_response, "No c_start_container_resonse returned by c_start_container.")
|
|
|
|
# 204 success 304 already on
|
2022-10-18 14:14:53 -05:00
|
|
|
|
2023-04-10 22:51:37 -05:00
|
|
|
def test_c_process_cont_list(self):
|
2023-06-25 23:46:10 -05:00
|
|
|
self.cont_obj = c_get_filtered_containers(self.req_obj,self.j_obj, self.host, self.port, self.access_token, self.endpoint,self.hypercare_containers,self.status_filters)
|
2023-04-10 22:51:37 -05:00
|
|
|
self.assertTrue(process_cont_list(self.cont_obj, c_start_container, self.req_obj, self.host, self.port, self.access_token, self.endpoint))
|
|
|
|
|
2022-10-18 14:14:53 -05:00
|
|
|
def test_c_stop_container(self):
|
2023-06-25 23:46:10 -05:00
|
|
|
self.cid = c_get_container_id(self.req_obj,self.j_obj, self.host, self.port, self.access_token, self.endpoint, self.hypercare_containers,self.status_filters)
|
2023-04-10 22:51:37 -05:00
|
|
|
self.c_stop_container_response = c_stop_container(self.req_obj, self.host, self.port, self.access_token, self.endpoint, self.cid)
|
2022-10-18 14:14:53 -05:00
|
|
|
self.assertTrue(self.c_stop_container_response, "No c_start_container_resonse returned by c_start_container.")
|
2022-10-21 12:48:19 -05:00
|
|
|
|
|
|
|
|
2022-10-17 17:18:17 -05:00
|
|
|
if __name__ == '__main__':
|
|
|
|
unittest.main()
|