How to use loadFixtures method of dbfixtures Package

Best Venom code snippet using dbfixtures.loadFixtures

server_test.go

Source:server_test.go Github

copy

Full Screen

...67 // Discard logging from the application to focus on test logs68 // NOTE: ConsoleLog MUST be false otherwise this will be overriden69 logger.Discard()70 // Load the fixtures then regenerate the test database if requested or required.71 require.NoError(s.loadFixtures())72 if _, err := os.Stat(dbtgz); *update || os.IsNotExist(err) {73 require.NoError(s.generateDB())74 }75 // Create the initial configuration, database fixture, and servers for the tests76 require.NoError(s.setupConfig())77 require.NoError(s.extractDB())78 require.NoError(s.setupServers())79 require.NoError(s.setupRemoteTrtl())80}81func (s *trtlTestSuite) TearDownSuite() {82 require := s.Require()83 require.NoError(s.cleanup())84 logger.ResetLogger()85}86//===========================================================================87// Server Tests88//===========================================================================89func TestTrtl(t *testing.T) {90 suite.Run(t, new(trtlTestSuite))91}92func (s *trtlTestSuite) TestMaintenance() {93 // Becasue we're modifying the configuration, ensure we reset the test environment94 defer s.reset()95 require := s.Require()96 s.conf.Maintenance = true97 server, err := trtl.New(*s.conf)98 require.NotEmpty(server, "no maintenance mode server was returned")99 require.NoError(err, "starting the server in maintenance mode caused an error")100 require.Nil(server.GetDB(), "maintenance mode database was not nil")101}102//===========================================================================103// Test Assertions104//===========================================================================105// StatusError is a helper assertion function that checks a gRPC status error106func (s *trtlTestSuite) StatusError(err error, code codes.Code, theError string) {107 require := s.Require()108 require.Error(err, "no status error returned")109 var serr *status.Status110 serr, ok := status.FromError(err)111 require.True(ok, "error is not a grpc status error")112 require.Equal(code, serr.Code(), "status code does not match")113 require.Equal(theError, serr.Message(), "status error message does not match")114}115// EqualMeta is a helper assertion function that checks if the actual metadata matches116// expectations about what the version should be. This helper function relies on some117// test fixture information to minimize what test users must supply.118func (s *trtlTestSuite) EqualMeta(expectedKey []byte, expectedNamespace string, expectedVersion, expectedParent *pb.Version, actualMeta *pb.Meta) {119 require := s.Require()120 require.NotNil(actualMeta, "cannot compare actual meta to expectations since actual meta is nil")121 expectedMeta := &pb.Meta{122 Key: expectedKey,123 Namespace: expectedNamespace,124 Region: s.conf.Replica.Region,125 Owner: fmt.Sprintf("%d:%s", s.conf.Replica.PID, s.conf.Replica.Name),126 Version: expectedVersion,127 Parent: expectedParent,128 }129 // Protocol buffer sanity check, this protects us from the case where the protocol buffers130 // have changed but the tests haven't been updated.131 if proto.Equal(actualMeta, expectedMeta) {132 // If the protocol buffers are equal, then we expect that everything is ok133 return134 }135 // At this point it's up to us to determine what the difference is between the versions136 require.Equal(expectedMeta.Key, actualMeta.Key, "meta keys do not match")137 require.Equal(expectedMeta.Namespace, actualMeta.Namespace, "meta namespace does not match")138 require.Equal(expectedMeta.Region, actualMeta.Region, "meta regions do not match")139 require.Equal(expectedMeta.Owner, actualMeta.Owner, "meta owners do not match")140 s.EqualVersion(expectedMeta.Version, actualMeta.Version, "version")141 s.EqualVersion(expectedMeta.Parent, actualMeta.Parent, "parent")142 // Could not determine the difference in the protocol buffers, so error generically143 require.True(proto.Equal(actualMeta, expectedMeta), "actual and expected protocol buffer metadata does not match")144}145// EqualVersion compares to Versions to see if they are the same146func (s *trtlTestSuite) EqualVersion(expectedVersion, actualVersion *pb.Version, versionType string) {147 require := s.Require()148 if expectedVersion == nil {149 require.Nil(actualVersion, "expected %s is nil but actual %s is not", versionType, versionType)150 return151 }152 require.Equal(expectedVersion.Pid, actualVersion.Pid, "expected %s PID does not match actual %s PID", versionType, versionType)153 require.Equal(expectedVersion.Version, actualVersion.Version, "expected %s scalar does not match actual %s scalar", versionType, versionType)154 require.Equal(expectedVersion.Region, actualVersion.Region, "expected %s region does not match actual %s region", versionType, versionType)155}156//===========================================================================157// Test setup helpers158//===========================================================================159// Creates a valid config for the tests so long as the current config is empty160func (s *trtlTestSuite) setupConfig() (err error) {161 if s.conf != nil || s.tmpdb != "" {162 return errors.New("cannot create configuration, run test suite cleanup first")163 }164 // Create a tmp directory for the database165 if s.tmpdb, err = ioutil.TempDir("testdata", "trtldb-*"); err != nil {166 return fmt.Errorf("could not create tmpdb: %s", err)167 }168 // Create the configuration without loading it from the environment169 conf := mock.Config()170 conf.Database.URL = fmt.Sprintf("leveldb:///%s", s.tmpdb)171 // Mark as processed since the config wasn't loaded from the environment172 if conf, err = conf.Mark(); err != nil {173 return fmt.Errorf("could not validate test configuration: %s", err)174 }175 // Set the configuration as a pointer so individual tests can modify the config as needed176 s.conf = &conf177 return nil178}179// creates and runs all of the trtl services in preparation for testing180func (s *trtlTestSuite) setupServers() (err error) {181 if s.conf == nil {182 return errors.New("no configuration, run s.setupConfig first")183 }184 // Create the trtl server185 if s.trtl, err = trtl.New(*s.conf); err != nil {186 return fmt.Errorf("could not create trtl service: %s", err)187 }188 // Create a bufconn listener(s) so that there are no actual network requests189 s.grpc = bufconn.New(bufSize, "")190 // Run the test server without signals, background routines or maintenance mode checks191 // TODO: do we need to check if there was an error when starting run?192 go s.trtl.Run(s.grpc.Listener)193 return nil194}195// creates and serves the RemoteTrtl server over bufconn using TLS credentials.196// Connections to trtl are always over mTLS, so the purpose of the remote trtl server197// is to have a peer that the tests can establish realistic connections to. In order to198// do this, the testdata directory contains two self-signed certificates generated by199// the certs CLI command. It doesn't matter which certificate the remote trtl server200// uses, but tests should use the opposite one to be able to connect over mTLS to the201// remote.202func (s *trtlTestSuite) setupRemoteTrtl() (err error) {203 // Create the server TLS configuration from the fixture204 var tls *tls.Config205 conf := config.MTLSConfig{206 ChainPath: serverCerts,207 CertPath: serverCerts,208 }209 if tls, err = conf.ParseTLSConfig(); err != nil {210 return fmt.Errorf("could not parse tls config: %s", err)211 }212 // Create the grpc server options with TLS213 opts := make([]grpc.ServerOption, 0)214 opts = append(opts, grpc.Creds(credentials.NewTLS(tls)))215 // Create the remote peer216 s.remote = mock.New(bufconn.New(bufSize, serverTarget), opts...)217 return nil218}219//===========================================================================220// Test cleanup helpers221//===========================================================================222// cleanup the current temporary directory, configuration, and running services.223func (s *trtlTestSuite) cleanup() (err error) {224 // Shutdown the trtl server if it is running225 // This should shutdown all the running services and close the database226 // Note that Shutdown should be graceful and not shutdown anything not running.227 if s.trtl != nil {228 if err = s.trtl.Shutdown(); err != nil {229 return err230 }231 }232 // Shutdown the gRPC connection if it's running233 if s.grpc != nil {234 s.grpc.Release()235 }236 // Cleanup the tmpdb and delete any stray files237 if s.tmpdb != "" {238 os.RemoveAll(s.tmpdb)239 }240 // Cleanup the remote trtl peer241 if s.remote != nil {242 s.remote.Shutdown()243 }244 // Reset all of the test suite variables245 s.tmpdb = ""246 s.conf = nil247 s.grpc = nil248 s.trtl = nil249 s.remote = nil250 return nil251}252// reset the test environment, refreshing the honu database fixture and all of the253// services. This is useful if the test makes changes to the database, though it is254// somewhat heavyweight since it blows away the prior configuration, running servers,255// and open database connection.256func (s *trtlTestSuite) reset() {257 require := s.Require()258 // Reset the previous environment259 s.resetEnvironment()260 // Run the trtl server on the new configuration261 require.NoError(s.setupServers(), "could not reset servers")262 // Run the remote trtl peer263 require.NoError(s.setupRemoteTrtl(), "could not reset remote trtl")264}265// shutdown the trtl servers and reset the configuration and fixtures266func (s *trtlTestSuite) resetEnvironment() {267 require := s.Require()268 // Cleanup previous configuration and shutdown servers, deleting the tmp database.269 require.NoError(s.cleanup(), "could not cleanup before reset")270 // Setup a new configuration and tmpdb271 require.NoError(s.setupConfig(), "could not reset configuration")272 // Extract the honu db fixture into tmpdb273 require.NoError(s.extractDB(), "could not reset db")274}275//===========================================================================276// Test fixtures management277//===========================================================================278// loads client credentials from disk and returns grpc.DialOptions to use for TLS279// client connections.280func (s *trtlTestSuite) loadClientCredentials() (opts []grpc.DialOption, err error) {281 opts = make([]grpc.DialOption, 0)282 // Load the client credentials from the fixtures283 var sz *trust.Serializer284 if sz, err = trust.NewSerializer(false); err != nil {285 return nil, err286 }287 var pool trust.ProviderPool288 if pool, err = sz.ReadPoolFile(clientCerts); err != nil {289 return nil, err290 }291 var provider *trust.Provider292 if provider, err = sz.ReadFile(clientCerts); err != nil {293 return nil, err294 }295 // Create the TLS configuration from the client credentials296 var cert tls.Certificate297 if cert, err = provider.GetKeyPair(); err != nil {298 return nil, err299 }300 var certPool *x509.CertPool301 if certPool, err = pool.GetCertPool(false); err != nil {302 return nil, err303 }304 var u *url.URL305 if u, err = url.Parse(serverTarget); err != nil {306 return nil, err307 }308 conf := &tls.Config{309 ServerName: u.Host,310 Certificates: []tls.Certificate{cert},311 RootCAs: certPool,312 }313 opts = append(opts, grpc.WithTransportCredentials(credentials.NewTLS(conf)))314 return opts, nil315}316// loads the test fixtures from a JSON file and stores them in the dbFixtures map -317// these fixtures are used both for generating the test database and for making318// comparative assertions in test code.319func (s *trtlTestSuite) loadFixtures() (err error) {320 var fixtures []byte321 if fixtures, err = ioutil.ReadFile("testdata/db.json"); err != nil {322 return fmt.Errorf("could not read fixtures at testdata/db.json: %s", err)323 }324 if err = json.Unmarshal(fixtures, &dbFixtures); err != nil {325 return fmt.Errorf("could not unmarshal fixtures at testdata/db.json: %s", err)326 }327 return nil328}329// extract honu database from generated fixtures - the honu db fixture is interacted330// with directly by the code and we expect that the data in the honu db fixture mates331// the in-memory fixtures defined by dbFixtures. If there is a mismatch, delete332// testdata/db.tgz or run the tests with the -update flag to regenerate them.333func (s *trtlTestSuite) extractDB() (err error) {334 if s.tmpdb == "" {335 return errors.New("no temporary database, run s.setupConfig first")336 }337 // Always extract the test database to a temporary directory.338 if _, err = utils.ExtractGzip(dbtgz, s.tmpdb, true); err != nil {339 return fmt.Errorf("unable to extract honu db fixture: %s", err)340 }341 return nil342}343// generates an updated database and compresses it to a gzip file.344// NOTE: loadFixtures must have been called before this method.345func (s *trtlTestSuite) generateDB() (err error) {346 // Create a new and temporary db to write the fixtures into347 if err = s.setupConfig(); err != nil {348 return err349 }350 // Ensure we cleanup so that subsequent tests can generate tmpdb directories351 defer s.cleanup()352 // Open a honu database, all fixtures will be written by Honu, which means that Honu353 // will be performing all version management, we expect that everything is at the354 // first version when the fixtures database is created.355 var db *honu.DB356 if db, err = honu.Open(s.conf.Database.URL, s.conf.GetHonuConfig()); err != nil {357 return fmt.Errorf("could not open tmp honu database in %s: %s", s.tmpdb, err)358 }...

