Coverage for backend/flowsheetInternals/propertyPackages/serializers/SimulationObjectPropertyPackageSerializer.py: 69%

1from rest_framework import serializers 

2from flowsheetInternals.propertyPackages.models.SimulationObjectPropertyPackages import SimulationObjectPropertyPackages 

3from flowsheetInternals.propertyPackages.models.StreamFactory import StreamFactory 

4 

5class SimulationObjectPropertyPackagesSerializer(serializers.ModelSerializer): 

6 

7 class Meta: 

8 model = SimulationObjectPropertyPackages 

9 fields = '__all__' 

10 read_only_fields = ['name', 'id', 'unitOp', 'connectorPort'] # only allow which property package to be updated 

11 

12 def update(self, instance, validated_data): 

13 updated_instance = super().update(instance, validated_data) 

14 factory = StreamFactory(updated_instance) 

15 factory.check_and_update_stream() 

16 return updated_instance