forked from p34709852/monkey
parent
bb40560fc7
commit
2ecc683216
|
@ -1,2 +0,0 @@
|
||||||
class MonkeyNotFoundError(Exception):
|
|
||||||
pass
|
|
|
@ -5,7 +5,6 @@ import mongoengine
|
||||||
from mongoengine import Document, StringField, ListField, BooleanField, EmbeddedDocumentField, DateField, \
|
from mongoengine import Document, StringField, ListField, BooleanField, EmbeddedDocumentField, DateField, \
|
||||||
ReferenceField
|
ReferenceField
|
||||||
|
|
||||||
from monkey_island.cc.models.errors import MonkeyNotFoundError
|
|
||||||
from monkey_island.cc.models.monkey_ttl import MonkeyTtl
|
from monkey_island.cc.models.monkey_ttl import MonkeyTtl
|
||||||
|
|
||||||
|
|
||||||
|
@ -55,3 +54,7 @@ class Monkey(Document):
|
||||||
# Trying to dereference unknown document - the monkey is MIA.
|
# Trying to dereference unknown document - the monkey is MIA.
|
||||||
monkey_is_dead = True
|
monkey_is_dead = True
|
||||||
return monkey_is_dead
|
return monkey_is_dead
|
||||||
|
|
||||||
|
|
||||||
|
class MonkeyNotFoundError(Exception):
|
||||||
|
pass
|
||||||
|
|
|
@ -7,7 +7,7 @@ from unittest import TestCase
|
||||||
import mongomock
|
import mongomock
|
||||||
|
|
||||||
from monkey import Monkey
|
from monkey import Monkey
|
||||||
from monkey_island.cc.models.errors import MonkeyNotFoundError
|
from monkey_island.cc.models.monkey import MonkeyNotFoundError
|
||||||
from monkey_ttl import MonkeyTtl
|
from monkey_ttl import MonkeyTtl
|
||||||
|
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue