Renamed collection from TTL to MonkeyTTL

monkey/monkey_island/cc/models/monkey.py

@@ -22,7 +22,7 @@ class Config(EmbeddedDocument):
 
 class Creds(EmbeddedDocument):
     """
-    TODO get an example of this data
+    TODO get an example of this data, and make it strict
     """
     meta = {'strict': False}
     pass
@@ -36,7 +36,7 @@ class PbaResults(EmbeddedDocument):
     result = ListField()
 
 
-class Ttl(Document):
+class MonkeyTtl(Document):
     meta = {
         'indexes': [
             {
@@ -72,7 +72,7 @@ class Monkey(Document):
     config_error = BooleanField()
     critical_services = ListField(StringField())
     pba_results = ListField()
-    ttl_ref = ReferenceField(Ttl)
+    ttl_ref = ReferenceField(MonkeyTtl)
 
     def is_dead(self):
         monkey_is_dead = False
@@ -80,7 +80,7 @@ class Monkey(Document):
             monkey_is_dead = True
         else:
             try:
-                if Ttl.objects(id=self.ttl_ref.id).count() == 0:
+                if MonkeyTtl.objects(id=self.ttl_ref.id).count() == 0:
                     # No TTLs - monkey has timed out. The monkey is MIA
                     monkey_is_dead = True
             except mongoengine.DoesNotExist:

monkey/monkey_island/cc/resources/monkey.py

@@ -48,7 +48,7 @@ class Monkey(flask_restful.Resource):
             tunnel_host_ip = monkey_json['tunnel'].split(":")[-2].replace("//", "")
             NodeService.set_monkey_tunnel(monkey["_id"], tunnel_host_ip)
 
-        current_ttl = models.monkey.Ttl(expire_at=datetime.now() + timedelta(seconds=30))
+        current_ttl = models.monkey.MonkeyTtl(expire_at=datetime.now() + timedelta(seconds=30))
         current_ttl.save()
 
         update['$set']['ttl_ref'] = current_ttl.id