Full Screen

Full Screen

dbfixtures.go

Source:dbfixtures.go Github

copy

Full Screen

...83 }84 venom.Debug(ctx, "applied %d migrations\n", n)85 }86 // Load fixtures in the databases.87 if err = loadFixtures(ctx, db, e.Files, e.Folder, getDialect(e.Database, e.SkipResetSequences), workdir); err != nil {88 return nil, err89 }90 r := Result{Executor: e}91 return r, nil92}93// ZeroValueResult return an empty implementation of this executor result94func (Executor) ZeroValueResult() interface{} {95 return Result{}96}97// GetDefaultAssertions return the default assertions of the executor.98func (e Executor) GetDefaultAssertions() venom.StepAssertions {99 return venom.StepAssertions{Assertions: []venom.Assertion{}}100}101// loadFixtures loads the fixtures in the database.102// It gives priority to the fixtures files found in folder,103// and switch to the list of files if no folder was specified.104func loadFixtures(ctx context.Context, db *sql.DB, files []string, folder string, dialect func(*fixtures.Loader) error, workdir string) error {105 if folder != "" {106 venom.Debug(ctx, "loading fixtures from folder %s\n", path.Join(workdir, folder))107 loader, err := fixtures.New(108 // By default the package refuse to load if the database109 // does not contains "test" to avoid wiping a production db.110 fixtures.DangerousSkipTestDatabaseCheck(),111 fixtures.Database(db),112 fixtures.Directory(path.Join(workdir, folder)),113 dialect)114 if err != nil {115 return errors.Wrapf(err, "failed to create folder loader")116 }117 if err = loader.Load(); err != nil {118 return errors.Wrapf(err, "failed to load fixtures from folder %q", path.Join(workdir, folder))...

