def test_translate_highlevel_save_bundle_modifs_only(self): """ Ensures that a Bundle is saved successfully to the DB when all the messages are there already. :return: """ # Create some bundles to attach. b = models.Bundle("all_ALL", "ALL") b.app = self.tapp m1 = models.Message("message1", "First Message") m2 = models.Message("message2", "Second Message") b.messages.append(m1) b.messages.append(m2) db.session.add(b) db.session.commit() bundle = ops_highlevel.load_bundle(self.tapp, "all_ALL", "ALL") bundle.add_msg("message1", "First Message Modified") bundle.add_msg("message2", "Second Message Modified") ops_highlevel.save_bundle(self.tapp, bundle) # Check that it was saved properly. m1 = db.session.query(models.Message).filter_by(bundle=b, key="message1").first() self.assertEquals("First Message Modified", m1.value) m2 = db.session.query(models.Message).filter_by(bundle=b, key="message2").first() self.assertEquals("Second Message Modified", m2.value) msgs = db.session.query(models.Message).filter_by(bundle=b).all() self.assertEquals(2, len(msgs))
def test_translate_highlevel_load_bundle(self): """ Ensures that if the bundle does exist it is retrieved successfully. :return: """ # Create some bundles to attach. b = models.Bundle("all_ALL", "ALL") b.app = self.tapp m1 = models.Message("message1", "First Message") m2 = models.Message("message2", "Second Message") b.messages.append(m1) b.messages.append(m2) db.session.add(b) db.session.commit() bundle = ops_highlevel.load_bundle(self.tapp, "all_ALL", "ALL") self.assertIs(type(bundle), bundles.Bundle) self.assertEquals(2, len(bundle.get_msgs())) self.assertEquals(bundle.get_msg("message1"), "First Message") self.assertEquals(bundle.get_msg("message2"), "Second Message")