def test_get_play_by_uuid(self):
play = factories.PlayFactory(
name="play1", uuid="6b838b6f-cfc7-4e11-a264-73df8683ee0e")
factories.PlayFactory(name="play2")
request = self.client.get(
"/api/v1/plays?uuid=6b838b6f-cfc7-4e11-a264-73df8683ee0e")
self.assertEqual(1, len(request.data["results"]))
self.assertEqual(play.name, request.data["results"][0]["name"])
def test_get_plays_by_name(self):
# Create a playbook and two plays
playbook = factories.PlaybookFactory()
play = factories.PlayFactory(name="first_play", playbook=playbook)
factories.PlayFactory(name="second_play", playbook=playbook)
# Query for the first play name and expect one result
request = self.client.get("/api/v1/plays?name=%s" % play.name)
self.assertEqual(1, len(request.data["results"]))
self.assertEqual(play.name, request.data["results"][0]["name"])
def test_partial_update_play(self):
play = factories.PlayFactory()
self.assertNotEqual("update", play.name)
request = self.client.patch("/api/v1/plays/%s" % play.id, {"name": "update"})
self.assertEqual(200, request.status_code)
play_updated = models.Play.objects.get(id=play.id)
self.assertEqual("update", play_updated.name)
def test_get_play_duration(self):
started = timezone.now()
ended = started + datetime.timedelta(hours=1)
play = factories.PlayFactory(started=started, ended=ended)
request = self.client.get("/api/v1/plays/%s" % play.id)
self.assertEqual(parse_duration(request.data["duration"]),
ended - started)
def test_update_wrong_play_status(self):
play = factories.PlayFactory()
self.assertNotEqual("wrong", play.status)
request = self.client.patch("/api/v1/plays/%s" % play.id, {"status": "wrong"})
self.assertEqual(400, request.status_code)
play_updated = models.Play.objects.get(id=play.id)
self.assertNotEqual("wrong", play_updated.status)
def test_expired_play(self):
play = factories.PlayFactory(status="running")
self.assertEqual("running", play.status)
request = self.client.patch("/api/v1/plays/%s" % play.id, {"status": "expired"})
self.assertEqual(200, request.status_code)
play_updated = models.Play.objects.get(id=play.id)
self.assertEqual("expired", play_updated.status)
def test_get_play_by_status(self):
play = factories.PlayFactory(status="running")
factories.PlayFactory(status="completed")
factories.PlayFactory(status="unknown")
# Confirm we have three objects
request = self.client.get("/api/v1/plays")
self.assertEqual(3, len(request.data["results"]))
# Test single status
request = self.client.get("/api/v1/plays?status=running")
self.assertEqual(1, len(request.data["results"]))
self.assertEqual(play.status, request.data["results"][0]["status"])
# Test multiple status
request = self.client.get("/api/v1/plays?status=running&status=completed")
self.assertEqual(2, len(request.data["results"]))
def test_get_tasks_by_play(self):
playbook = factories.PlaybookFactory()
# Create two plays
play = factories.PlayFactory(name="first_play", playbook=playbook)
second_play = factories.PlayFactory(name="second_play",
playbook=playbook)
request = self.client.get("/api/v1/plays")
self.assertEqual(2, len(request.data["results"]))
# Add tasks for the first play and validate
task = factories.TaskFactory(name="inside_first", play=play)
factories.TaskFactory(name="inside_second", play=play)
request = self.client.get("/api/v1/tasks?play=%s" % play.id)
self.assertEqual(2, len(request.data["results"]))
self.assertEqual(task.name, request.data["results"][1]["name"])
self.assertEqual("inside_second", request.data["results"][0]["name"])
request = self.client.get("/api/v1/tasks?play=%s" % second_play.id)
self.assertEqual(0, len(request.data["results"]))
def test_get_play_order(self):
old_started = timezone.now() - datetime.timedelta(hours=12)
old_ended = old_started + datetime.timedelta(minutes=30)
old_play = factories.PlayFactory(started=old_started, ended=old_ended)
new_started = timezone.now() - datetime.timedelta(hours=6)
new_ended = new_started + datetime.timedelta(hours=1)
new_play = factories.PlayFactory(started=new_started, ended=new_ended)
# Ensure we have two objects
request = self.client.get("/api/v1/plays")
self.assertEqual(2, len(request.data["results"]))
order_fields = ["id", "created", "updated", "started", "ended", "duration"]
# Ascending order
for field in order_fields:
request = self.client.get("/api/v1/plays?order=%s" % field)
self.assertEqual(request.data["results"][0]["id"], old_play.id)
# Descending order
for field in order_fields:
request = self.client.get("/api/v1/plays?order=-%s" % field)
self.assertEqual(request.data["results"][0]["id"], new_play.id)
def test_get_play_by_date(self):
play = factories.PlayFactory()
past = datetime.datetime.now() - datetime.timedelta(hours=12)
negative_date_fields = ["created_before", "started_before", "updated_before"]
positive_date_fields = ["created_after", "started_after", "updated_after"]
# Expect no play when searching before it was created
for field in negative_date_fields:
request = self.client.get("/api/v1/plays?%s=%s" % (field, past.isoformat()))
self.assertEqual(request.data["count"], 0)
# Expect a play when searching after it was created
for field in positive_date_fields:
request = self.client.get("/api/v1/plays?%s=%s" % (field, past.isoformat()))
self.assertEqual(request.data["count"], 1)
self.assertEqual(request.data["results"][0]["id"], play.id)
def test_create_task(self):
play = factories.PlayFactory()
file = factories.FileFactory()
self.assertEqual(0, models.Task.objects.count())
request = self.client.post(
"/api/v1/tasks",
{
"name": "create",
"action": "test",
"lineno": 2,
"handler": False,
"status": "running",
"play": play.id,
"file": file.id,
"playbook": play.playbook.id,
},
)
self.assertEqual(201, request.status_code)
self.assertEqual(1, models.Task.objects.count())
def test_task_serializer_compress_tags(self):
play = factories.PlayFactory()
file = factories.FileFactory()
serializer = serializers.TaskSerializer(
data={
"name": "compress",
"action": "test",
"lineno": 2,
"status": "running",
"handler": False,
"play": play.id,
"file": file.id,
"tags": factories.TASK_TAGS,
"playbook": play.playbook.id,
})
serializer.is_valid()
task = serializer.save()
task.refresh_from_db()
self.assertEqual(task.tags, utils.compressed_obj(factories.TASK_TAGS))
def test_task_serializer(self):
play = factories.PlayFactory()
file = factories.FileFactory()
serializer = serializers.TaskSerializer(
data={
"name": "serializer",
"action": "test",
"lineno": 2,
"status": "completed",
"handler": False,
"play": play.id,
"file": file.id,
"playbook": play.playbook.id,
})
serializer.is_valid()
task = serializer.save()
task.refresh_from_db()
self.assertEqual(task.name, "serializer")
self.assertEqual(task.status, "completed")
def test_get_play_by_playbook(self):
play = factories.PlayFactory(name="play1")
factories.PlayFactory(name="play2")
request = self.client.get("/api/v1/plays?playbook=1")
self.assertEqual(1, len(request.data["results"]))
self.assertEqual(play.name, request.data["results"][0]["name"])
def test_get_play(self):
play = factories.PlayFactory()
request = self.client.get("/api/v1/plays/%s" % play.id)
self.assertEqual(play.name, request.data["name"])
def test_delete_play(self):
play = factories.PlayFactory()
self.assertEqual(1, models.Play.objects.all().count())
request = self.client.delete("/api/v1/plays/%s" % play.id)
self.assertEqual(204, request.status_code)
self.assertEqual(0, models.Play.objects.all().count())
def test_get_plays(self):
play = factories.PlayFactory()
request = self.client.get("/api/v1/plays")
self.assertEqual(1, len(request.data["results"]))
self.assertEqual(play.name, request.data["results"][0]["name"])
def test_play_factory(self):
play = factories.PlayFactory(name="play factory")
self.assertEqual(play.name, "play factory")
def test_get_playbook_arguments(self):
play = factories.PlayFactory()
request = self.client.get("/api/v1/plays/%s" % play.id)
self.assertIn("inventory", request.data["playbook"]["arguments"])
