123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183 |
- // Copyright (C) 2018 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 https://mozilla.org/MPL/2.0/.
- package db
- import (
- "math/bits"
- "sort"
- "testing"
- "github.com/syncthing/syncthing/lib/events"
- "github.com/syncthing/syncthing/lib/protocol"
- )
- func TestEachFlagBit(t *testing.T) {
- cases := []struct {
- flags uint32
- iterations int
- }{
- {0, 0},
- {1<<0 | 1<<3, 2},
- {1 << 0, 1},
- {1 << 31, 1},
- {1<<10 | 1<<20 | 1<<30, 3},
- }
- for _, tc := range cases {
- var flags uint32
- iterations := 0
- eachFlagBit(tc.flags, func(f uint32) {
- iterations++
- flags |= f
- if bits.OnesCount32(f) != 1 {
- t.Error("expected exactly one bit to be set in every call")
- }
- })
- if flags != tc.flags {
- t.Errorf("expected 0x%x flags, got 0x%x", tc.flags, flags)
- }
- if iterations != tc.iterations {
- t.Errorf("expected %d iterations, got %d", tc.iterations, iterations)
- }
- }
- }
- func TestMetaDevices(t *testing.T) {
- d1 := protocol.DeviceID{1}
- d2 := protocol.DeviceID{2}
- meta := newMetadataTracker(nil, events.NoopLogger)
- meta.addFile(d1, protocol.FileInfo{Sequence: 1})
- meta.addFile(d1, protocol.FileInfo{Sequence: 2, LocalFlags: 1})
- meta.addFile(d2, protocol.FileInfo{Sequence: 1})
- meta.addFile(d2, protocol.FileInfo{Sequence: 2, LocalFlags: 2})
- meta.addFile(protocol.LocalDeviceID, protocol.FileInfo{Sequence: 1})
- // There are five device/flags combos
- if l := len(meta.counts.Counts); l < 5 {
- t.Error("expected at least five buckets, not", l)
- }
- // There are only two non-local devices
- devs := meta.devices()
- if l := len(devs); l != 2 {
- t.Fatal("expected two devices, not", l)
- }
- // Check that we got the two devices we expect
- sort.Slice(devs, func(a, b int) bool {
- return devs[a].Compare(devs[b]) == -1
- })
- if devs[0] != d1 {
- t.Error("first device should be d1")
- }
- if devs[1] != d2 {
- t.Error("second device should be d2")
- }
- }
- func TestMetaSequences(t *testing.T) {
- d1 := protocol.DeviceID{1}
- meta := newMetadataTracker(nil, events.NoopLogger)
- meta.addFile(d1, protocol.FileInfo{Sequence: 1})
- meta.addFile(d1, protocol.FileInfo{Sequence: 2, RawInvalid: true})
- meta.addFile(d1, protocol.FileInfo{Sequence: 3})
- meta.addFile(d1, protocol.FileInfo{Sequence: 4, RawInvalid: true})
- meta.addFile(protocol.LocalDeviceID, protocol.FileInfo{Sequence: 1})
- meta.addFile(protocol.LocalDeviceID, protocol.FileInfo{Sequence: 2})
- meta.addFile(protocol.LocalDeviceID, protocol.FileInfo{Sequence: 3, LocalFlags: 1})
- meta.addFile(protocol.LocalDeviceID, protocol.FileInfo{Sequence: 4, LocalFlags: 2})
- if seq := meta.Sequence(d1); seq != 4 {
- t.Error("sequence of first device should be 4, not", seq)
- }
- if seq := meta.Sequence(protocol.LocalDeviceID); seq != 4 {
- t.Error("sequence of first device should be 4, not", seq)
- }
- }
- func TestRecalcMeta(t *testing.T) {
- ldb := newLowlevelMemory(t)
- defer ldb.Close()
- // Add some files
- s1 := newFileSet(t, "test", ldb)
- files := []protocol.FileInfo{
- {Name: "a", Size: 1000},
- {Name: "b", Size: 2000},
- }
- s1.Update(protocol.LocalDeviceID, files)
- // Verify local/global size
- snap := snapshot(t, s1)
- ls := snap.LocalSize()
- gs := snap.GlobalSize()
- snap.Release()
- if ls.Bytes != 3000 {
- t.Fatalf("Wrong initial local byte count, %d != 3000", ls.Bytes)
- }
- if gs.Bytes != 3000 {
- t.Fatalf("Wrong initial global byte count, %d != 3000", gs.Bytes)
- }
- // Reach into the database to make the metadata tracker intentionally
- // wrong and out of date
- curSeq := s1.meta.Sequence(protocol.LocalDeviceID)
- tran, err := ldb.newReadWriteTransaction()
- if err != nil {
- t.Fatal(err)
- }
- s1.meta.mut.Lock()
- s1.meta.countsPtr(protocol.LocalDeviceID, 0).Sequence = curSeq - 1 // too low
- s1.meta.countsPtr(protocol.LocalDeviceID, 0).Bytes = 1234 // wrong
- s1.meta.countsPtr(protocol.GlobalDeviceID, 0).Bytes = 1234 // wrong
- s1.meta.dirty = true
- s1.meta.mut.Unlock()
- if err := s1.meta.toDB(tran, []byte("test")); err != nil {
- t.Fatal(err)
- }
- if err := tran.Commit(); err != nil {
- t.Fatal(err)
- }
- // Verify that our bad data "took"
- snap = snapshot(t, s1)
- ls = snap.LocalSize()
- gs = snap.GlobalSize()
- snap.Release()
- if ls.Bytes != 1234 {
- t.Fatalf("Wrong changed local byte count, %d != 1234", ls.Bytes)
- }
- if gs.Bytes != 1234 {
- t.Fatalf("Wrong changed global byte count, %d != 1234", gs.Bytes)
- }
- // Create a new fileset, which will realize the inconsistency and recalculate
- s2 := newFileSet(t, "test", ldb)
- // Verify local/global size
- snap = snapshot(t, s2)
- ls = snap.LocalSize()
- gs = snap.GlobalSize()
- snap.Release()
- if ls.Bytes != 3000 {
- t.Fatalf("Wrong fixed local byte count, %d != 3000", ls.Bytes)
- }
- if gs.Bytes != 3000 {
- t.Fatalf("Wrong fixed global byte count, %d != 3000", gs.Bytes)
- }
- }
- func TestMetaKeyCollisions(t *testing.T) {
- if protocol.LocalAllFlags&needFlag != 0 {
- t.Error("Collision between need flag and protocol local file flags")
- }
- }
|