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