func (s *EnsurePasswordSuite) tearDownKey(c *gc.C, shouldBeDeleted bool) { err := s.k.DeleteValue(s.tempRegEntry) if shouldBeDeleted { c.Assert(err, gc.ErrorMatches, utils.NoSuchFileErrRegexp) } else { c.Assert(err, jc.ErrorIsNil) } s.k.Close() err = registry.DeleteKey(registry.LOCAL_MACHINE, s.tempRegKey[6:]) c.Assert(err, jc.ErrorIsNil) }
func (s *EnsurePasswordSuite) setUpKey(c *gc.C) { k, exist, err := registry.CreateKey(registry.LOCAL_MACHINE, s.tempRegKey[6:], registry.ALL_ACCESS) // If we get in here it means cleanup failed at some point earlier if exist { err = registry.DeleteKey(registry.LOCAL_MACHINE, s.tempRegKey[6:]) k, exist, err = registry.CreateKey(registry.LOCAL_MACHINE, s.tempRegKey[6:], registry.ALL_ACCESS) } c.Assert(err, jc.ErrorIsNil) c.Assert(exist, jc.IsFalse) s.k = k }
func (s *windowsSeriesSuite) createRegKey(c *gc.C, key *string) { salt, err := utils.RandomPassword() c.Assert(err, jc.ErrorIsNil) regKey := fmt.Sprintf(`SOFTWARE\JUJU\%s`, salt) s.PatchValue(key, regKey) k, _, err := registry.CreateKey(registry.LOCAL_MACHINE, *key, registry.ALL_ACCESS) c.Assert(err, jc.ErrorIsNil) err = k.Close() c.Assert(err, jc.ErrorIsNil) s.AddCleanup(func(*gc.C) { registry.DeleteKey(registry.LOCAL_MACHINE, *series.CurrentVersionKey) }) }
func (s *windowsVersionSuite) SetUpTest(c *gc.C) { s.BaseSuite.SetUpTest(c) salt, err := utils.RandomPassword() c.Assert(err, jc.ErrorIsNil) regKey := fmt.Sprintf(`SOFTWARE\JUJU\%s`, salt) s.PatchValue(version.CurrentVersionKey, regKey) k, _, err := registry.CreateKey(registry.LOCAL_MACHINE, *version.CurrentVersionKey, registry.ALL_ACCESS) c.Assert(err, jc.ErrorIsNil) err = k.Close() c.Assert(err, jc.ErrorIsNil) s.AddCleanup(func(*gc.C) { registry.DeleteKey(registry.LOCAL_MACHINE, *version.CurrentVersionKey) }) }
// tearDownFeatureFlags restores the old registry values func (s *JujuOSEnvSuite) tearDownFeatureFlags(c *gc.C) { regKey := osenv.JujuRegistryKey[len(`HKLM:\`):] if s.regKeyExisted { k, err := registry.OpenKey(registry.LOCAL_MACHINE, regKey, registry.ALL_ACCESS) if s.regEntryExisted { err := k.SetStringValue(osenv.JujuFeatureFlagEnvKey, s.oldRegEntryValue) c.Assert(err, jc.ErrorIsNil) } else { err := k.DeleteValue(osenv.JujuFeatureFlagEnvKey) c.Assert(err, jc.ErrorIsNil) } err = k.Close() c.Assert(err, jc.ErrorIsNil) } else { err := registry.DeleteKey(registry.LOCAL_MACHINE, regKey) c.Assert(err, jc.ErrorIsNil) } }
func (s *flagWinSuite) TearDownTest(c *gc.C) { s.k.DeleteValue("JUJU_TESTING_FEATURE") s.k.Close() registry.DeleteKey(registry.LOCAL_MACHINE, regKey[6:]) }