def test_is_image_old(self): janitor = Janitor(etcd_client=EtcdMockCorrect(), docker_client=DockerMock()) self.assertFalse( janitor.is_image_old({ 'name': 'shiny-service', 'version': 20 })) self.assertFalse( janitor.is_image_old({ 'name': 'shiny-service', 'version': '19' })) self.assertFalse( janitor.is_image_old({ 'name': 'shiny-service', 'version': 18 })) self.assertTrue( janitor.is_image_old({ 'name': 'shiny-service', 'version': 17 })) self.assertTrue( janitor.is_image_old({ 'name': 'shiny-service', 'version': 0 })) self.assertTrue( janitor.is_image_old({ 'name': 'shiny-service', 'version': '0' }))
def test_delete_unused_images(self): dockermock = DockerMock() janitor = Janitor(etcd_client=EtcdMockCorrect(), docker_client=dockermock, delete_images=1) count = len(dockermock.images()) tags = 0 for image in dockermock.images(): tags = tags + len(image['RepoTags']) self.assertEqual(tags, 20) janitor.delete_unused_images() tags = 0 for image in dockermock.images(): tags = tags + len(image['RepoTags']) self.assertEqual(tags, 13) self.assertEqual(count - 5, len(dockermock.images()))
def test_delete_image(self): janitor = Janitor(etcd_client=EtcdMockCorrect(), docker_client=DockerMock()) image = {'Id': 'myid'} self.assertTrue(janitor.delete_image(image)) janitor = Janitor(etcd_client=EtcdMockCorrect(), docker_client=DockerMock(), delete_images=True) image = {'Id': 'myid', 'RepoTags': []} self.assertFalse(janitor.delete_image(image))
def test_parse_tag(self): janitor = Janitor(etcd_client=EtcdMockCorrect(), docker_client=DockerMock()) self.assertEqual(janitor.parse_tag(''), None) self.assertEqual(janitor.parse_tag('host:v1'), None) self.assertEqual(janitor.parse_tag('host/name:v1'), { 'host': 'host', 'name': 'name', 'version': '1' }) self.assertEqual(janitor.parse_tag('h-o-s-t/name1:v12'), { 'host': 'h-o-s-t', 'name': 'name1', 'version': '12' })
def test_get_current_apps_empty_etcd_catalog(self): janitor = Janitor(etcd_client=EtcdMockEmpty(), docker_client=DockerMock()) self.assertEqual({}, janitor.get_current_apps())
import discord from discord.ext import commands import os from dotenv import load_dotenv from janitor import Janitor, config from filereader import filereader load_dotenv() janitor = Janitor() inifile = filereader("bot.ini") config = config.config botclient = commands.Bot(command_prefix=config.get('General', 'Sign')) content = {} def addcomands(): inifile.readbetween("[CustomCommands]", "[EOFCustomCommands]") for line in inifile.commands: line = line.split("=") line = [linecontent.strip() for linecontent in line] content[line[0]] = line[1] print("Content added") async def process_own_commands(message): for key in content: if message.content == config.get('General', 'Sign') + key: if "@author" in content[key]: await message.channel.send(message.author.mention + content[key].strip("@author"))