Coverage for manila/db/migrations/alembic/versions/293fac1130ca_add_replication_attrs.py: 100%

1# Copyright 2015 Goutham Pacha Ravi. 

2# All Rights Reserved. 

3# Licensed under the Apache License, Version 2.0 (the "License"); you may 

4# not use this file except in compliance with the License. You may obtain 

5# a copy of the License at 

6# 

7# http://www.apache.org/licenses/LICENSE-2.0 

8# 

9# Unless required by applicable law or agreed to in writing, software 

10# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT 

11# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the 

12# License for the specific language governing permissions and limitations 

13# under the License. 

14 

15"""Add replication attributes to Share and ShareInstance models. 

16 

17Revision ID: 293fac1130ca 

18Revises: 344c1ac4747f 

19Create Date: 2015-09-10 15:45:07.273043 

20 

21""" 

22 

23# revision identifiers, used by Alembic. 

24revision = '293fac1130ca' 

25down_revision = '344c1ac4747f' 

26 

27from alembic import op 

28import sqlalchemy as sa 

29 

30 

31def upgrade(): 

32 """Add replication attributes to Shares and ShareInstances.""" 

33 op.add_column('shares', sa.Column('replication_type', sa.String(255))) 

34 op.add_column('share_instances', 

35 sa.Column('replica_state', sa.String(255))) 

36 

37 

38def downgrade(): 

39 """Remove replication attributes from Shares and ShareInstances.""" 

40 op.drop_column('shares', 'replication_type') 

41 op.drop_column('share_instances', 'replica_state')