Coverage for crateweb/research/tests/factories.py: 100%

1""" 

2crate_anon/crateweb/research/tests/factories.py 

3 

4=============================================================================== 

5 

6 Copyright (C) 2015, University of Cambridge, Department of Psychiatry. 

7 Created by Rudolf Cardinal (rnc1001@cam.ac.uk). 

8 

9 This file is part of CRATE. 

10 

11 CRATE is free software: you can redistribute it and/or modify 

12 it under the terms of the GNU General Public License as published by 

13 the Free Software Foundation, either version 3 of the License, or 

14 (at your option) any later version. 

15 

16 CRATE is distributed in the hope that it will be useful, 

17 but WITHOUT ANY WARRANTY; without even the implied warranty of 

18 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 

19 GNU General Public License for more details. 

20 

21 You should have received a copy of the GNU General Public License 

22 along with CRATE. If not, see <https://www.gnu.org/licenses/>. 

23 

24=============================================================================== 

25 

26Factory Boy test factories. 

27 

28""" 

29 

30import factory 

31 

32from django.conf import settings 

33 

34from crate_anon.crateweb.research.models import ( 

35 PatientExplorer, 

36 Query, 

37 SitewideQuery, 

38) 

39 

40 

41class UserFactory(factory.django.DjangoModelFactory): 

42 class Meta: 

43 model = settings.AUTH_USER_MODEL 

44 

45 username = factory.Sequence(lambda n: f"user-{n}") 

46 

47 

48class QueryFactory(factory.django.DjangoModelFactory): 

49 class Meta: 

50 model = Query 

51 

52 user = factory.SubFactory(UserFactory) 

53 

54 

55class SitewideQueryFactory(factory.django.DjangoModelFactory): 

56 class Meta: 

57 model = SitewideQuery 

58 

59 

60class PatientExplorerFactory(factory.django.DjangoModelFactory): 

61 class Meta: 

62 model = PatientExplorer 

63 

64 user = factory.SubFactory(UserFactory)