add newly-required field to test data

18F · Nov 30, 2016 · e04b58f · e04b58f
1 parent 0ee80d7
commit e04b58f
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/projects/tests/test_field_locking.py b/projects/tests/test_field_locking.py
@@ -2,10 +2,11 @@
 from datetime import date
 from django.core.exceptions import ValidationError
 from projects.models import Buy, Project
-from projects.factories import BuyFactory, UserFactory, ProjectFactory, \
+from projects.factories import BuyFactory, ProjectFactory, \
                                ContractingOfficeFactory, \
                                ContractingOfficerFactory, \
-                               ContractingSpecialistFactory
+                               ContractingSpecialistFactory, \
+                               ContractingOfficerRepresentativeFactory
 
 
 class TestLocking:
@@ -22,6 +23,7 @@ def buy_plus(self, buy):
         buy.contracting_office = ContractingOfficeFactory()
         buy.contracting_officer = ContractingOfficerFactory()
         buy.contracting_specialist = ContractingSpecialistFactory()
+        buy.contracting_officer_representative = ContractingOfficerRepresentativeFactory()
         buy.base_period_length = '3 months'
         buy.option_periods = 3
         buy.option_period_length = '3 months'