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

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