Blame src/python-problem/tests/util.py

Packit Service 8a8a03
import datetime
Packit Service 8a8a03
Packit Service 8a8a03
import problem
Packit Service 8a8a03
Packit Service 8a8a03
class FakeProxy(object):
Packit Service 8a8a03
    data = dict()
Packit Service 8a8a03
Packit Service 8a8a03
    def get_item(self, dump_dir, name):
Packit Service 8a8a03
        if dump_dir not in self.data:
Packit Service 8a8a03
            raise problem.exception.InvalidProblem()
Packit Service 8a8a03
Packit Service 8a8a03
        try:
Packit Service 8a8a03
            return self.data[dump_dir][name]
Packit Service 8a8a03
        except KeyError:
Packit Service 8a8a03
            return None
Packit Service 8a8a03
Packit Service 8a8a03
    def set_item(self, dump_dir, name, value):
Packit Service 8a8a03
        self.data[dump_dir][name] = value
Packit Service 8a8a03
Packit Service 8a8a03
    def del_item(self, dump_dir, name):
Packit Service 8a8a03
        del self.data[dump_dir][name]
Packit Service 8a8a03
Packit Service 8a8a03
    def create(self, problem_dict):
Packit Service 8a8a03
        datestr = str(datetime.datetime.now()).replace(' ', '-')
Packit Service 8a8a03
        name = '{0}-{1}'.format(problem_dict['type'], datestr)
Packit Service 8a8a03
        self.data[name] = problem_dict
Packit Service 8a8a03
        return name
Packit Service 8a8a03
Packit Service 8a8a03
    def delete(self, dump_dir):
Packit Service 8a8a03
        del self.data[dump_dir]
Packit Service 8a8a03
Packit Service 8a8a03
    def list(self):
Packit Service 8a8a03
        return self.data.keys()
Packit Service 8a8a03
Packit Service 8a8a03
    def list_all(self):
Packit Service 8a8a03
        return self.data.keys()