Full Screen

Full Screen

repository_test.go

Source:repository_test.go Github

copy

Full Screen

...83 sort.Strings(resultChirps[0].Tags)84 assert.Equal(s.T(), reflect.DeepEqual(resultChirps[0].Tags, pq.StringArray{"tag1", "tag2"}), true, "chirp should contain correct tags")85}86func (s *Suite) Test_repository_GetChirps() {87 s.loadFixtures()88 resultChirps, err := s.repository.GetChirps([]string{"tag1"})89 require.NoError(s.T(), err)90 assert.Equal(s.T(), len(resultChirps), 1, "there should be one chirp")91 assert.Equal(s.T(), resultChirps[0].Message, "test_message", "chirp should contain correct message")92 assert.Equal(s.T(), resultChirps[0].Author, "test_user", "chirp should contain correct author")93 sort.Strings(resultChirps[0].Tags)94 assert.Equal(s.T(), reflect.DeepEqual(resultChirps[0].Tags, pq.StringArray{"tag1", "tag2"}), true, "chirp should contain correct tags")95}96func (s *Suite) Test_repository_CountChirps() {97 s.loadFixtures()98 chirpsCount, err := s.repository.CountChirps("2016-01-01", "2016-01-02", []string{"tag1"})99 require.NoError(s.T(), err)100 assert.Equal(s.T(), chirpsCount, 1, "there should be one chirp")101}102func (s *Suite) clearDB() {103 tx := s.DB.MustBegin()104 tx.MustExec(`DELETE FROM chirps_tags;`)105 tx.MustExec(`DELETE FROM chirps;`)106 tx.MustExec(`DELETE FROM tags;`)107 tx.MustExec(`DELETE FROM users;`)108 if err := tx.Commit(); err != nil {109 log.Fatal(err)110 }111}112func (s *Suite) loadFixtures() {113 err := s.fixtures.Load()114 if err != nil {115 log.Fatal(errors.Wrap(err, "could not load fixtures"))116 }117}...

