def test_add_remove(): # Original state: # pc has c1 and c2 # pc2 has c3 # c4 is not with any pc user = User.query.filter().first() values = yaml2json('pc-components.db') pc = values['device'] c1, c2 = (d.Component(**c) for c in values['components']) pc = d.Desktop(**pc, components=OrderedSet([c1, c2])) db.session.add(pc) c3 = d.Component(serial_number='nc1', owner_id=user.id) pc2 = d.Desktop(serial_number='s2', components=OrderedSet([c3]), chassis=ComputerChassis.Microtower) c4 = d.Component(serial_number='c4s', owner_id=user.id) db.session.add(pc2) db.session.add(c4) db.session.commit() # Test: # pc has only c3 actions = Sync.add_remove(device=pc, components={c3, c4}) db.session.add_all(actions) db.session.commit() # We enforce the appliance of order_by assert len(actions) == 1 assert isinstance(actions[0], Remove) assert actions[0].device == pc2 assert actions[0].components == OrderedSet([c3])
def test_add_remove(): # Original state: # pc has c1 and c2 # pc2 has c3 # c4 is not with any pc values = file('pc-components.db') pc = values['device'] c1, c2 = (Component(**c) for c in values['components']) pc = Computer(**pc, components=OrderedSet([c1, c2])) db.session.add(pc) c3 = Component(serial_number='nc1') pc2 = Computer(serial_number='s2', components=OrderedSet([c3])) c4 = Component(serial_number='c4s') db.session.add(pc2) db.session.add(c4) db.session.commit() # Test: # pc has only c3 events = Sync.add_remove(device=pc, components={c3, c4}) db.session.add_all(events) db.session.commit() # We enforce the appliance of order_by assert len(events) == 1 assert isinstance(events[0], Remove) assert events[0].device == pc2 assert events[0].components == OrderedSet([c3])