2015-06-11 08:46:57 +02:00
|
|
|
// Copyright (C) 2015 The Syncthing Authors.
|
|
|
|
//
|
|
|
|
// This Source Code Form is subject to the terms of the Mozilla Public
|
|
|
|
// License, v. 2.0. If a copy of the MPL was not distributed with this file,
|
|
|
|
// You can obtain one at http://mozilla.org/MPL/2.0/.
|
|
|
|
|
2015-06-03 09:47:39 +02:00
|
|
|
package config
|
|
|
|
|
|
|
|
import (
|
|
|
|
"errors"
|
|
|
|
"testing"
|
|
|
|
)
|
|
|
|
|
2016-07-04 22:32:34 +02:00
|
|
|
type requiresRestart struct {
|
|
|
|
committed chan struct{}
|
|
|
|
}
|
2015-06-03 09:47:39 +02:00
|
|
|
|
|
|
|
func (requiresRestart) VerifyConfiguration(_, _ Configuration) error {
|
|
|
|
return nil
|
|
|
|
}
|
2016-07-04 22:32:34 +02:00
|
|
|
func (c requiresRestart) CommitConfiguration(_, _ Configuration) bool {
|
|
|
|
select {
|
|
|
|
case c.committed <- struct{}{}:
|
|
|
|
default:
|
|
|
|
}
|
2015-06-03 09:47:39 +02:00
|
|
|
return false
|
|
|
|
}
|
|
|
|
func (requiresRestart) String() string {
|
|
|
|
return "requiresRestart"
|
|
|
|
}
|
|
|
|
|
|
|
|
type validationError struct{}
|
|
|
|
|
|
|
|
func (validationError) VerifyConfiguration(_, _ Configuration) error {
|
|
|
|
return errors.New("some error")
|
|
|
|
}
|
2016-07-04 22:32:34 +02:00
|
|
|
func (c validationError) CommitConfiguration(_, _ Configuration) bool {
|
2015-06-03 09:47:39 +02:00
|
|
|
return true
|
|
|
|
}
|
|
|
|
func (validationError) String() string {
|
|
|
|
return "validationError"
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestReplaceCommit(t *testing.T) {
|
|
|
|
w := Wrap("/dev/null", Configuration{Version: 0})
|
|
|
|
if w.Raw().Version != 0 {
|
|
|
|
t.Fatal("Config incorrect")
|
|
|
|
}
|
|
|
|
|
|
|
|
// Replace config. We should get back a clean response and the config
|
|
|
|
// should change.
|
|
|
|
|
2016-07-04 22:32:34 +02:00
|
|
|
err := w.Replace(Configuration{Version: 1})
|
|
|
|
if err != nil {
|
|
|
|
t.Fatal("Should not have a validation error:", err)
|
2015-06-03 09:47:39 +02:00
|
|
|
}
|
2016-07-04 22:32:34 +02:00
|
|
|
if w.RequiresRestart() {
|
2015-06-03 09:47:39 +02:00
|
|
|
t.Fatal("Should not require restart")
|
|
|
|
}
|
2016-08-07 18:21:59 +02:00
|
|
|
if w.Raw().Version != CurrentVersion {
|
2015-06-03 09:47:39 +02:00
|
|
|
t.Fatal("Config should have changed")
|
|
|
|
}
|
|
|
|
|
|
|
|
// Now with a subscriber requiring restart. We should get a clean response
|
|
|
|
// but with the restart flag set, and the config should change.
|
|
|
|
|
2016-07-04 22:32:34 +02:00
|
|
|
sub0 := requiresRestart{committed: make(chan struct{}, 1)}
|
|
|
|
w.Subscribe(sub0)
|
2015-06-03 09:47:39 +02:00
|
|
|
|
2016-07-04 22:32:34 +02:00
|
|
|
err = w.Replace(Configuration{Version: 2})
|
|
|
|
if err != nil {
|
|
|
|
t.Fatal("Should not have a validation error:", err)
|
2015-06-03 09:47:39 +02:00
|
|
|
}
|
2016-07-04 22:32:34 +02:00
|
|
|
|
|
|
|
<-sub0.committed
|
|
|
|
if !w.RequiresRestart() {
|
2015-06-03 09:47:39 +02:00
|
|
|
t.Fatal("Should require restart")
|
|
|
|
}
|
2016-08-07 18:21:59 +02:00
|
|
|
if w.Raw().Version != CurrentVersion {
|
2015-06-03 09:47:39 +02:00
|
|
|
t.Fatal("Config should have changed")
|
|
|
|
}
|
|
|
|
|
|
|
|
// Now with a subscriber that throws a validation error. The config should
|
|
|
|
// not change.
|
|
|
|
|
|
|
|
w.Subscribe(validationError{})
|
|
|
|
|
2016-07-04 22:32:34 +02:00
|
|
|
err = w.Replace(Configuration{Version: 3})
|
|
|
|
if err == nil {
|
2015-06-03 09:47:39 +02:00
|
|
|
t.Fatal("Should have a validation error")
|
|
|
|
}
|
2016-07-04 22:32:34 +02:00
|
|
|
if !w.RequiresRestart() {
|
|
|
|
t.Fatal("Should still require restart")
|
2015-06-03 09:47:39 +02:00
|
|
|
}
|
2016-08-07 18:21:59 +02:00
|
|
|
if w.Raw().Version != CurrentVersion {
|
2015-06-03 09:47:39 +02:00
|
|
|
t.Fatal("Config should not have changed")
|
|
|
|
}
|
|
|
|
}
|