Full Screen

Full Screen

loadFixtures

Using AI Code Generation

copy

Full Screen

1import (2type ApiTestSuite struct {3}4func (s *ApiTestSuite) SetupSuite() {5 s.server = httptest.NewServer(server.NewRouter())6}7func (s *ApiTestSuite) TearDownSuite() {8 s.server.Close()9}10func (s *ApiTestSuite) TestGetUsers() {11 fixtures := tests.LoadFixtures(s.T(), "users")12 defer fixtures.Clean()13 Body(`[14 {

Full Screen

Full Screen

loadFixtures

Using AI Code Generation

copy

Full Screen

1dbfixtures.loadFixtures('fixtures', function (err) {2 if (err) {3 console.log(err);4 } else {5 console.log('Fixtures loaded');6 }7});8dbfixtures.loadFixtures('fixtures', function (err) {9 if (err) {10 console.log(err);11 } else {12 console.log('Fixtures loaded');13 }14});15dbfixtures.loadFixtures('fixtures', function (err) {16 if (err) {17 console.log(err);18 } else {19 console.log('Fixtures loaded');20 }21});22dbfixtures.loadFixtures('fixtures', function (err) {23 if (err) {24 console.log(err);25 } else {26 console.log('Fixtures loaded');27 }28});29dbfixtures.loadFixtures('fixtures', function (err) {30 if (err) {31 console.log(err);32 } else {33 console.log('Fixtures loaded');34 }35});36dbfixtures.loadFixtures('fixtures', function (err) {37 if (err) {38 console.log(err);39 } else {40 console.log('Fixtures loaded');41 }42});43dbfixtures.loadFixtures('fixtures', function (err) {44 if (err) {45 console.log(err);46 } else {47 console.log('Fixtures loaded');48 }49});50dbfixtures.loadFixtures('fixtures', function (err) {51 if (err) {52 console.log(err);53 } else {54 console.log('Fixtures loaded');55 }56});57dbfixtures.loadFixtures('fixtures', function (err) {58 if (err) {59 console.log(err);60 }

Full Screen

Full Screen

loadFixtures

Using AI Code Generation

copy

Full Screen

1import (2func TestLoadFixtures(t *testing.T) {3 db := NewDB()4 db.LoadFixtures()5 user := db.GetUserById(1)6 assert.Equal(t, "John", user.Name)7 assert.Equal(t, "Doe", user.Surname)8 assert.Equal(t, "

Full Screen

Full Screen

loadFixtures

Using AI Code Generation

copy

Full Screen

1func main() {2 dbFixtures := dbfixtures.NewDbFixtures(db)3 err := dbFixtures.LoadFixtures("path/to/fixtures")4 if err != nil {5 log.Fatal(err)6 }7}8func main() {9 dbFixtures := dbfixtures.NewDbFixtures(db)10 err := dbFixtures.LoadFixtures("path/to/fixtures")11 if err != nil {12 log.Fatal(err)13 }14}15func main() {16 dbFixtures := dbfixtures.NewDbFixtures(db)17 err := dbFixtures.LoadFixtures("path/to/fixtures")18 if err != nil {19 log.Fatal(err)20 }21}22func main() {23 dbFixtures := dbfixtures.NewDbFixtures(db)24 err := dbFixtures.LoadFixtures("path/to/fixtures")25 if err != nil {26 log.Fatal(err)27 }28}29func main() {30 dbFixtures := dbfixtures.NewDbFixtures(db)31 err := dbFixtures.LoadFixtures("path/to/fixtures")32 if err != nil {33 log.Fatal(err)34 }35}36func main() {37 dbFixtures := dbfixtures.NewDbFixtures(db)38 err := dbFixtures.LoadFixtures("path/to/fixtures")39 if err != nil {40 log.Fatal(err)41 }42}43func main() {44 dbFixtures := dbfixtures.NewDbFixtures(db)45 err := dbFixtures.LoadFixtures("path/to/fixtures")46 if err != nil {47 log.Fatal(err)48 }49}50func main() {51 dbFixtures := dbfixtures.NewDbFixtures(db)52 err := dbFixtures.LoadFixtures("path/to/fixtures")53 if err != nil {54 log.Fatal(err)55 }56}

Full Screen

Full Screen

loadFixtures

Using AI Code Generation

copy

Full Screen

1func main() {2 dbfixtures.LoadFixtures("fixtures", "users")3 dbfixtures.LoadFixtures("fixtures", "posts")4 dbfixtures.LoadFixtures("fixtures", "comments")5}6func main() {7 dbfixtures.LoadFixtures("fixtures", "users")8 dbfixtures.LoadFixtures("fixtures", "posts")9 dbfixtures.LoadFixtures("fixtures", "comments")10}11func main() {12 dbfixtures.LoadFixtures("fixtures", "users")13 dbfixtures.LoadFixtures("fixtures", "posts")14 dbfixtures.LoadFixtures("fixtures", "comments")15}16func main() {17 dbfixtures.LoadFixtures("fixtures", "users")18 dbfixtures.LoadFixtures("fixtures", "posts")19 dbfixtures.LoadFixtures("fixtures", "comments")20}

Full Screen

Full Screen

Automation Testing Tutorials

Learn to execute automation testing from scratch with LambdaTest Learning Hub. Right from setting up the prerequisites to run your first automation test, to following best practices and diving deeper into advanced test scenarios. LambdaTest Learning Hubs compile a list of step-by-step guides to help you be proficient with different test automation frameworks i.e. Selenium, Cypress, TestNG etc.

LambdaTest Learning Hubs:

YouTube

You could also refer to video tutorials over LambdaTest YouTube channel to get step by step demonstration from industry experts.

Run Venom automation tests on LambdaTest cloud grid

Perform automation testing on 3000+ real desktop and mobile devices online.

... View sample repo... View DocumentationSign up Free
›
_

Most used method in

Try LambdaTest Now !!

Get 100 minutes of automation test minutes FREE!!

...
Sign up with GoogleSign up with Email
Next-Gen App & Browser Testing Cloud

Was this article helpful?

Helpful

NotHelpful