def create_cluster_model(creator=None, cpus=1, memory=1):
if creator is None:
creator = create_user_model()
cluster = Cluster(creator=creator, cpus=cpus, memory=memory)
cluster.save()
return cluster
def test_cluster_invalid_without_creator():
with pytest.raises(ValidationError) as error:
cluster = Cluster(creator=None, cpus=1, memory=1)
cluster.save()
assert [*error.value.error_dict.keys()] == ['creator']
