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"))