github.com/klaytn/klaytn@v1.10.2/datasync/chaindatafetcher/kas/repository_checkpoint_test.go (about) 1 // Copyright 2020 The klaytn Authors 2 // This file is part of the klaytn library. 3 // 4 // The klaytn library is free software: you can redistribute it and/or modify 5 // it under the terms of the GNU Lesser General Public License as published by 6 // the Free Software Foundation, either version 3 of the License, or 7 // (at your option) any later version. 8 // 9 // The klaytn library is distributed in the hope that it will be useful, 10 // but WITHOUT ANY WARRANTY; without even the implied warranty of 11 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 12 // GNU Lesser General Public License for more details. 13 // 14 // You should have received a copy of the GNU Lesser General Public License 15 // along with the klaytn library. If not, see <http://www.gnu.org/licenses/>. 16 17 package kas 18 19 import "strings" 20 21 func (s *SuiteRepository) TestCheckpoint_Success() { 22 // Clean up the table before test. 23 s.NoError(s.repo.db.Delete(FetcherMetadata{}).Error) 24 25 expected := int64(1912) 26 err := s.repo.WriteCheckpoint(expected) 27 s.NoError(err) 28 29 actual, err := s.repo.ReadCheckpoint() 30 s.NoError(err) 31 s.Equal(expected, actual) 32 } 33 34 func (s *SuiteRepository) TestCheckpoint_Fail_RecordNotFound() { 35 // Clean up the table before test. 36 s.NoError(s.repo.db.Delete(FetcherMetadata{}).Error) 37 38 // readCheckpoint returns an error if it is failed to read a checkpoint from database. 39 _, err := s.repo.readCheckpoint() 40 s.Error(err) 41 s.True(strings.Contains(err.Error(), "record not found")) 42 } 43 44 func (s *SuiteRepository) TestCheckpoint_Success_RecordNotFound() { 45 // Clean up the table before test. 46 s.NoError(s.repo.db.Delete(FetcherMetadata{}).Error) 47 48 // ReadCheckpoint filters "record not found" error and returns 0. 49 checkpoint, err := s.repo.ReadCheckpoint() 50 s.NoError(err) 51 s.Equal(int64(0), checkpoint) 52 }