def test_delete_cascade_tenant(self):
now = datetime.now()
tenant_model = TenantModel.objects.create(tenant_name="test_tenant",
created_at=now,
updated_at=now)
objects_create = NotificationDestinationModel.objects.create(
name="test", tenant=tenant_model, created_at=now, updated_at=now)
objects_create.save()
objects_all = NotificationDestinationModel.all()
self.assertEqual(objects_all.count(), 1)
tenant_model.delete()
model_objects_all = NotificationDestinationModel.all()
self.assertEqual(model_objects_all.count(), 0)
def test_update(self):
now = datetime.now()
tenant_model = TenantModel.objects.create(tenant_name="test_tenant",
created_at=now,
updated_at=now)
objects_create = NotificationDestinationModel.objects.create(
name="test", tenant=tenant_model, created_at=now, updated_at=now)
objects_create.save()
objects_all = NotificationDestinationModel.all()
self.assertEqual(objects_all.count(), 1)
notification_destination_model = objects_all[0]
notification_destination_model.name = "update"
notification_destination_model.save()
objects_get = NotificationDestinationModel.objects.get(name="update")
self.assertEqual(objects_get.name, "update")
def test_telephone(self):
now = datetime.now()
tenant_model = TenantModel.objects.create(
tenant_name="test_tenant",
created_at=now,
updated_at=now
)
objects_create = TelephoneDestination.objects.create(
name="test", tenant=tenant_model, phone_number="080-1234-5678", country_code=81, created_at=now, updated_at=now)
objects_create.save()
# 通知先として取得したときに区別できているか
objects_all = NotificationDestinationModel.all()
self.assertTrue(isinstance(objects_all[0], TelephoneDestination))
# Telephone通知先として取得したときに区別できているか
destination_objects_all = TelephoneDestination.all()
self.assertTrue(isinstance(destination_objects_all[0], TelephoneDestination))
def test_email(self):
now = datetime.now()
tenant_model = TenantModel.objects.create(
tenant_name="test_tenant",
created_at=now,
updated_at=now
)
objects_create = EmailDestination.objects.create(
name="test", tenant=tenant_model, address="*****@*****.**", created_at=now, updated_at=now)
objects_create.save()
# 通知先として取得したときに区別できているか
objects_all = NotificationDestinationModel.all()
self.assertTrue(isinstance(objects_all[0], EmailDestination))
# Email通知先として取得したときに区別できているか
destination_objects_all = EmailDestination.all()
self.assertTrue(isinstance(destination_objects_all[0], EmailDestination))
def test_notification_message(self):
tenant_model = TenantModel.objects.create(tenant_name="test_name",
email="*****@*****.**",
tel="03-1234-5678")
tenant_model.save()
aws_environment_model = AwsEnvironmentModel.objects.create(
aws_account_id="1234567890",
name="test_name",
aws_role="test_role",
aws_external_id="test_external",
tenant=tenant_model)
aws_environment_model.save()
alarm_message = {
"Region": "Asia Pacific (Tokyo)",
"Trigger": {
"Namespace":
"AWS/EC2",
"Dimensions": [{
"name": "InstanceId",
"value": "i-1234567890123456"
}],
"MetricName":
"NetworkOut"
},
"AlarmName": "NARUKO-EC2-i-1234567890123456-NetworkOut-DANGER",
"StateChangeTime": "2018-12-01T00:00:00.000+0000",
"AWSAccountId": "1234567890"
}
message = NotificationDestinationModel.NotificationMessage(
alarm_message)
self.assertEqual(message.resource.region, "Asia Pacific (Tokyo)")
self.assertEqual(message.resource.resource_id, "i-1234567890123456")
self.assertEqual(message.metric, "NetworkOut")
self.assertEqual(message.level, "危険")
self.assertEqual(message.aws.aws_account_id, "1234567890")
self.assertEqual(message.time, "2018年12月01日 09時00分00秒")
self.assertTrue(isinstance(message.resource, Ec2))